def test_removal(self): su = ServiceUser(UID=1) su.save() su2 = ServiceUser(UID=2) su2.save() ServiceUser.connect(1, 2) try: ServiceUser.remove('aaa', 1) except Exception as e: self.assertIsInstance(e, InvalidUIDError) try: ServiceUser.remove(1, 'bbb') except Exception as e: self.assertIsInstance(e, InvalidUIDError) try: ServiceUser.remove(1, 7) except Exception as e: self.assertIsInstance(e, NonExistingUIDError) try: ServiceUser.remove(1, 1) except Exception as e: self.assertIsInstance(e, EqualUIDError) ServiceUser.remove(1, 2) self.assertEquals(len(su.connected_users.all()), 0) ServiceUser.remove(1, 2) self.assertEquals(len(su.connected_users.all()), 0)
def test_connection(self): su = ServiceUser(UID=1) su.save() su2 = ServiceUser(UID=2) su2.save() try: ServiceUser.connect('aaa', 1) except Exception as e: self.assertIsInstance(e, InvalidUIDError) try: ServiceUser.connect(1, 'bbb') except Exception as e: self.assertIsInstance(e, InvalidUIDError) try: ServiceUser.connect(1, 7) except Exception as e: self.assertIsInstance(e, NonExistingUIDError) try: ServiceUser.connect(1, 1) except Exception as e: self.assertIsInstance(e, EqualUIDError) ServiceUser.connect(1, 2) self.assertEquals(len(su.connected_users.all()), 1) self.assertEquals(su.connected_users.first(), ServiceUser.get_by_uid(2)) ServiceUser.connect(1, 2) self.assertEquals(len(su.connected_users.all()), 1) self.assertEquals(su.connected_users.first(), ServiceUser.get_by_uid(2))
def test_getting_by_uid(self): su = ServiceUser(UID=1) su.save() try: ServiceUser.get_by_uid(2) except Exception as e: self.assertIsInstance(e, NonExistingUIDError) su = ServiceUser.get_by_uid(1) self.assertIsInstance(su, ServiceUser) self.assertEquals(su, ServiceUser.objects.get(UID=1))
def test_creation(self): try: ServiceUser(UID='aaa') except Exception as e: self.assertIsInstance(e, InvalidUIDError) try: ServiceUser(UID=-1) except Exception as e: self.assertIsInstance(e, InvalidUIDError) su = ServiceUser(UID=1) su.save() self.assertIsInstance(su, ServiceUser) self.assertEquals(su, ServiceUser.objects.get(UID=1))
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())