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_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 post(self, request, *args, **kwargs):
uid = request.POST['uid']
try:
serializer = ConnectionsListSerializer(instance=ServiceUser.get_by_uid(uid))
result = response.Response(serializer.data, status=status.HTTP_200_OK)
except (InvalidUIDError, NonExistingUIDError):
result = response.Response(status=status.HTTP_400_BAD_REQUEST)
return result
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_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)
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 get_object(self):
uid = self.kwargs['uid']
user = ServiceUser.get_by_uid(uid)
return user
