Added test_worker_closed

huashengdun · Nov 21, 2022 · 50fc990 · 50fc990
1 parent 9238c01
commit 50fc990
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_handler.py b/tests/test_handler.py
@@ -297,3 +297,21 @@ def __call__(self):
         WsockHandler.on_message(obj, b'{"data": "somestuff"}')
         self.assertGreaterEqual(ref.count, 1)
         obj.close.assert_called_with(reason='No worker found')
+
+    def test_worker_closed(self):
+        request = HTTPServerRequest(uri='/')
+        obj = Mock(spec=WsockHandler, request=request)
+        obj.src_addr = ("127.0.0.1", 8888)
+
+        class Worker:
+            def __init__(self):
+                self.closed = True
+
+        class FakeWeakRef:
+            def __call__(self):
+                return Worker()
+
+        ref = FakeWeakRef()
+        obj.worker_ref = ref
+        WsockHandler.on_message(obj, b'{"data": "somestuff"}')
+        obj.close.assert_called_with(reason='Worker closed')