Add e2e tests for billing journals

Author: Robert Segal

e2e_config.test.json

@@ -11,6 +11,7 @@
   "accounts.seller.id": "SEL-7310-3075",
   "accounts.user.id": "USR-9673-3314",
   "accounts.user_group.id": "UGR-6822-0561",
+  "billing.journal.id": "BJO-6562-0928",
   "catalog.authorization.id": "AUT-9288-6146",
   "catalog.listing.id": "LST-5489-0806",
   "catalog.price_list.id": "PRC-7255-3950-0245",

mpt_api_client/resources/billing/journal_upload.py

@@ -1,10 +1,5 @@
 from mpt_api_client.http import AsyncService, Service
-from mpt_api_client.http.mixins import (
-    AsyncCollectionMixin,
-    AsyncFilesOperationsMixin,
-    CollectionMixin,
-    FilesOperationsMixin,
-)
+from mpt_api_client.http.types import FileTypes
 from mpt_api_client.models import Model
 
 
@@ -18,21 +13,73 @@ class JournalUploadServiceConfig:
     _endpoint = "/public/v1/billing/journals/{journal_id}/upload"
     _model_class = JournalUpload
     _collection_key = "data"
+    _upload_file_key = "file"
+    _upload_data_key = "id"
 
 
 class JournalUploadService(
-    FilesOperationsMixin[JournalUpload],
-    CollectionMixin[JournalUpload],
     Service[JournalUpload],
     JournalUploadServiceConfig,
 ):
     """Journal Upload service."""
 
+    def upload(self, resource_data: str, file: FileTypes | None = None) -> Model:  # noqa: WPS110
+        """Upload journal file.
+
+        Args:
+            resource_data: Resource data.
+            file: File image.
+
+        Returns:
+            Model: Created resource.
+        """
+        files = {}
+
+        if file:
+            files[self._upload_file_key] = file  # UNUSED type: ignore[attr-defined]
+            files[self._upload_data_key] = resource_data  # UNUSED type: ignore
+
+        response = self.http_client.request(  # UNUSED type: ignore[attr-defined]
+            "post",
+            self.path,  # UNUSED type: ignore[attr-defined]
+            files=files,
+            force_multipart=True,
+        )
+
+        return self._model_class.from_response(
+            response
+        )  # UNUSED type: ignore[attr-defined, no-any-return]
+
 
 class AsyncJournalUploadService(
-    AsyncFilesOperationsMixin[JournalUpload],
-    AsyncCollectionMixin[JournalUpload],
     AsyncService[JournalUpload],
     JournalUploadServiceConfig,
 ):
     """Journal Upload service."""
+
+    async def upload(self, resource_data: str, file: FileTypes | None = None) -> Model:  # noqa: WPS110
+        """Upload journal file.
+
+        Args:
+            resource_data: Resource data.
+            file: File image.
+
+        Returns:
+            Model: Created resource.
+        """
+        files = {}
+
+        if file:
+            files[self._upload_file_key] = file  # UNUSED type: ignore[attr-defined]
+            files[self._upload_data_key] = resource_data  # UNUSED type: ignore
+
+        response = await self.http_client.request(  # UNUSED type: ignore[attr-defined]
+            "post",
+            self.path,  # UNUSED type: ignore[attr-defined]
+            files=files,
+            force_multipart=True,
+        )
+
+        return self._model_class.from_response(
+            response
+        )  # UNUSED type: ignore[attr-defined, no-any-return]

tests/data/test_billing_journal.jsonl

@@ -0,0 +1 @@
+{"externalIds": {"vendor": "ext-seeded-billing-sub-vendor-id", "invoice": "INV12345", "reference": "ORD-7924-7691-0805"}, "search": {"source": {"type": "Subscription", "criteria": "id", "value": "SUB-5839-4140-9574"},"order":{"criteria":"order.id","value":"ORD-7924-7691-0805"}, "item": {"criteria": "item.id", "value": "ITM-1767-7355-0001"}}, "period": {"start": "2025-12-22", "end": "2026-12-21"}, "price": {"unitPP": 10, "PPx1": 8.33}, "quantity": 10, "segment": "COM", "description": {"value1": "desc-1", "value2": "desc-2"}}

tests/data/test_billing_journal.xlsx

@@ -0,0 +1,14 @@
+import pathlib
+
+import pytest
+
+
+@pytest.fixture
+def billing_journal_fd():
+    file_path = pathlib.Path("tests/data/test_billing_journal.xlsx").resolve()
+    return file_path.open("rb")
+
+
+@pytest.fixture
+def billing_journal_id(e2e_config):
+    return e2e_config["billing.journal.id"]

tests/e2e/billing/conftest.py

@@ -0,0 +1,21 @@
+import pytest
+
+
+@pytest.fixture
+def invalid_billing_journal_id():
+    return "BJO-0000-0000"
+
+
+@pytest.fixture
+def billing_journal_factory(authorization_id):
+    def factory(
+        name: str = "E2E Created Billing Journal",
+    ):
+        return {
+            "authorization": {"id": authorization_id},
+            "dueDate": "2026-01-02T19:00:00.000Z",
+            "externalIds": {},
+            "name": name,
+        }
+
+    return factory

tests/e2e/billing/journal/conftest.py

@@ -0,0 +1,114 @@
+import pytest
+
+from mpt_api_client.exceptions import MPTAPIError
+from mpt_api_client.rql.query_builder import RQLQuery
+
+pytestmark = [pytest.mark.flaky]
+
+
+@pytest.fixture
+async def created_billing_journal(async_mpt_vendor, billing_journal_factory):
+    new_billing_journal_request_data = billing_journal_factory(
+        name="E2E Created Billing Journal",
+    )
+
+    created_billing_journal = await async_mpt_vendor.billing.journals.create(
+        new_billing_journal_request_data
+    )
+
+    yield created_billing_journal
+
+    try:
+        await async_mpt_vendor.billing.journals.delete(created_billing_journal.id)
+    except MPTAPIError as error:
+        print(f"TEARDOWN - Unable to delete billing journal: {error.title}")  # noqa: WPS421
+
+
+@pytest.fixture
+async def submitted_billing_journal(async_mpt_vendor, created_billing_journal, billing_journal_fd):
+    billing_journal_file_data = created_billing_journal.id
+    await async_mpt_vendor.billing.journals.submit(created_billing_journal.id)
+    await async_mpt_vendor.billing.journals.upload(created_billing_journal.id).upload(
+        resource_data=billing_journal_file_data,
+        file=billing_journal_fd,
+    )
+
+    return created_billing_journal
+
+
+@pytest.fixture
+async def completed_billing_journal(async_mpt_vendor, submitted_billing_journal):
+    await async_mpt_vendor.billing.journals.accept(submitted_billing_journal.id)
+    await async_mpt_vendor.billing.journals.complete(submitted_billing_journal.id)
+    return submitted_billing_journal
+
+
+async def test_get_billing_journal_by_id(async_mpt_vendor, billing_journal_id):
+    result = await async_mpt_vendor.billing.journals.get(billing_journal_id)
+
+    assert result is not None
+
+
+async def test_list_billing_journals(async_mpt_vendor):
+    limit = 10
+
+    result = await async_mpt_vendor.billing.journals.fetch_page(limit=limit)
+
+    assert len(result) > 0
+
+
+async def test_get_billing_journal_by_id_not_found(async_mpt_vendor, invalid_billing_journal_id):
+    with pytest.raises(MPTAPIError, match=r"404 Not Found"):
+        await async_mpt_vendor.billing.journals.get(invalid_billing_journal_id)
+
+
+async def test_filter_billing_journals(async_mpt_vendor, billing_journal_id):
+    select_fields = ["-value"]
+    filtered_billing_journals = (
+        async_mpt_vendor.billing.journals.filter(RQLQuery(id=billing_journal_id))
+        .filter(RQLQuery(name="E2E Seeded Billing Journal"))
+        .select(*select_fields)
+    )
+
+    result = [billing_journal async for billing_journal in filtered_billing_journals.iterate()]
+
+    assert len(result) == 1
+
+
+def test_create_billing_journal(created_billing_journal):
+    result = created_billing_journal
+
+    assert result is not None
+
+
+async def test_update_billing_journal(
+    async_mpt_vendor, created_billing_journal, billing_journal_factory
+):
+    updated_name = "E2E Updated Billing Journal Name"
+    updated_billing_journal_data = billing_journal_factory(name=updated_name)
+
+    result = await async_mpt_vendor.billing.journals.update(
+        created_billing_journal.id,
+        updated_billing_journal_data,
+    )
+
+    assert result.name == updated_name
+
+
+async def test_delete_billing_journal(async_mpt_vendor, created_billing_journal):
+    result = created_billing_journal
+
+    await async_mpt_vendor.billing.journals.delete(result.id)
+
+
+async def test_upload_billing_journal(
+    async_mpt_vendor, created_billing_journal, billing_journal_fd
+):
+    billing_journal_file_data = created_billing_journal.id
+
+    result = await async_mpt_vendor.billing.journals.upload(created_billing_journal.id).upload(
+        resource_data=billing_journal_file_data,
+        file=billing_journal_fd,
+    )
+
+    assert result is not None

tests/e2e/billing/journal/test_async_journal.py

@@ -0,0 +1,108 @@
+import pytest
+
+from mpt_api_client.exceptions import MPTAPIError
+from mpt_api_client.rql.query_builder import RQLQuery
+
+pytestmark = [pytest.mark.flaky]
+
+
+@pytest.fixture
+def created_billing_journal(mpt_vendor, billing_journal_factory):
+    new_billing_journal_request_data = billing_journal_factory(
+        name="E2E Created Billing Journal",
+    )
+
+    created_billing_journal = mpt_vendor.billing.journals.create(new_billing_journal_request_data)
+
+    yield created_billing_journal
+
+    try:
+        mpt_vendor.billing.journals.delete(created_billing_journal.id)
+    except MPTAPIError as error:
+        print(f"TEARDOWN - Unable to delete billing journal: {error.title}")  # noqa: WPS421
+
+
+@pytest.fixture
+def submitted_billing_journal(mpt_vendor, created_billing_journal, billing_journal_fd):
+    billing_journal_file_data = created_billing_journal.id
+    mpt_vendor.billing.journals.submit(created_billing_journal.id)
+    mpt_vendor.billing.journals.upload(created_billing_journal.id).upload(
+        resource_data=billing_journal_file_data,
+        file=billing_journal_fd,
+    )
+
+    return created_billing_journal
+
+
+@pytest.fixture
+def completed_billing_journal(mpt_vendor, submitted_billing_journal):
+    mpt_vendor.billing.journals.accept(submitted_billing_journal.id)
+    mpt_vendor.billing.journals.complete(submitted_billing_journal.id)
+    return submitted_billing_journal
+
+
+def test_get_billing_journal_by_id(mpt_vendor, billing_journal_id):
+    result = mpt_vendor.billing.journals.get(billing_journal_id)
+
+    assert result is not None
+
+
+def test_list_billing_journals(mpt_vendor):
+    limit = 10
+
+    result = mpt_vendor.billing.journals.fetch_page(limit=limit)
+
+    assert len(result) > 0
+
+
+def test_get_billing_journal_by_id_not_found(mpt_vendor, invalid_billing_journal_id):
+    with pytest.raises(MPTAPIError, match=r"404 Not Found"):
+        mpt_vendor.billing.journals.get(invalid_billing_journal_id)
+
+
+def test_filter_billing_journals(mpt_vendor, billing_journal_id):
+    select_fields = ["-value"]
+    filtered_billing_journals = (
+        mpt_vendor.billing.journals.filter(RQLQuery(id=billing_journal_id))
+        .filter(RQLQuery(name="E2E Seeded Billing Journal"))
+        .select(*select_fields)
+    )
+
+    result = list(filtered_billing_journals.iterate())
+
+    assert len(result) == 1
+
+
+def test_create_billing_journal(created_billing_journal):
+    result = created_billing_journal
+
+    assert result is not None
+
+
+def test_update_billing_journal(mpt_vendor, created_billing_journal, billing_journal_factory):
+    updated_name = "E2E Updated Billing Journal Name"
+    updated_billing_journal_data = billing_journal_factory(name=updated_name)
+
+    result = mpt_vendor.billing.journals.update(
+        created_billing_journal.id,
+        updated_billing_journal_data,
+    )
+
+    assert result.name == updated_name
+
+
+def test_delete_billing_journal(mpt_vendor, created_billing_journal):
+    result = created_billing_journal
+
+    mpt_vendor.billing.journals.delete(result.id)
+
+
+def test_upload_billing_journal(mpt_vendor, created_billing_journal, billing_journal_fd):
+    billing_journal_file_data = created_billing_journal.id
+
+    result = mpt_vendor.billing.journals.upload(created_billing_journal.id).upload(
+        resource_data=billing_journal_file_data,
+        file=billing_journal_fd,
+    )
+
+    assert result is not None