MPT-16512 E2E for Notification batch attachments

Author: albertsola

e2e_config.test.json

@@ -46,6 +46,7 @@
   "commerce.subscription.agreement.id": "AGR-2473-3299-1721",
   "commerce.subscription.id": "SUB-3678-1831-2188",
   "commerce.subscription.product.item.id": "ITM-1767-7355-0001",
+  "notifications.batch.attachment.id": "ATT-7183-1965-7758",
   "notifications.batch.id": "MST-3638-2460-4825",
   "notifications.category.id": "NTC-6157-0397",
   "notifications.subscriber.id": "NTS-0829-7123-7123",

mpt_api_client/resources/notifications/batches.py

@@ -33,8 +33,8 @@ class BatchesService(
 ):
     """Notifications Batches service."""
 
-    def get_batch_attachment(self, batch_id: str, attachment_id: str) -> FileModel:
-        """Get batch attachment.
+    def download_attachment(self, batch_id: str, attachment_id: str) -> FileModel:
+        """Download batch attachment.
 
         Args:
             batch_id: Batch ID.
@@ -59,8 +59,8 @@ class AsyncBatchesService(
 ):
     """Async Notifications Batches service."""
 
-    async def get_batch_attachment(self, batch_id: str, attachment_id: str) -> FileModel:
-        """Get batch attachment.
+    async def download_attachment(self, batch_id: str, attachment_id: str) -> FileModel:
+        """Download batch attachment.
 
         Args:
             batch_id: Batch ID.

tests/e2e/notifications/batch/conftest.py

@@ -24,3 +24,8 @@ def batch_data(category_id, short_uuid):
         "body": "Hello world",
         "contacts": [{"email": f"{short_uuid}@example.com"}],
     }
+
+
+@pytest.fixture
+def batch_attachment_id(e2e_config):
+    return e2e_config["notifications.batch.attachment.id"]

tests/e2e/notifications/batch/test_async_batches.py

@@ -11,7 +11,7 @@ async def test_create_batch(async_batch_service, batch_data):
 
 
 async def test_get_batch(async_batch_service, batch_id):
-    result = await async_batch_service.get(batch_id)
+    result = await async_batch_service.get(batch_id, select=["attachments"])
 
     assert result.id == batch_id
 
@@ -30,7 +30,7 @@ async def test_create_batch_with_file(async_batch_service, batch_data, logo_fd):
     assert result is not None
 
 
-@pytest.mark.skip(reason="Batches attachments not implemented")
-async def test_download_attachment():
-    # TODO - Implement get and download E2E tests for attachments
-    raise NotImplementedError
+async def test_download_attachment(async_batch_service, batch_id, batch_attachment_id):
+    result = await async_batch_service.download_attachment(batch_id, batch_attachment_id)
+
+    assert result.filename == "logo.png"

tests/e2e/notifications/batch/test_batches.py

@@ -11,7 +11,7 @@ def test_create_batch(batch_service, batch_data):
 
 
 def test_get_batch(batch_service, batch_id):
-    result = batch_service.get(batch_id)
+    result = batch_service.get(batch_id, select=["attachments"])
 
     assert result.id == batch_id
 
@@ -30,7 +30,7 @@ def test_create_batch_with_file(batch_service, batch_data, logo_fd):
     assert result is not None
 
 
-@pytest.mark.skip(reason="Batches attachments not implemented")  # noqa: AAA01
-def test_download_attachment():
-    # TODO - Implement get and download E2E tests for attachment
-    raise NotImplementedError
+def test_download_attachment(batch_service, batch_id, batch_attachment_id):
+    result = batch_service.download_attachment(batch_id, batch_attachment_id)
+
+    assert result.filename == "logo.png"

tests/unit/resources/notifications/test_batches.py

@@ -1,30 +1,77 @@
 import pytest
 
+from mpt_api_client.http import AsyncHTTPClient, HTTPClient
+from mpt_api_client.http.types import Response
+from mpt_api_client.models import FileModel
 from mpt_api_client.resources.notifications.batches import (
     AsyncBatchesService,
     BatchesService,
 )
 
 
 @pytest.fixture
-def batches_service(http_client):
+def batches_service(http_client: HTTPClient) -> BatchesService:
     return BatchesService(http_client=http_client)
 
 
 @pytest.fixture
-def async_batches_service(async_http_client):
+def async_batches_service(async_http_client: AsyncHTTPClient) -> AsyncBatchesService:
     return AsyncBatchesService(http_client=async_http_client)
 
 
-@pytest.mark.parametrize("method", ["get", "create", "iterate", "get_batch_attachment"])
-def test_sync_batches_service_methods(batches_service, method):
+@pytest.fixture
+def attachment_response() -> Response:
+    return Response(
+        headers={"content-disposition": 'attachment; filename="test.csv"'},
+        status_code=200,
+        content=b"content",
+    )
+
+
+@pytest.mark.parametrize("method", ["get", "create", "iterate", "download_attachment"])
+def test_sync_batches_service_methods(batches_service: BatchesService, method: str) -> None:
     result = hasattr(batches_service, method)
 
     assert result is True
 
 
-@pytest.mark.parametrize("method", ["get", "create", "iterate", "get_batch_attachment"])
-def test_async_batches_service_methods(async_batches_service, method):
+@pytest.mark.parametrize("method", ["get", "create", "iterate", "download_attachment"])
+def test_async_batches_service_methods(
+    async_batches_service: AsyncBatchesService, method: str
+) -> None:
     result = hasattr(async_batches_service, method)
 
     assert result is True
+
+
+def test_download_attachment(mocker, batches_service, attachment_response):
+    batch_id = "B-123"
+    attachment_id = "A-123"
+    mock_request = mocker.patch.object(
+        batches_service.http_client, "request", return_value=attachment_response
+    )
+
+    result = batches_service.download_attachment(batch_id, attachment_id)
+
+    assert isinstance(result, FileModel)
+    mock_request.assert_called_once_with(
+        "get", f"/public/v1/notifications/batches/{batch_id}/attachments/{attachment_id}"
+    )
+
+
+async def test_async_download_attachment(  # noqa: WPS210
+    mocker, async_batches_service, attachment_response
+):
+    batch_id = "B-123"
+    attachment_id = "A-123"
+    mock_request = mocker.patch.object(
+        async_batches_service.http_client, "request", return_value=attachment_response
+    )
+
+    result = await async_batches_service.download_attachment(batch_id, attachment_id)
+
+    assert isinstance(result, FileModel)
+
+    mock_request.assert_called_once_with(
+        "get", f"/public/v1/notifications/batches/{batch_id}/attachments/{attachment_id}"
+    )