Test removing both readable and writable side of stream when closing

clue · clue · commit 1416302619c8 · 2017-12-19T17:42:48.000+01:00
diff --git a/tests/AbstractLoopTest.php b/tests/AbstractLoopTest.php
@@ -184,6 +184,83 @@ public function testRemoveStreamForWriteOnly()
         $this->tickLoop($this->loop);
     }
 
+    public function testRemoveReadAndWriteStreamFromLoopOnceResourceClosesEndsLoop()
+    {
+        list($stream, $other) = $this->createSocketPair();
+        stream_set_blocking($stream, false);
+        stream_set_blocking($other, false);
+
+        // dummy writable handler
+        $this->loop->addWriteStream($stream, function () { });
+
+        // remove stream when the stream is readable (closes)
+        $this->loop->addReadStream($stream, function ($stream) {
+            $this->loop->removeReadStream($stream);
+            $this->loop->removeWriteStream($stream);
+            fclose($stream);
+        });
+
+        // close other side
+        fclose($other);
+
+        $this->assertRunFasterThan($this->tickTimeout);
+    }
+
+    public function testRemoveReadAndWriteStreamFromLoopOnceResourceClosesOnEndOfFileEndsLoop()
+    {
+        list($stream, $other) = $this->createSocketPair();
+        stream_set_blocking($stream, false);
+        stream_set_blocking($other, false);
+
+        // dummy writable handler to fill outgoing buffer
+        fwrite($stream, str_repeat('.', 60000));
+        $this->loop->addWriteStream($stream, function ($stream) use ($other) {
+            fwrite($stream, '.');
+
+            // close other side once outgoing buffer is filled
+            fclose($other);
+        });
+
+        // remove stream when the stream is readable (closes)
+        $never = $this->expectCallableNever();
+        $this->loop->addReadStream($stream, function ($stream) use ($never) {
+            $data = fread($stream, 1024);
+            if ($data !== '') {
+                // we expect to read an empty string => stream closed
+                return $never();
+            }
+
+            $this->loop->removeReadStream($stream);
+            $this->loop->removeWriteStream($stream);
+            fclose($stream);
+        });
+
+        $this->loop->addTimer(0.01, function () use ($other) {
+            fclose($other);
+        });
+
+        $this->assertRunFasterThan(0.1);
+    }
+
+    public function testRemoveReadAndWriteStreamFromLoopWithClosingResourceEndsLoop()
+    {
+        // get only one part of the pair to ensure the other side will close immediately
+        list($stream) = $this->createSocketPair();
+        stream_set_blocking($stream, false);
+
+        // dummy writable handler
+        $this->loop->addWriteStream($stream, function () { });
+
+        // remove stream when the stream is readable (closes)
+        $this->loop->addReadStream($stream, function ($stream) {
+            $this->loop->removeReadStream($stream);
+            $this->loop->removeWriteStream($stream);
+            fclose($stream);
+        });
+
+        $this->assertRunFasterThan($this->tickTimeout);
+    }
+
     public function testRemoveInvalid()
     {
         list ($stream) = $this->createSocketPair();
