implemented required fields for csv data load

This commit is contained in:
Sam
2021-02-03 11:32:58 +00:00
parent 569ba6515b
commit e6ada44704
3 changed files with 13 additions and 8 deletions

View File

@@ -403,6 +403,7 @@ class LoadCoopManagerTestCase(APITestCase):
self.csv_path = 'datasets/test_coop.csv'
def test_admin_can_load_csv(self):
# delete all instances
company_count = self.company_model.objects.count()
user_count = self.user_model.objects.count()
@@ -420,8 +421,8 @@ class LoadCoopManagerTestCase(APITestCase):
# check re sponse
self.assertEqual(response.status_code, 200)
# check for object creation
self.assertNotEqual(company_count, self.company_model.objects.count())
self.assertNotEqual(user_count, self.user_model.objects.count())
self.assertEquals(company_count + 5, self.company_model.objects.count())
self.assertEquals(user_count + 5, self.user_model.objects.count())
def test_auth_user_cannot_load_csv(self):
company_count = self.company_model.objects.count()