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)