pyapiary 2.3.1__tar.gz → 2.3.2__tar.gz

{pyapiary-2.3.1 → pyapiary-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyapiary
-Version: 2.3.1
+Version: 2.3.2
 Summary: A simple, lightweight set of connectors and functions to various APIs and DBMSs, controlled by a central broker.
 Author: Rob D'Aveta
 Author-email: rob.daveta@gmail.com

{pyapiary-2.3.1 → pyapiary-2.3.2}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "pyapiary"
 packages = [{ include = "pyapiary", from = "src" }]
-version = "2.3.1"
+version = "2.3.2"
 description = "A simple, lightweight set of connectors and functions to various APIs and DBMSs, controlled by a central broker."
 authors = ["Rob D'Aveta <rob.daveta@gmail.com>"]
 readme = "README.md"

{pyapiary-2.3.1 → pyapiary-2.3.2}/src/pyapiary/api_connectors/flashpoint.py

@@ -44,6 +44,16 @@ class FlashpointConnector(Broker):
         """
         return self.post("/sources/v2/fraud", json={"query": query, **kwargs})
 
+    @log_method_call
+    def search_fraud_checks(self, query: str, **kwargs) -> httpx.Response:
+        """Checks search provides the ability to search and read data from our Checks dataset.
+
+        Args:
+            query (str): The search string used in the API query.
+            **kwargs: Additional query logic per the Flashpoint API documentation.
+        """
+        return self.post("/sources/v2/fraud/checks", json={"query": query, **kwargs})
+
     @log_method_call
     def search_marketplaces(self, query: str, **kwargs) -> httpx.Response:
         """
@@ -140,6 +150,16 @@ class AsyncFlashpointConnector(AsyncBroker):
         """
         return await self.post("/sources/v2/fraud", json={"query": query, **kwargs})
 
+    @log_method_call
+    async def search_fraud_checks(self, query: str, **kwargs) -> httpx.Response:
+        """Checks search provides the ability to search and read data from our Checks dataset.
+
+        Args:
+            query (str): The search string used in the API query.
+            **kwargs: Additional query logic per the Flashpoint API documentation.
+        """
+        return await self.post("/sources/v2/fraud/checks", json={"query": query, **kwargs})
+
     @log_method_call
     async def search_marketplaces(self, query: str, **kwargs) -> httpx.Response:
         """

pyapiary-2.3.2/src/pyapiary/tests/test_flashpoint/test_unit_async_flashpoint.py

@@ -0,0 +1,100 @@
+import pytest
+import httpx
+from unittest.mock import patch, AsyncMock
+from pyapiary.api_connectors.flashpoint import AsyncFlashpointConnector
+
+
+@pytest.mark.asyncio
+async def test_async_init_with_api_key():
+    connector = AsyncFlashpointConnector(api_key="test_token")
+    assert connector.api_key == "test_token"
+    assert connector.headers["Authorization"] == "Bearer test_token"
+
+
+@patch.dict("os.environ", {"FLASHPOINT_API_KEY": "env_token"}, clear=True)
+@pytest.mark.asyncio
+async def test_async_init_with_env_key():
+    connector = AsyncFlashpointConnector(load_env_vars=True)
+    assert connector.api_key == "env_token"
+    assert connector.headers["Authorization"] == "Bearer env_token"
+
+
+@pytest.mark.asyncio
+async def test_async_init_missing_key():
+    with patch.dict("os.environ", {}, clear=True):
+        with pytest.raises(ValueError, match="FLASHPOINT_API_KEY is required"):
+            AsyncFlashpointConnector()
+
+
+@patch("pyapiary.api_connectors.flashpoint.AsyncFlashpointConnector.post", new_callable=AsyncMock)
+@pytest.mark.asyncio
+async def test_async_search_fraud_checks(mock_post):
+    import json
+
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": []}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = AsyncFlashpointConnector(api_key="mock_token")
+    result = await connector.search_fraud_checks("stolen checks")
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_awaited_once_with(
+        "/sources/v2/fraud/checks", json={"query": "stolen checks"}
+    )
+
+
+@patch("pyapiary.api_connectors.flashpoint.AsyncFlashpointConnector.post", new_callable=AsyncMock)
+@pytest.mark.asyncio
+async def test_async_search_fraud_checks_with_kwargs(mock_post):
+    import json
+
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": [{"id": "abc123"}]}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = AsyncFlashpointConnector(api_key="mock_token")
+    result = await connector.search_fraud_checks("routing number", size=10, from_=0)
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_awaited_once_with(
+        "/sources/v2/fraud/checks",
+        json={"query": "routing number", "size": 10, "from_": 0},
+    )
+
+
+@patch("pyapiary.api_connectors.flashpoint.AsyncFlashpointConnector.post", new_callable=AsyncMock)
+@pytest.mark.asyncio
+async def test_async_search_fraud(mock_post):
+    import json
+
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": []}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = AsyncFlashpointConnector(api_key="mock_token")
+    result = await connector.search_fraud("credential stuffing")
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_awaited_once()

pyapiary-2.3.2/src/pyapiary/tests/test_flashpoint/test_unit_flashpoint.py

@@ -0,0 +1,94 @@
+import httpx
+import pytest
+from unittest.mock import patch, MagicMock
+from pyapiary.api_connectors.flashpoint import FlashpointConnector
+
+def test_init_with_api_key():
+    connector = FlashpointConnector(api_key="test_token")
+    assert connector.api_key == "test_token"
+    assert connector.headers["Authorization"] == "Bearer test_token"
+
+
+@patch.dict("os.environ", {"FLASHPOINT_API_KEY": "env_token"}, clear=True)
+def test_init_with_env_key():
+    connector = FlashpointConnector(load_env_vars=True)
+    assert connector.api_key == "env_token"
+    assert connector.headers["Authorization"] == "Bearer env_token"
+
+
+def test_init_missing_key():
+    with patch.dict("os.environ", {}, clear=True):
+        with pytest.raises(ValueError, match="FLASHPOINT_API_KEY is required"):
+            FlashpointConnector()
+
+
+@patch("pyapiary.api_connectors.flashpoint.FlashpointConnector.post")
+def test_search_fraud_checks(mock_post):
+    import json
+
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": []}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = FlashpointConnector(api_key="mock_token")
+    result = connector.search_fraud_checks("stolen checks")
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_called_once_with(
+        "/sources/v2/fraud/checks", json={"query": "stolen checks"}
+    )
+
+
+@patch("pyapiary.api_connectors.flashpoint.FlashpointConnector.post")
+def test_search_fraud_checks_with_kwargs(mock_post):
+    import json
+
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": [{"id": "abc123"}]}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = FlashpointConnector(api_key="mock_token")
+    result = connector.search_fraud_checks("routing number", size=10, from_=0)
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_called_once_with(
+        "/sources/v2/fraud/checks",
+        json={"query": "routing number", "size": 10, "from_": 0},
+    )
+
+
+@patch("pyapiary.api_connectors.flashpoint.FlashpointConnector.post")
+def test_search_fraud(mock_post):
+    import json
+
+    # Use a real httpx.Response so our test matches the new return type
+    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
+    payload = {"success": True, "data": []}
+    mock_response = httpx.Response(
+        200,
+        request=request,
+        content=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+    )
+    mock_post.return_value = mock_response
+
+    connector = FlashpointConnector(api_key="mock_token")
+    result = connector.search_fraud("credential stuffing")
+
+    assert isinstance(result, httpx.Response)
+    assert result.json() == payload
+    mock_post.assert_called_once()

pyapiary-2.3.1/src/pyapiary/tests/test_flashpoint/test_unit_async_flashpoint.py

@@ -1,49 +0,0 @@
-import pytest
-import httpx
-from unittest.mock import patch, AsyncMock
-from pyapiary.api_connectors.flashpoint import AsyncFlashpointConnector
-
-
-@pytest.mark.asyncio
-async def test_async_init_with_api_key():
-    connector = AsyncFlashpointConnector(api_key="test_token")
-    assert connector.api_key == "test_token"
-    assert connector.headers["Authorization"] == "Bearer test_token"
-
-
-@patch.dict("os.environ", {"FLASHPOINT_API_KEY": "env_token"}, clear=True)
-@pytest.mark.asyncio
-async def test_async_init_with_env_key():
-    connector = AsyncFlashpointConnector(load_env_vars=True)
-    assert connector.api_key == "env_token"
-    assert connector.headers["Authorization"] == "Bearer env_token"
-
-
-@pytest.mark.asyncio
-async def test_async_init_missing_key():
-    with patch.dict("os.environ", {}, clear=True):
-        with pytest.raises(ValueError, match="FLASHPOINT_API_KEY is required"):
-            AsyncFlashpointConnector()
-
-
-@patch("pyapiary.api_connectors.flashpoint.AsyncFlashpointConnector.post", new_callable=AsyncMock)
-@pytest.mark.asyncio
-async def test_async_search_fraud(mock_post):
-    import json
-
-    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
-    payload = {"success": True, "data": []}
-    mock_response = httpx.Response(
-        200,
-        request=request,
-        content=json.dumps(payload).encode("utf-8"),
-        headers={"Content-Type": "application/json"},
-    )
-    mock_post.return_value = mock_response
-
-    connector = AsyncFlashpointConnector(api_key="mock_token")
-    result = await connector.search_fraud("credential stuffing")
-
-    assert isinstance(result, httpx.Response)
-    assert result.json() == payload
-    mock_post.assert_awaited_once()

pyapiary-2.3.1/src/pyapiary/tests/test_flashpoint/test_unit_flashpoint.py

@@ -1,45 +0,0 @@
-import httpx
-import pytest
-from unittest.mock import patch, MagicMock
-from pyapiary.api_connectors.flashpoint import FlashpointConnector
-
-def test_init_with_api_key():
-    connector = FlashpointConnector(api_key="test_token")
-    assert connector.api_key == "test_token"
-    assert connector.headers["Authorization"] == "Bearer test_token"
-
-
-@patch.dict("os.environ", {"FLASHPOINT_API_KEY": "env_token"}, clear=True)
-def test_init_with_env_key():
-    connector = FlashpointConnector(load_env_vars=True)
-    assert connector.api_key == "env_token"
-    assert connector.headers["Authorization"] == "Bearer env_token"
-
-
-def test_init_missing_key():
-    with patch.dict("os.environ", {}, clear=True):
-        with pytest.raises(ValueError, match="FLASHPOINT_API_KEY is required"):
-            FlashpointConnector()
-
-
-@patch("pyapiary.api_connectors.flashpoint.FlashpointConnector.post")
-def test_search_fraud(mock_post):
-    import json
-
-    # Use a real httpx.Response so our test matches the new return type
-    request = httpx.Request("POST", "https://api.flashpoint.io/mock")
-    payload = {"success": True, "data": []}
-    mock_response = httpx.Response(
-        200,
-        request=request,
-        content=json.dumps(payload).encode("utf-8"),
-        headers={"Content-Type": "application/json"},
-    )
-    mock_post.return_value = mock_response
-
-    connector = FlashpointConnector(api_key="mock_token")
-    result = connector.search_fraud("credential stuffing")
-
-    assert isinstance(result, httpx.Response)
-    assert result.json() == payload
-    mock_post.assert_called_once()