def test_no_ip_connected(self): ProcessTask.objects.create(name="example.Foo", args=[1], params={'bar': 'baz'}) response = self.client.get(self.url) request = APIRequestFactory().get(self.url) self.assertEqual(response.status_code, status.HTTP_200_OK) tasks = ProcessTask.objects.all() serializer = ProcessTaskSerializer(tasks, many=True, context={'request': request}) self.assertEqual(response.data, serializer.data)
def test_user_in_organization(self): membership = GroupMember.objects.create(member=self.member, group=self.org) membership.roles.add(self.user_role) ProcessTask.objects.create(name="example.Foo", args=[1], params={'bar': 'baz'}, information_package=self.ip) response = self.client.get(self.url) request = APIRequestFactory().get(self.url) self.assertEqual(response.status_code, status.HTTP_200_OK) tasks = ProcessTask.objects.all() serializer = ProcessTaskSerializer(tasks, many=True, context={'request': request}) self.assertEqual(len(response.data), len(serializer.data)) self.assertEqual(response.data, serializer.data)
def test_authenticated(self): # get API response self.client.force_authenticate(user=self.user) request = APIRequestFactory().get(self.url) response = self.client.get(self.url) # get data from DB tasks = ProcessTask.objects.all() serializer = ProcessTaskSerializer(tasks, many=True, context={'request': request}) self.assertEqual(response.data, serializer.data) self.assertEqual(response.status_code, status.HTTP_200_OK)