fix: remove unused export and extend mock interface

VorTECHsa · Jul 17, 2023 · 8298e31 · 8298e31
1 parent 0b60785
commit 8298e31
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tests/endpoints/test_cargo_movements_real.py b/tests/endpoints/test_cargo_movements_real.py
@@ -1,6 +1,4 @@
-import pytest
 from datetime import datetime
-import os
 
 from tests.testcases import TestCaseUsingRealAPI
 from tests.timer import Timer

diff --git a/tests/mock_client.py b/tests/mock_client.py
@@ -13,6 +13,7 @@
     VESSELS_REFERENCE,
 )
 from vortexasdk.search_response import SearchResponse
+from vortexasdk.utils import PAGINATION_STRATEGIES
 
 
 def _read(example_file) -> List[Dict]:
@@ -49,3 +50,12 @@ def search(
         self, resource: str, response_type=None, **data
     ) -> SearchResponse:
         return {"data": MockVortexaClient._results[resource], "reference": {}}
+
+    def search_base(
+        self,
+        resource: str,
+        response_type=None,
+        pagination_strategy: PAGINATION_STRATEGIES = None,
+        **data,
+    ) -> SearchResponse:
+        return {"data": MockVortexaClient._results[resource], "reference": {}}