From 1cadee077c5418ac0d9df749831cd4279a06691e Mon Sep 17 00:00:00 2001 From: Miguel Grinberg Date: Sun, 24 Mar 2024 09:33:36 +0000 Subject: [PATCH] add unit test --- tests/common/test_server.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/common/test_server.py b/tests/common/test_server.py index b568c68..6e074ae 100644 --- a/tests/common/test_server.py +++ b/tests/common/test_server.py @@ -578,16 +578,21 @@ class TestServer(unittest.TestCase): s = server.Server(async_handlers=False) sid_foo = s.manager.connect('123', '/foo') sid_bar = s.manager.connect('123', '/bar') + sid_baz = s.manager.connect('123', '/baz') connect_star_handler = mock.MagicMock() msg_foo_handler = mock.MagicMock() msg_star_handler = mock.MagicMock() star_foo_handler = mock.MagicMock() star_star_handler = mock.MagicMock() + my_message_baz_handler = mock.MagicMock() + star_baz_handler = mock.MagicMock() s.on('connect', connect_star_handler, namespace='*') s.on('msg', msg_foo_handler, namespace='/foo') s.on('msg', msg_star_handler, namespace='*') s.on('*', star_foo_handler, namespace='/foo') s.on('*', star_star_handler, namespace='*') + s.on('my message', my_message_baz_handler, namespace='/baz') + s._trigger_event('connect', '/bar', sid_bar) s._handle_eio_message('123', '2/foo,["msg","a","b"]') s._handle_eio_message('123', '2/bar,["msg","a","b"]') @@ -602,6 +607,11 @@ class TestServer(unittest.TestCase): star_star_handler.assert_called_once_with( 'my message', '/bar', sid_bar, 'a', 'b', 'c') + s._handle_eio_message('123', '2/baz,["my message","a","b","c"]') + s._handle_eio_message('123', '2/baz,["msg","a","b"]') + my_message_baz_handler.assert_called_once_with(sid_baz, 'a', 'b', 'c') + msg_star_handler.assert_called_with('/baz', sid_baz, 'a', 'b') + def test_handle_event_with_disconnected_namespace(self, eio): s = server.Server(async_handlers=False) s.manager.connect('123', '/foo')