def test_get_address_list(self): rm = Room("ad1", "al1", "newRoom") compare = ["ad1"] for i in range(2, 45): ad = "ad" + str(i) al = "al" + str(i) rm.add_user(ad, al) compare.append(ad) self.assertEqual(rm.get_address_list(), compare)
def test_remove_by_address(self): rm = Room("ad1", "alA", "newRoom") rm.add_user("ad2", "alB") self.assertEqual(rm.remove_user_by_address("ad1"), "alA") self.assertEqual(rm.get_address_list(), ["ad2"]) self.assertEqual(rm.get_alias_list(), ["alB"])
def test_initializer(self): rm = Room("ad1", "alA", "newRoom") self.assertEqual(rm.get_address_list(), ["ad1"]) self.assertEqual(rm.get_alias_list(), ["alA"]) self.assertEqual(rm.get_name(), "newRoom")