Added missing billing statement attachments endpoints

Author: Robert Segal

mpt_api_client/resources/billing/statement_attachments.py

@@ -0,0 +1,56 @@
+from mpt_api_client.http import AsyncService, Service
+from mpt_api_client.http.mixins import (
+    AsyncCollectionMixin,
+    AsyncCreateFileMixin,
+    AsyncDeleteMixin,
+    AsyncDownloadFileMixin,
+    AsyncGetMixin,
+    AsyncUpdateMixin,
+    CollectionMixin,
+    CreateFileMixin,
+    DeleteMixin,
+    DownloadFileMixin,
+    GetMixin,
+    UpdateMixin,
+)
+from mpt_api_client.models import Model
+
+
+class StatementAttachment(Model):
+    """Statement Attachment resource."""
+
+
+class StatementAttachmentsServiceConfig:
+    """Statement Attachments service configuration."""
+
+    _endpoint = "/public/v1/billing/statements/{statement_id}/attachments"
+    _model_class = StatementAttachment
+    _collection_key = "data"
+    _upload_file_key = "file"
+    _upload_data_key = "attachment"
+
+
+class StatementAttachmentsService(
+    CreateFileMixin[StatementAttachment],
+    UpdateMixin[StatementAttachment],
+    DownloadFileMixin[StatementAttachment],
+    DeleteMixin,
+    GetMixin[StatementAttachment],
+    CollectionMixin[StatementAttachment],
+    Service[StatementAttachment],
+    StatementAttachmentsServiceConfig,
+):
+    """Statement Attachments service."""
+
+
+class AsyncStatementAttachmentsService(
+    AsyncCreateFileMixin[StatementAttachment],
+    AsyncUpdateMixin[StatementAttachment],
+    AsyncDownloadFileMixin[StatementAttachment],
+    AsyncDeleteMixin,
+    AsyncGetMixin[StatementAttachment],
+    AsyncCollectionMixin[StatementAttachment],
+    AsyncService[StatementAttachment],
+    StatementAttachmentsServiceConfig,
+):
+    """Statement Attachments service."""

mpt_api_client/resources/billing/statements.py

@@ -9,6 +9,10 @@
 )
 from mpt_api_client.models import Model
 from mpt_api_client.resources.billing.mixins import AsyncIssuableMixin, IssuableMixin
+from mpt_api_client.resources.billing.statement_attachments import (
+    AsyncStatementAttachmentsService,
+    StatementAttachmentsService,
+)
 from mpt_api_client.resources.billing.statement_charges import (
     AsyncStatementChargesService,
     StatementChargesService,
@@ -44,6 +48,13 @@ def charges(self, statement_id: str) -> StatementChargesService:
             endpoint_params={"statement_id": statement_id},
         )
 
+    def attachments(self, statement_id: str) -> StatementAttachmentsService:
+        """Return statement attachments service."""
+        return StatementAttachmentsService(
+            http_client=self.http_client,
+            endpoint_params={"statement_id": statement_id},
+        )
+
 
 class AsyncStatementsService(
     AsyncUpdateMixin[Statement],
@@ -61,3 +72,10 @@ def charges(self, statement_id: str) -> AsyncStatementChargesService:
             http_client=self.http_client,
             endpoint_params={"statement_id": statement_id},
         )
+
+    def attachments(self, statement_id: str) -> AsyncStatementAttachmentsService:
+        """Return statement attachments service."""
+        return AsyncStatementAttachmentsService(
+            http_client=self.http_client,
+            endpoint_params={"statement_id": statement_id},
+        )

tests/unit/resources/billing/test_statement_attachments.py

@@ -0,0 +1,52 @@
+import pytest
+
+from mpt_api_client.resources.billing.statement_attachments import (
+    AsyncStatementAttachmentsService,
+    StatementAttachmentsService,
+)
+
+
+@pytest.fixture
+def statement_attachments_service(http_client) -> StatementAttachmentsService:
+    return StatementAttachmentsService(
+        http_client=http_client, endpoint_params={"statement_id": "STM-0000-0001"}
+    )
+
+
+@pytest.fixture
+def async_statement_attachments_service(async_http_client) -> AsyncStatementAttachmentsService:
+    return AsyncStatementAttachmentsService(
+        http_client=async_http_client, endpoint_params={"statement_id": "STM-0000-0001"}
+    )
+
+
+def test_endpoint(statement_attachments_service) -> None:
+    result = (
+        statement_attachments_service.path
+        == "/public/v1/billing/statements/STM-0000-0001/attachments"
+    )
+
+    assert result is True
+
+
+def test_async_endpoint(async_statement_attachments_service) -> None:
+    result = (
+        async_statement_attachments_service.path
+        == "/public/v1/billing/statements/STM-0000-0001/attachments"
+    )
+
+    assert result is True
+
+
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
+def test_methods_present(statement_attachments_service, method: str) -> None:
+    result = hasattr(statement_attachments_service, method)
+
+    assert result is True
+
+
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
+def test_async_methods_present(async_statement_attachments_service, method: str) -> None:
+    result = hasattr(async_statement_attachments_service, method)
+
+    assert result is True

tests/unit/resources/billing/test_statements.py

@@ -1,5 +1,9 @@
 import pytest
 
+from mpt_api_client.resources.billing.statement_attachments import (
+    AsyncStatementAttachmentsService,
+    StatementAttachmentsService,
+)
 from mpt_api_client.resources.billing.statement_charges import (
     AsyncStatementChargesService,
     StatementChargesService,
@@ -19,7 +23,19 @@ def async_statements_service(async_http_client):
 
 @pytest.mark.parametrize(
     "method",
-    ["get", "update", "issue", "cancel", "error", "pending", "queue", "retry", "recalculate"],
+    [
+        "get",
+        "update",
+        "issue",
+        "cancel",
+        "error",
+        "pending",
+        "queue",
+        "retry",
+        "recalculate",
+        "attachments",
+        "charges",
+    ],
 )
 def test_mixins_present(statements_service, method):
     result = hasattr(statements_service, method)
@@ -29,7 +45,19 @@ def test_mixins_present(statements_service, method):
 
 @pytest.mark.parametrize(
     "method",
-    ["get", "update", "issue", "cancel", "error", "pending", "queue", "retry", "recalculate"],
+    [
+        "get",
+        "update",
+        "issue",
+        "cancel",
+        "error",
+        "pending",
+        "queue",
+        "retry",
+        "recalculate",
+        "attachments",
+        "charges",
+    ],
 )
 def test_async_mixins_present(async_statements_service, method):
     result = hasattr(async_statements_service, method)
@@ -41,6 +69,7 @@ def test_async_mixins_present(async_statements_service, method):
     ("service_method", "expected_service_class"),
     [
         ("charges", StatementChargesService),
+        ("attachments", StatementAttachmentsService),
     ],
 )
 def test_property_services(statements_service, service_method, expected_service_class):
@@ -54,6 +83,7 @@ def test_property_services(statements_service, service_method, expected_service_
     ("service_method", "expected_service_class"),
     [
         ("charges", AsyncStatementChargesService),
+        ("attachments", AsyncStatementAttachmentsService),
     ],
 )
 def test_async_property_services(async_statements_service, service_method, expected_service_class):