Make tag tests more generic

This commit is contained in:
Joel Lavikainen
2018-11-18 17:52:34 +02:00
parent 97e6ce3d8d
commit 1f8e9e582a
+8 -4
View File
@@ -25,7 +25,8 @@ class TagsTestCase(APITestCase):
# We dont care about icon, so response is sliced
sliced_response = OrderedDict(islice(response.data[0].items(), 3))
self.assertEqual(sliced_response, {'id': 1, 'slug': 'Party', 'name': 'Bileet'})
tag1 = Tag.objects.get(slug="Party")
self.assertEqual(sliced_response, {'id': tag1.id, 'slug': 'Party', 'name': 'Bileet'})
def test_get_multiple_tags(self):
self.assertEqual(Tag.objects.count(), 1)
@@ -37,12 +38,15 @@ class TagsTestCase(APITestCase):
self.assertTrue(status.is_success(response.status_code))
# We dont care about icon, so response is sliced
tag1 = Tag.objects.get(slug="Party")
sliced_response = OrderedDict(islice(response.data[0].items(), 3))
self.assertEqual(sliced_response, {'id': 1, 'slug': 'Party', 'name': 'Bileet'})
self.assertEqual(sliced_response, {'id': tag1.id, 'slug': 'Party', 'name': 'Bileet'})
sliced_response = OrderedDict(islice(response.data[1].items(), 3))
self.assertEqual(sliced_response, {'id': 2, 'slug': 'Freshmen', 'name': 'Fuksit'})
tag2 = Tag.objects.get(slug="Freshmen")
self.assertEqual(sliced_response, {'id': tag2.id, 'slug': 'Freshmen', 'name': 'Fuksit'})
sliced_response = OrderedDict(islice(response.data[2].items(), 3))
self.assertEqual(sliced_response, {'id': 3, 'slug': 'International', 'name': 'Ulkkarit'})
tag3 = Tag.objects.get(slug="International")
self.assertEqual(sliced_response, {'id': tag3.id, 'slug': 'International', 'name': 'Ulkkarit'})
def test_create_tag(self):
self.assertEqual(Tag.objects.count(), 1)