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