Add additional mixin tests for activatable resource

Author: Robert Segal

setup.cfg

@@ -41,6 +41,7 @@ per-file-ignores =
   tests/http/test_service.py: WPS204 WPS202
   tests/http/test_mixins.py: WPS204 WPS202
   tests/resources/catalog/test_products.py: WPS202 WPS210
+  tests/resources/catalog/test_mixins.py: WPS118 WPS202 WPS204
 
   tests/*:
     # Allow magic strings.

tests/resources/catalog/test_mixins.py

@@ -2,6 +2,11 @@
 import pytest
 import respx
 
+from mpt_api_client.resources.catalog.pricing_policy_attachments import (
+    AsyncPricingPolicyAttachmentsService,
+    PricingPolicyAttachment,
+    PricingPolicyAttachmentsService,
+)
 from mpt_api_client.resources.catalog.products import AsyncProductsService, Product, ProductsService
 
 
@@ -15,6 +20,20 @@ def async_product_service(async_http_client):
     return AsyncProductsService(http_client=async_http_client)
 
 
+@pytest.fixture
+def pricing_policy_attachments_service(http_client):
+    return PricingPolicyAttachmentsService(
+        http_client=http_client, endpoint_params={"pricing_policy_id": "PRP-0000-0001"}
+    )
+
+
+@pytest.fixture
+def async_pricing_policy_attachments_service(async_http_client):
+    return AsyncPricingPolicyAttachmentsService(
+        http_client=async_http_client, endpoint_params={"pricing_policy_id": "PRP-0000-0001"}
+    )
+
+
 @pytest.mark.parametrize(
     ("action", "input_status"),
     [
@@ -31,7 +50,7 @@ def test_custom_resource_actions(product_service, action, input_status):
             f"https://api.example.com/public/v1/catalog/products/PRD-123/{action}"
         ).mock(
             return_value=httpx.Response(
-                status_code=200,
+                status_code=httpx.codes.OK,
                 headers={"content-type": "application/json"},
                 json=response_expected_data,
             )
@@ -62,7 +81,7 @@ def test_custom_resource_actions_no_data(product_service, action, input_status):
             f"https://api.example.com/public/v1/catalog/products/PRD-123/{action}"
         ).mock(
             return_value=httpx.Response(
-                status_code=200,
+                status_code=httpx.codes.OK,
                 headers={"content-type": "application/json"},
                 json=response_expected_data,
             )
@@ -93,7 +112,7 @@ async def test_async_custom_resource_actions(async_product_service, action, inpu
             f"https://api.example.com/public/v1/catalog/products/PRD-123/{action}"
         ).mock(
             return_value=httpx.Response(
-                status_code=200,
+                status_code=httpx.codes.OK,
                 headers={"content-type": "application/json"},
                 json=response_expected_data,
             )
@@ -124,7 +143,7 @@ async def test_async_custom_resource_actions_no_data(async_product_service, acti
             f"https://api.example.com/public/v1/catalog/products/PRD-123/{action}"
         ).mock(
             return_value=httpx.Response(
-                status_code=200,
+                status_code=httpx.codes.OK,
                 headers={"content-type": "application/json"},
                 json=response_expected_data,
             )
@@ -137,3 +156,143 @@ async def test_async_custom_resource_actions_no_data(async_product_service, acti
         assert request.content == request_expected_content
         assert order.to_dict() == response_expected_data
         assert isinstance(order, Product)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("activate", {"id": "PRP-0000-0000-0001", "status": "update"}),
+        ("deactivate", {"id": "PRP-0000-0000-0001", "status": "update"}),
+    ],
+)
+def test_custom_resource_activatable_actions(
+    pricing_policy_attachments_service, action, input_status
+):
+    request_expected_content = b'{"id":"PRP-0000-0000-0001","status":"update"}'
+    response_expected_data = {"id": "PRP-0000-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/catalog/pricing-policies/PRP-0000-0001"
+            f"/attachments/PRP-0000-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        attachment = getattr(pricing_policy_attachments_service, action)(
+            "PRP-0000-0000-0001", input_status
+        )
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert attachment.to_dict() == response_expected_data
+        assert isinstance(attachment, PricingPolicyAttachment)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("activate", None),
+        ("deactivate", None),
+    ],
+)
+def test_custom_resource_activatable_actions_no_data(
+    pricing_policy_attachments_service, action, input_status
+):
+    request_expected_content = b""
+    response_expected_data = {"id": "PRP-0000-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/catalog/pricing-policies/PRP-0000-0001"
+            f"/attachments/PRP-0000-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        attachment = getattr(pricing_policy_attachments_service, action)(
+            "PRP-0000-0000-0001", input_status
+        )
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert attachment.to_dict() == response_expected_data
+        assert isinstance(attachment, PricingPolicyAttachment)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("activate", {"id": "PRP-0000-0000-0001", "status": "update"}),
+        ("deactivate", {"id": "PRP-0000-0000-0001", "status": "update"}),
+    ],
+)
+async def test_async_custom_resource_activatable_actions(
+    async_pricing_policy_attachments_service, action, input_status
+):
+    request_expected_content = b'{"id":"PRP-0000-0000-0001","status":"update"}'
+    response_expected_data = {"id": "PRP-0000-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/catalog/pricing-policies/PRP-0000-0001"
+            f"/attachments/PRP-0000-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        attachment = await getattr(async_pricing_policy_attachments_service, action)(
+            "PRP-0000-0000-0001", input_status
+        )
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert attachment.to_dict() == response_expected_data
+        assert isinstance(attachment, PricingPolicyAttachment)
+
+
+@pytest.mark.parametrize(
+    ("action", "input_status"),
+    [
+        ("activate", None),
+        ("deactivate", None),
+    ],
+)
+async def test_async_custom_resource_activatable_actions_no_data(
+    async_pricing_policy_attachments_service, action, input_status
+):
+    request_expected_content = b""
+    response_expected_data = {"id": "PRP-0000-0000-0001", "status": "new_status"}
+    with respx.mock:
+        mock_route = respx.post(
+            f"https://api.example.com/public/v1/catalog/pricing-policies/PRP-0000-0001"
+            f"/attachments/PRP-0000-0000-0001/{action}"
+        ).mock(
+            return_value=httpx.Response(
+                status_code=httpx.codes.OK,
+                headers={"content-type": "application/json"},
+                json=response_expected_data,
+            )
+        )
+        attachment = await getattr(async_pricing_policy_attachments_service, action)(
+            "PRP-0000-0000-0001", input_status
+        )
+
+        assert mock_route.call_count == 1
+        request = mock_route.calls[0].request
+
+        assert request.content == request_expected_content
+        assert attachment.to_dict() == response_expected_data
+        assert isinstance(attachment, PricingPolicyAttachment)