Simplify OutputBuffer tests

amphp · Oct 17, 2017 · e00c6d4 · e00c6d4
1 parent 48bb2eb
commit e00c6d4
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/test/OutputBufferTest.php b/test/OutputBufferTest.php
@@ -3,7 +3,6 @@
 namespace Amp\ByteStream\Test;
 
 use Amp\ByteStream\ClosedException;
-use Amp\ByteStream\InMemoryStream;
 use Amp\ByteStream\OutputBuffer;
 use Amp\Loop;
 use Amp\PHPUnit\TestCase;
@@ -15,9 +14,7 @@ public function testWrite() {
             $output->write('foo');
             $output->end();
 
-            $memory = new InMemoryStream(yield $output);
-
-            $this->assertSame('foo', yield $memory->read());
+            $this->assertSame('foo', yield $output);
         });
     }
 
@@ -27,9 +24,7 @@ public function testEnd() {
             $output->write('foo');
             $output->end('bar');
 
-            $memory = new InMemoryStream(yield $output);
-
-            $this->assertSame('foobar', yield $memory->read());
+            $this->assertSame('foobar', yield $output);
         });
     }