def setUp(self): self.name_server = RegisterServer(4)
class TestNameServer(TestCase): def setUp(self): self.name_server = RegisterServer(4) def test_register_server(self): self.name_server.register_server('127.0.0.1:9090') self.assertEqual(self.name_server.list_servers(), ['127.0.0.1:9090']) with self.assertRaises(ValueError): self.name_server.register_server('127.0.0.1:9090') def test_list_rooms(self): self.assertEqual(self.name_server.list_rooms(), []) def test_create_room(self): pass def test_join_room(self): self.name_server.register_server('127.0.0.1:9090') self.name_server.join_room('War Room') self.assertEqual(self.name_server.list_rooms(), {'War Room'}) self.name_server.join_room('Peace Room') self.assertEqual(self.name_server.list_rooms(), {'War Room', 'Peace Room'}) # Test if it avoids duplicates self.name_server.join_room('Peace Room') self.assertEqual(self.name_server.list_rooms(), {'War Room', 'Peace Room'})