implemented required fields for csv data load
This commit is contained in:
@@ -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()
|
||||
|
||||
@@ -103,6 +103,9 @@ def load_coop_managers(request):
|
||||
coop_counter = 0
|
||||
user_counter = 0
|
||||
for row in csv_reader:
|
||||
if '' in (row['cif'], row['nombre-coop'], row['email']):
|
||||
logging.error(f"Required data missing: {row}")
|
||||
continue
|
||||
try:
|
||||
coop_data = {
|
||||
'cif': row['cif'].strip(),
|
||||
@@ -111,6 +114,7 @@ def load_coop_managers(request):
|
||||
'shop': bool(row['es-tienda'].strip()),
|
||||
'shop_link': row['url'].strip(),
|
||||
}
|
||||
# import ipdb; ipdb.set_trace()
|
||||
coop = Company.objects.create(**coop_data)
|
||||
logging.info(f"Created Coop: {coop_data}")
|
||||
coop_counter += 1
|
||||
|
||||
Reference in New Issue
Block a user