From 1f8e9e582ad56c448757177f3357bdd2f6147d77 Mon Sep 17 00:00:00 2001 From: Joel Lavikainen Date: Sun, 18 Nov 2018 17:52:34 +0200 Subject: [PATCH] Make tag tests more generic --- webapp/tests.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/webapp/tests.py b/webapp/tests.py index 0a9f1fa..1d0c016 100644 --- a/webapp/tests.py +++ b/webapp/tests.py @@ -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)