def post(self, request, *args, **kwargs): uid1 = request.POST['uid1'] uid2 = request.POST['uid2'] try: ServiceUser.remove(uid1, uid2) result = response.Response({u'first_uid': uid1, u'second_uid': uid2}, status=status.HTTP_200_OK) except (InvalidUIDError, NonExistingUIDError, EqualUIDError): result = response.Response(status=status.HTTP_400_BAD_REQUEST) return result
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())
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)