wip

tests/test_watchlist.py

@@ -0,0 +1,69 @@
+from datetime import datetime
+
+import pytest
+
+from onfido import ApplicantBuilder, ReportName, WatchlistMonitor
+from tests.conftest import create_applicant
+
+
+@pytest.fixture(scope="function")
+def applicant_id(onfido_api):
+    applicant_builder = ApplicantBuilder(
+        first_name="John",
+        last_name="Smith",
+        dob=datetime(year=1990, month=1, day=1),
+    )
+    return create_applicant(onfido_api, applicant_builder=applicant_builder).id
+
+
+@pytest.fixture(scope="function")
+def watchlist_monitor(onfido_api, applicant_id):
+    return onfido_api.create_watchlist_monitor(
+        WatchlistMonitor(
+            applicant_id=applicant_id, report_name=ReportName.WATCHLIST_STANDARD
+        )
+    )
+
+
+def test_create_watchlist_standard_monitor(onfido_api, applicant_id, watchlist_monitor):
+    assert watchlist_monitor.applicant_id == applicant_id
+    assert watchlist_monitor.actual_instance.name == ReportName.WATCHLIST_STANDARD
+
+
+def test_create_watchlist_aml_monitor(onfido_api, applicant_id):
+    watchlist_monitor = onfido_api.create_watchlist_monitor(
+        WatchlistMonitor(
+            applicant_id=applicant_id, report_name=ReportName.WATCHLIST_AML
+        )
+    )
+
+    assert watchlist_monitor.applicant_id == applicant_id
+    assert watchlist_monitor.actual_instance.name == ReportName.WATCHLIST_AML
+
+
+def test_list_watchlist_monitors(onfido_api, applicant_id, watchlist_monitor):
+    list_of_monitors = onfido_api.list_watchlist_monitors(
+        applicant_id, include_deleted=False
+    )
+
+    assert len(list_of_monitors) > 0
+
+
+def test_find_watchlist_monitor(onfido_api, watchlist_monitor):
+    get_watchlist_monitor = onfido_api.find_watchlist_monitor(watchlist_monitor.id)
+
+    assert get_watchlist_monitor.id == watchlist_monitor.id
+
+
+def test_delete_watchlist_monitor(onfido_api, watchlist_monitor):
+    onfido_api.delete_watchlist_monitor(watchlist_monitor.id)
+
+
+def test_list_watchlist_monitor_matches(onfido_api, watchlist_monitor):
+    matches_list = onfido_api.list_watchlist_monitor_matches(watchlist_monitor.id)
+
+    assert len(matches_list) == 0
+
+
+def test_force_report_creation(onfido_api, watchlist_monitor):
+    onfido_api.force_new_record_creation(watchlist_monitor.id)

tests/test_workflow_run_outputs.py

@@ -2,7 +2,8 @@
 
 from onfido import (
     ApplicantBuilder,
-    CompleteTaskRequest,
+    CompleteTaskBuilder,
+    CompleteTaskBuilderData,
     CountryCodes,
     DocumentBreakdown,
     DocumentProperties,
@@ -78,7 +79,7 @@ def test_profile_data_as_output(onfido_api, applicant_id, profile_data):
     tasks = onfido_api.list_tasks(workflow_run_id)
     profile_data_task_id = list(filter(lambda task: "profile" in task.id, tasks))[0].id
 
-    complete_task_request = CompleteTaskRequest(data=profile_data)
+    complete_task_request = CompleteTaskRequest(data=CompleteTaskBuilderData(profile_data))
 
     onfido_api.complete_task(
         workflow_run_id=workflow_run_id,
@@ -106,36 +107,35 @@ def test_document_and_facial_similarity_report_as_output(
     tasks = onfido_api.list_tasks(workflow_run_id)
     profile_data_task_id = list(filter(lambda task: "profile" in task.id, tasks))[0].id
 
-    complete_task_request = CompleteTaskRequest(
-        data={"first_name": "Jane", "last_name": "Doe"}
+    complete_task_builder = CompleteTaskBuilder(
+        data=CompleteTaskBuilderData({"first_name": "Jane", "last_name": "Doe"})
     )
     onfido_api.complete_task(
         workflow_run_id=workflow_run_id,
         task_id=profile_data_task_id,
-        complete_task_request=complete_task_request,
+        complete_task_request=complete_task_builder,
     )
 
     tasks = onfido_api.list_tasks(workflow_run_id)
     document_capture_task_id = list(
         filter(lambda task: "document_photo" in task.id, tasks)
     )[0].id
 
-    complete_document_capture_task_request = CompleteTaskRequest(
-        data=[{"id": document_id}]
+    complete_document_capture_task_builder = CompleteTaskBuilder(
+        data=CompleteTaskBuilderData([{"id": document_id}])
     )
     onfido_api.complete_task(
         workflow_run_id=workflow_run_id,
         task_id=document_capture_task_id,
-        complete_task_request=complete_document_capture_task_request,
+        complete_task_request=complete_document_capture_task_builder,
     )
 
     tasks = onfido_api.list_tasks(workflow_run_id)
     live_photo_capture_task_id = list(
         filter(lambda task: "face_photo" in task.id, tasks)
     )[0].id
-
-    complete_live_photo_capture_task_request = CompleteTaskRequest(
-        data=[{"id": live_photo_id}]
+    complete_live_photo_capture_task_request = CompleteTaskBuilder(
+        data=CompleteTaskBuilderData([{"id": live_photo_id}])
     )
     onfido_api.complete_task(
         workflow_run_id=workflow_run_id,