Unit test in progress...

TravisCarden · TravisCarden · commit b5d8d759de31 · 2016-09-19T22:06:59.000-05:00
diff --git a/src/Symfony/Component/HttpFoundation/Tests/File/FileTest.php b/src/Symfony/Component/HttpFoundation/Tests/File/FileTest.php
@@ -13,6 +13,7 @@
 
 use Symfony\Component\HttpFoundation\File\File;
 use Symfony\Component\HttpFoundation\File\MimeType\MimeTypeGuesser;
+use Symfony\Component\HttpFoundation\File\Tests\Fixtures\PaddingStreamWrapper;
 
 class FileTest extends \PHPUnit_Framework_TestCase
 {
@@ -28,6 +29,24 @@ public function testGetMimeTypeUsesMimeTypeGuessers()
         $this->assertEquals('image/gif', $file->getMimeType());
     }
 
+    public function testGetSizeWithStreamFilter()
+    {
+        // @todo Where should these fixture classes go so this isn't necessary?
+        require_once 'Fixtures/PaddingStreamWrapper.php';
+        require_once 'Fixtures/PaddingStreamFilter.php';
+
+        stream_wrapper_register('pad', PaddingStreamWrapper::class);
+        $path = 'pad://test';
+
+        $file = new File($path);
+
+        ob_start();
+        $filtered_size = readfile($path);
+        ob_end_clean();
+
+        $this->assertEquals($filtered_size, $file->getSize());
+    }
+
     public function testGuessExtensionWithoutGuesser()
     {
         $file = new File(__DIR__.'/Fixtures/directory/.empty');
diff --git a/src/Symfony/Component/HttpFoundation/Tests/File/Fixtures/PaddingStreamFilter.php b/src/Symfony/Component/HttpFoundation/Tests/File/Fixtures/PaddingStreamFilter.php
@@ -0,0 +1,21 @@
+<?php
+
+namespace Symfony\Component\HttpFoundation\File\Tests\Fixtures;
+
+class PaddingStreamFilter extends \php_user_filter
+{
+    const NAME = 'pad';
+
+    const PADDING = '     ';
+
+    public function filter($in, $out, &$consumed, $closing)
+    {
+        while ($bucket = stream_bucket_make_writeable($in)) {
+            $bucket->data .= self::PADDING;
+            $consumed += $bucket->datalen;
+            stream_bucket_append($out, $bucket);
+        }
+
+        return PSFS_PASS_ON;
+    }
+}
diff --git a/src/Symfony/Component/HttpFoundation/Tests/File/Fixtures/PaddingStreamWrapper.php b/src/Symfony/Component/HttpFoundation/Tests/File/Fixtures/PaddingStreamWrapper.php
@@ -0,0 +1,62 @@
+<?php
+
+namespace Symfony\Component\HttpFoundation\File\Tests\Fixtures;
+
+class PaddingStreamWrapper
+{
+    protected $file;
+
+    public function stream_open($path, $mode, $options, &$opened_path)
+    {
+        $this->file = fopen('php://temp', 'r+');
+
+        $path_parts = explode('://', $path);
+        $file_contents = end($path_parts);
+        fwrite($this->file, $file_contents);
+        fseek($this->file, 0);
+
+        stream_filter_register(PaddingStreamFilter::NAME, PaddingStreamFilter::class);
+        stream_filter_append($this->file, PaddingStreamFilter::NAME, STREAM_FILTER_READ);
+
+        return (bool) $this->file;
+    }
+
+    public function stream_read($count)
+    {
+        return fread($this->file, $count);
+    }
+
+    public function stream_write($data)
+    {
+        return fwrite($this->file, $data);
+    }
+
+    public function stream_tell()
+    {
+        return ftell($this->file);
+    }
+
+    public function stream_eof()
+    {
+        return feof($this->file);
+    }
+
+    public function stream_seek($offset, $whence)
+    {
+        return fseek($this->file, $offset, $whence);
+    }
+
+    public function stream_stat()
+    {
+        return fstat($this->file);
+    }
+
+    public function url_stat($path, $flags)
+    {
+        $file = fopen($path, 'r');
+        $stat = fstat($file);
+        fclose($file);
+
+        return $stat;
+    }
+}
