def test_get_valid_groups(self): """Test getting valid groups.""" GroupFactory.create_batch(2) response = self.client.get(reverse('group-detail', kwargs={'pk': 1})) self.assertEqual(response.status_code, status.HTTP_200_OK) response = self.client.get(reverse('group-list')) self.assertEqual(response.status_code, status.HTTP_200_OK)
def create_groups(max_groups): """ Create groups. :param max_groups: Maximum count of groups. :return: Created groups. """ groups = GroupFactory.create_batch(max_groups) return groups
def test_get_invalid_group(self): """Test getting invalid groups.""" GroupFactory.create_batch(2) response = self.client.get(reverse('group-detail', kwargs={'pk': 999})) self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)