Added billing custom ledgers endpoints

Author: Robert Segal

mpt_api_client/resources/billing/billing.py

@@ -1,4 +1,8 @@
 from mpt_api_client.http import AsyncHTTPClient, HTTPClient
+from mpt_api_client.resources.billing.custom_ledgers import (
+    AsyncCustomLedgersService,
+    CustomLedgersService,
+)
 from mpt_api_client.resources.billing.journals import AsyncJournalsService, JournalsService
 from mpt_api_client.resources.billing.ledgers import AsyncLedgersService, LedgersService
 
@@ -19,6 +23,11 @@ def ledgers(self) -> LedgersService:
         """Ledgers service."""
         return LedgersService(http_client=self.http_client)
 
+    @property
+    def custom_ledgers(self) -> CustomLedgersService:
+        """Custom ledgers service."""
+        return CustomLedgersService(http_client=self.http_client)
+
 
 class AsyncBilling:
     """Billing MPT API Module."""
@@ -35,3 +44,8 @@ def journals(self) -> AsyncJournalsService:
     def ledgers(self) -> AsyncLedgersService:
         """Ledgers service."""
         return AsyncLedgersService(http_client=self.http_client)
+
+    @property
+    def custom_ledgers(self) -> AsyncCustomLedgersService:
+        """Custom ledgers service."""
+        return AsyncCustomLedgersService(http_client=self.http_client)

mpt_api_client/resources/billing/custom_ledgers.py

@@ -0,0 +1,45 @@
+from mpt_api_client.http import AsyncService, Service
+from mpt_api_client.http.mixins import (
+    AsyncCreateMixin,
+    AsyncDeleteMixin,
+    AsyncUpdateMixin,
+    CreateMixin,
+    DeleteMixin,
+    UpdateMixin,
+)
+from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AcceptableMixin, AsyncAcceptableMixin
+
+
+class CustomLedger(Model):
+    """Custom Ledger resource."""
+
+
+class CustomLedgersServiceConfig:
+    """Custom Ledgers service configuration."""
+
+    _endpoint = "/public/v1/billing/custom-ledgers"
+    _model_class = CustomLedger
+    _collection_key = "data"
+
+
+class CustomLedgersService(
+    CreateMixin[CustomLedger],
+    DeleteMixin,
+    UpdateMixin[CustomLedger],
+    AcceptableMixin[CustomLedger],
+    Service[CustomLedger],
+    CustomLedgersServiceConfig,
+):
+    """Custom Ledgers service."""
+
+
+class AsyncCustomLedgersService(
+    AsyncCreateMixin[CustomLedger],
+    AsyncDeleteMixin,
+    AsyncUpdateMixin[CustomLedger],
+    AsyncAcceptableMixin[CustomLedger],
+    AsyncService[CustomLedger],
+    CustomLedgersServiceConfig,
+):
+    """Async Custom Ledgers service."""

mpt_api_client/resources/billing/mixins.py

@@ -174,3 +174,55 @@ async def queue(self, resource_id: str, resource_data: ResourceData | None = Non
         return await self._resource_action(  # type: ignore[attr-defined, no-any-return]
             resource_id, "POST", "queue", json=resource_data
         )
+
+
+class AcceptableMixin[Model]:
+    """Acceptable mixin adds the ability to accept resources."""
+
+    def accept(self, resource_id: str, resource_data: ResourceData | None = None) -> Model:
+        """Accept resource.
+
+        Args:
+            resource_id: Resource ID
+            resource_data: Resource data will be updated
+        """
+        return self._resource_action(  # type: ignore[attr-defined, no-any-return]
+            resource_id, "POST", "accept", json=resource_data
+        )
+
+    def queue(self, resource_id: str, resource_data: ResourceData | None = None) -> Model:
+        """Queue resource.
+
+        Args:
+            resource_id: Resource ID
+            resource_data: Resource data will be updated
+        """
+        return self._resource_action(  # type: ignore[attr-defined, no-any-return]
+            resource_id, "POST", "queue", json=resource_data
+        )
+
+
+class AsyncAcceptableMixin[Model]:
+    """Acceptable mixin adds the ability to accept resources."""
+
+    async def accept(self, resource_id: str, resource_data: ResourceData | None = None) -> Model:
+        """Accept resource.
+
+        Args:
+            resource_id: Resource ID
+            resource_data: Resource data will be updated
+        """
+        return await self._resource_action(  # type: ignore[attr-defined, no-any-return]
+            resource_id, "POST", "accept", json=resource_data
+        )
+
+    async def queue(self, resource_id: str, resource_data: ResourceData | None = None) -> Model:
+        """Queue resource.
+
+        Args:
+            resource_id: Resource ID
+            resource_data: Resource data will be updated
+        """
+        return await self._resource_action(  # type: ignore[attr-defined, no-any-return]
+            resource_id, "POST", "queue", json=resource_data
+        )

tests/resources/billing/test_billing.py

@@ -1,6 +1,10 @@
 import pytest
 
 from mpt_api_client.resources.billing.billing import AsyncBilling, Billing
+from mpt_api_client.resources.billing.custom_ledgers import (
+    AsyncCustomLedgersService,
+    CustomLedgersService,
+)
 from mpt_api_client.resources.billing.journals import AsyncJournalsService, JournalsService
 from mpt_api_client.resources.billing.ledgers import AsyncLedgersService, LedgersService
 
@@ -20,6 +24,7 @@ def async_billing(async_http_client):
     [
         ("journals", JournalsService),
         ("ledgers", LedgersService),
+        ("custom_ledgers", CustomLedgersService),
     ],
 )
 def test_billing_properties(billing, property_name, expected_service_class):
@@ -35,6 +40,7 @@ def test_billing_properties(billing, property_name, expected_service_class):
     [
         ("journals", AsyncJournalsService),
         ("ledgers", AsyncLedgersService),
+        ("custom_ledgers", AsyncCustomLedgersService),
     ],
 )
 def test_async_billing_properties(async_billing, property_name, expected_service_class):

tests/resources/billing/test_custom_ledgers.py

@@ -0,0 +1,26 @@
+import pytest
+
+from mpt_api_client.resources.billing.custom_ledgers import (
+    AsyncCustomLedgersService,
+    CustomLedgersService,
+)
+
+
+@pytest.fixture
+def custom_ledgers_service(http_client):
+    return CustomLedgersService(http_client=http_client)
+
+
+@pytest.fixture
+def async_custom_ledgers_service(http_client):
+    return AsyncCustomLedgersService(http_client=http_client)
+
+
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "accept", "queue"])
+def test_mixins_present(custom_ledgers_service, method):
+    assert hasattr(custom_ledgers_service, method)
+
+
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "accept", "queue"])
+def test_async_mixins_present(async_custom_ledgers_service, method):
+    assert hasattr(async_custom_ledgers_service, method)

tests/resources/billing/test_journals.py

@@ -19,15 +19,15 @@ def async_journals_service(async_http_client):
 
 @pytest.mark.parametrize(
     "method",
-    ["get", "create", "update", "delete"],
+    ["get", "create", "update", "delete", "regenerate", "submit", "enquiry", "accept"],
 )
 def test_mixins_present(journals_service, method):
     assert hasattr(journals_service, method)
 
 
 @pytest.mark.parametrize(
     "method",
-    ["get", "create", "update", "delete"],
+    ["get", "create", "update", "delete", "regenerate", "submit", "enquiry", "accept"],
 )
 def test_async_mixins_present(async_journals_service, method):
     assert hasattr(async_journals_service, method)

tests/resources/billing/test_mixins.py

@@ -5,6 +5,8 @@
 from mpt_api_client.http.async_service import AsyncService
 from mpt_api_client.http.service import Service
 from mpt_api_client.resources.billing.mixins import (
+    AcceptableMixin,
+    AsyncAcceptableMixin,
     AsyncRecalculatableMixin,
     AsyncRegeneratableMixin,
     RecalculatableMixin,
@@ -49,6 +51,24 @@ class DummyAsyncRecalculatableService(
     _collection_key = "data"
 
 
+class DummyAcceptableService(
+    AcceptableMixin[DummyModel],
+    Service[DummyModel],
+):
+    _endpoint = "/public/v1/dummy/acceptable/"
+    _model_class = DummyModel
+    _collection_key = "data"
+
+
+class DummyAsyncAcceptableService(
+    AsyncAcceptableMixin[DummyModel],
+    AsyncService[DummyModel],
+):
+    _endpoint = "/public/v1/dummy/acceptable/"
+    _model_class = DummyModel
+    _collection_key = "data"
+
+
 @pytest.fixture
 def regeneratable_service(http_client):
     return DummyRegeneratableService(http_client=http_client)
@@ -69,6 +89,16 @@ def async_recalculatable_service(async_http_client):
     return DummyAsyncRecalculatableService(http_client=async_http_client)
 
 
+@pytest.fixture
+def acceptable_service(http_client):
+    return DummyAcceptableService(http_client=http_client)
+
+
+@pytest.fixture
+def async_acceptable_service(async_http_client):
+    return DummyAsyncAcceptableService(http_client=async_http_client)
+
+
 @pytest.mark.parametrize(
     ("action", "input_status"),
     [
@@ -321,3 +351,119 @@ async def test_async_recalculate_resource_actions_no_data(
         assert request.content == request_expected_content
         assert recalc_obj.to_dict() == response_expected_data
         assert isinstance(recalc_obj, DummyModel)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("accept", {"id": "OBJ-0000-0001", "status": "update"}),
+        ("queue", {"id": "OBJ-0000-0001", "status": "update"}),
+    ],
+)
+def test_acceptable_resource_actions(acceptable_service, action, input_status):
+    request_expected_content = b'{"id":"OBJ-0000-0001","status":"update"}'
+    response_expected_data = {"id": "OBJ-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/dummy/acceptable/OBJ-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        accept_obj = getattr(acceptable_service, action)("OBJ-0000-0001", input_status)
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert accept_obj.to_dict() == response_expected_data
+        assert isinstance(accept_obj, DummyModel)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [("accept", None), ("queue", None)],
+)
+def test_acceptable_resource_actions_no_data(acceptable_service, action, input_status):
+    request_expected_content = b""
+    response_expected_data = {"id": "OBJ-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/dummy/acceptable/OBJ-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        accept_obj = getattr(acceptable_service, action)("OBJ-0000-0001", input_status)
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert accept_obj.to_dict() == response_expected_data
+        assert isinstance(accept_obj, DummyModel)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("accept", {"id": "OBJ-0000-0001", "status": "update"}),
+        ("queue", {"id": "OBJ-0000-0001", "status": "update"}),
+    ],
+)
+async def test_async_acceptable_resource_actions(async_acceptable_service, action, input_status):
+    request_expected_content = b'{"id":"OBJ-0000-0001","status":"update"}'
+    response_expected_data = {"id": "OBJ-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/dummy/acceptable/OBJ-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        accept_obj = await getattr(async_acceptable_service, action)("OBJ-0000-0001", input_status)
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert accept_obj.to_dict() == response_expected_data
+        assert isinstance(accept_obj, DummyModel)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [("accept", None), ("queue", None)],
+)
+async def test_async_acceptable_resource_actions_no_data(
+    async_acceptable_service, action, input_status
+):
+    request_expected_content = b""
+    response_expected_data = {"id": "OBJ-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/dummy/acceptable/OBJ-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        accept_obj = await getattr(async_acceptable_service, action)("OBJ-0000-0001", input_status)
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert accept_obj.to_dict() == response_expected_data
+        assert isinstance(accept_obj, DummyModel)