jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Updated test_failed_weak_ref

pull/296/merge
Sheng 2022-11-21 16:39:13 +08:00
parent 2a46b52eac
commit 9238c01c35
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_handler.py
View File

@ -283,9 +283,11 @@ class TestWsockHandler(unittest.TestCase):
request = HTTPServerRequest(uri='/') request = HTTPServerRequest(uri='/')
obj = Mock(spec=WsockHandler, request=request) obj = Mock(spec=WsockHandler, request=request)
obj.src_addr = ("127.0.0.1", 8888) obj.src_addr = ("127.0.0.1", 8888)
class FakeWeakRef: class FakeWeakRef:
def __init__(self): def __init__(self):
self.count = 0 self.count = 0
def __call__(self): def __call__(self):
self.count += 1 self.count += 1
return None return None
@ -294,3 +296,4 @@ class TestWsockHandler(unittest.TestCase):
obj.worker_ref = ref obj.worker_ref = ref
WsockHandler.on_message(obj, b'{"data": "somestuff"}') WsockHandler.on_message(obj, b'{"data": "somestuff"}')
self.assertGreaterEqual(ref.count, 1) self.assertGreaterEqual(ref.count, 1)
obj.close.assert_called_with(reason='No worker found')