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 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 get_object(self):
uid = self.kwargs['uid']
user = ServiceUser.get_by_uid(uid)
return user
