def test_getting_connected_users(self): su = ServiceUser(UID=1) su.save() su2 = ServiceUser(UID=2) su2.save() ServiceUser.connect(1, 2) try: ServiceUser.get_connected_users(3) except Exception as e: self.assertIsInstance(e, NonExistingUIDError) con_list = ServiceUser.get_connected_users(1) self.assertEquals(list(con_list), list([su2])) con_list = ServiceUser.get_connected_users(2) self.assertEquals(list(con_list), list([su])) ServiceUser.remove(1, 2) con_list = ServiceUser.get_connected_users(1) self.assertEquals(list(con_list), list()) con_list = ServiceUser.get_connected_users(2) self.assertEquals(list(con_list), list())