Merge a5e908a into 2120bdd

amphp · Mar 23, 2019 · 5a40af1 · 5a40af1
2 parents 2120bdd + a5e908a
commit 5a40af1
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/lib/ResourceInputStream.php b/lib/ResourceInputStream.php
@@ -204,6 +204,11 @@ public function getResource()
         return $this->resource;
     }
 
+    public function setChunkSize(int $chunkSize)
+    {
+        $this->chunkSize = $chunkSize;
+    }
+
     /**
      * References the read watcher, so the loop keeps running in case there's an active read.
      *

diff --git a/lib/ResourceOutputStream.php b/lib/ResourceOutputStream.php
@@ -274,6 +274,11 @@ public function getResource()
         return $this->resource;
     }
 
+    public function setChunkSize(int $chunkSize)
+    {
+        $this->chunkSize = $chunkSize;
+    }
+
     public function __destruct()
     {
         if ($this->resource !== null) {

diff --git a/test/ResourceStreamTest.php b/test/ResourceStreamTest.php
@@ -274,4 +274,19 @@ public function testIssue47()
             $this->assertStringEqualsFile(__FILE__, $buffer);
         });
     }
+
+    public function testSetChunkSize()
+    {
+        Loop::run(function () {
+            list($a, $b) = $this->getStreamPair();
+            $a->setChunkSize(1);
+            $b->setChunkSize(1);
+
+            $this->assertSame(3, yield $a->write('foo'));
+            $this->assertSame('f', yield $b->read());
+
+            $b->setChunkSize(3);
+            $this->assertSame('oo', yield $b->read());
+        });
+    }
 }