Improve folder and tag creation api test.

2016-03-14 20:38:00 -04:00
parent 9823202405
commit c2c95787cb
2 changed files with 6 additions and 6 deletions
--- a/mayan/apps/folders/tests/test_api.py
+++ b/mayan/apps/folders/tests/test_api.py
@@ -37,11 +37,11 @@ class FolderAPITestCase(APITestCase):
            reverse('rest_api:folder-list'), {'label': TEST_FOLDER_LABEL}
        )

-        self.assertEqual(response.data['id'], 1)
-        self.assertEqual(response.data['label'], TEST_FOLDER_LABEL)
-
        folder = Folder.objects.first()

+        self.assertEqual(response.data['id'], folder.pk)
+        self.assertEqual(response.data['label'], TEST_FOLDER_LABEL)
+
        self.assertEqual(Folder.objects.count(), 1)
        self.assertEqual(folder.label, TEST_FOLDER_LABEL)