Fix tests

CybercentreCanada · Jul 5, 2024 · 1c077cd · 1c077cd
1 parent 10ac697
commit 1c077cd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_submit.py b/test/test_submit.py
@@ -104,9 +104,9 @@ def test_submit_dynamic(datastore, client):
     assert res['sid'] is not None
     assert res == datastore.submission.get(res['sid'], as_obj=False)
     assert 'Dynamic Analysis' in res['params']['services']['selected']
-    for k in res['params']:
+    for k, v in submission_data['params'].items():
         if k not in ['submitter', 'services', 'description', 'quota_item']:
-            assert res['params'][k] == submission_data['params'][k]
+            assert res['params'].get(k) == v
 
 
 def test_resubmit(datastore, client):
@@ -117,6 +117,6 @@ def test_resubmit(datastore, client):
     assert res is not None
     assert res['sid'] is not None
     assert res == datastore.submission.get(res['sid'], as_obj=False)
-    for k in res['params']:
+    for k, v in submission_data['params'].items():
         if k not in ['submitter', 'description', 'quota_item']:
-            assert res['params'][k] == submission_data['params'][k]
+            assert res['params'].get(k) == v