chore: add kwargs parameter to client methods

eduNEXT · Jul 29, 2024 · d5c6534 · d5c6534
1 parent d9114ea
commit d5c6534
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/eox_nelp/api_clients/pearson_engine.py b/eox_nelp/api_clients/pearson_engine.py
@@ -82,7 +82,7 @@ def _get_exam_data(self, exam_id):
             "external_key": exam_id,
         }
 
-    def import_candidate_demographics(self, user):
+    def import_candidate_demographics(self, user, **kwargs):
         """
         Import candidate demographics into Pearson Engine.
 
@@ -97,11 +97,12 @@ def import_candidate_demographics(self, user):
         candidate = {
             "user_data": self._get_user_data(user),
             "platform_data": self._get_platform_data(),
+            **kwargs
         }
 
         return self.make_post(path, candidate)
 
-    def import_exam_authorization(self, user, exam_id, transaction_type="Add"):
+    def import_exam_authorization(self, user, exam_id, **kwargs):
         """
         Import exam authorization data into Pearson Engine.
 
@@ -116,12 +117,12 @@ def import_exam_authorization(self, user, exam_id, transaction_type="Add"):
         exam_data = {
             "user_data": {"username": user.username},
             "exam_data": self._get_exam_data(exam_id),
-            "transaction_type": transaction_type,
+            **kwargs
         }
 
         return self.make_post(path, exam_data)
 
-    def real_time_import(self, user, exam_id):
+    def real_time_import(self, user, exam_id, **kwargs):
         """
         Perform a real-time import of exam authorization data.
 
@@ -137,6 +138,7 @@ def real_time_import(self, user, exam_id):
             "user_data": self._get_user_data(user),
             "exam_data": self._get_exam_data(exam_id),
             "platform_data": self._get_platform_data(),
+            **kwargs
         }
 
         return self.make_post(path, data)
diff --git a/eox_nelp/api_clients/tests/test_pearson_engine.py b/eox_nelp/api_clients/tests/test_pearson_engine.py
@@ -82,7 +82,7 @@ def test_import_exam_authorization(self, auth_mock, post_mock):
         exam_id = "exam-123"
         api_client = self.api_class()
 
-        response = api_client.import_exam_authorization(user, exam_id)
+        response = api_client.import_exam_authorization(user, exam_id, transaction_type="Add")
 
         self.assertDictEqual(response, expected_value)
         post_mock.assert_called_with("rti/api/v1/exam-authorization/", {