more work on load_coop_managers endpoint, unfinished

core/tests.py

@@ -3,14 +3,19 @@ import string
 import json
 import hashlib
 import base64
+import csv
 
 from django.test import TestCase
 
 from rest_framework.test import APITestCase
 from rest_framework import status
 
+import requests
+
 from core.utils import get_tokens_for_user
 
+from companies.models import Company
+
 from . import models
 from . import factories
 # Create your tests here.
@@ -376,3 +381,82 @@ class UpdateUserViewTest(APITestCase):
         # Assert forbidden code
         self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
 
+
+class LoadCoopManagerTestCase(TestCase):
+
+    def setUp(self):
+        """Tests setup
+        """
+        self.endpoint = 'http://127.0.0.1:8000/api/v1/load_coops/'
+        self.user_factory = factories.CustomUserFactory
+        self.user_model = models.CustomUser
+        self.company_model = Company
+        # create admin user
+        self.admin_email = f"admin_user@mail.com"
+        self.password = ''.join(random.choices(string.ascii_uppercase, k = 10))
+        self.admin_user = self.user_factory(email=self.admin_email, password=self.password, is_staff=True, is_active=True)
+        # create regular user
+        self.reg_email = f"user@mail.com"
+        self.user = self.user_factory(email=self.reg_email, is_active=True)
+        self.user.set_password(self.password)
+        self.user.save()
+
+    def test_admin_can_load_csv(self):
+        company_count = self.company_model.objects.count()
+        user_count = self.user_model.objects.count()
+
+        # read csv file
+        csv_file = '../coop.csv'
+        files = {'csv_file': open(csv_file,'rt')}
+
+        # Authenticate
+        token = get_tokens_for_user(self.admin_user)
+        headers = {'Authorization': f"Bearer {token['access']}"}
+
+        # send in request
+        response = requests.post(self.endpoint, files=files, headers=headers)
+
+        # check response
+
+        # check for object creation
+        self.assertNotEqual(company_count, self.company_model.objects.count())
+        self.assertNotEqual(user_count, self.user_model.objects.count())
+
+    def test_auth_user_cannot_load_csv(self):
+        company_count = self.company_model.objects.count()
+        user_count = self.user_model.objects.count()
+
+        # read csv file
+        csv_file = '../coop.csv'
+        files = {'csv_file': open(csv_file,'r')}
+
+        # Authenticate
+        token = get_tokens_for_user(self.user)
+        headers = {'Authorization': f"Bearer {token['access']}"}
+
+        # send in request
+        response = requests.post(self.endpoint, files=files, headers=headers)
+
+        # check response
+
+        # check for object creation
+        self.assertEqual(company_count, self.company_model.objects.count())
+        self.assertEqual(user_count, self.user_model.objects.count())
+
+    def test_anon_user_cannot_load_csv(self):
+        company_count = self.company_model.objects.count()
+        user_count = self.user_model.objects.count()
+
+        # read csv file
+        csv_file = '../coop.csv'
+        files = {'csv_file': open(csv_file,'r')}
+
+        # Query endpoint
+        response = self.client.post(self.endpoint, files=files)
+
+        # check response
+
+        # check for object creation
+        self.assertEqual(company_count, self.company_model.objects.count())
+        self.assertEqual(user_count, self.user_model.objects.count())
+