further fixed testing

ucb-rit · Nov 6, 2023 · d6b6682 · d6b6682
1 parent e6b7c35
commit d6b6682
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/coldfront/config/test_settings.py.sample b/coldfront/config/test_settings.py.sample
@@ -66,17 +66,20 @@ REQUEST_APPROVAL_CC_LIST = ['[email protected]']
 GOOGLE_DRIVE_PERMISSIONS = ()
 
 FILE_STORAGE = {
-    'backend': 'local_storage',
+    'backend': 'google_drive',
     'details': {
         'NEW_PROJECT_REQUEST_MOU': {
-            'location': 'New Project Request MOUs/'
+            'location': 'New Project Request MOUs/',
+            'filename_type': 'NewProject_MOU',
+        },
+        'SERVICE_UNITS_PURCHASE_REQUEST_MOU': {
+            'location': 'Service Units Purchase Request MOUs/',
+            'filename_type': 'AllowancePurchase_MOU',
         },
         'SECURE_DIRECTORY_REQUEST_MOU': {
-            'location': 'Secure Directory Request MOUs/'
+            'location': 'Secure Directory Request MOUs/',
+            'filename_type': 'SecureDirectory_RUA',
         },
-        'SERVICE_UNITS_PURCHASE_REQUEST_MOU': {
-            'location': 'Service Units Purchase Request MOUs/'
-        }
     },
 }
 

diff --git a/coldfront/core/project/tests/test_views/test_addition_views/test_request_view.py b/coldfront/core/project/tests/test_views/test_addition_views/test_request_view.py
@@ -181,10 +181,7 @@ def test_valid_post(self):
         extra_fields = request.extra_fields
         self.assertEqual(len(extra_fields), len(data))
         for field in data:
-            if field == 'num_service_units':
-                self.assertNotIn(field, extra_fields)
-            else:
-                self.assertEqual(data[field], extra_fields[field])
+            self.assertEqual(data[field], extra_fields[field])
 
         # A notification email should have been sent to admins.
         self.assertEqual(len(mail.outbox), 1)