def test_remove_client(self): # It removes the client remaining_client = Client(websockets.protocol.WebSocketCommonProtocol(host="1")) removed_client = Client(websockets.protocol.WebSocketCommonProtocol(host="2")) channel = Channel("test-channel") channel.add_client(remaining_client) channel.add_client(removed_client) self.assertTrue(len(channel.clients) == 2) # It removes a client channel.remove_client(removed_client) self.assertTrue(len(channel.clients) == 1) # It removes the right client actual_remaining_client = channel.clients.pop() self.assertEqual(remaining_client, actual_remaining_client)