def test_handle_message(self): room = Chatroom("../config/test_config/chat.yaml") fake_websocket = Mwsc() fake_websocket2 = Mwsc() self.connect_fake_client(fake_websocket, room) self.connect_fake_client(fake_websocket2, room) asyncio.get_event_loop().run_until_complete( room.handle_message(fake_websocket, "Fake message")) self.assertIn(fake_websocket, room.connected) self.assertEqual(room.connected[fake_websocket].websocket, fake_websocket) self.assertTrue( len(room.connected[fake_websocket2].websocket.incoming) > 1)
def test_send_to_all(self): room = Chatroom("../config/test_config/chat.yaml") fake_websocket = Mwsc() fake_websocket2 = Mwsc() self.connect_fake_client(fake_websocket, room) self.connect_fake_client(fake_websocket2, room) resp = Response("TO ALL") asyncio.get_event_loop().run_until_complete( room.send_to_all(resp, fake_websocket2)) self.assertEqual(resp.json(), room.connected[fake_websocket].websocket.incoming[-1]) self.assertNotEqual( resp.json(), room.connected[fake_websocket2].websocket.incoming[-1])
def test_handle_disconnect(self): room = Chatroom("../config/test_config/chat.yaml") fake_websocket = Mwsc() self.assertEqual(len(room.connected), 0) self.connect_fake_client(fake_websocket, room) self.assertEqual(len(room.connected), 1) asyncio.get_event_loop().run_until_complete( room.handle_disconnect(fake_websocket)) self.assertEqual(len(room.connected), 0)
def test_handle_connection(self): room = Chatroom("../config/test_config/chat.yaml") fake_websocket = Mwsc() self.connect_fake_client(fake_websocket, room) self.assertIn(fake_websocket, room.connected) self.assertEqual(room.connected[fake_websocket].websocket, fake_websocket) self.assertTrue( len(room.connected[fake_websocket].websocket.incoming) > 0) self.assertEqual(len(room.connected), 1)