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)