test: generate actions clients tests for each resource client

Author: jooola

tests/unit/actions/test_client.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import inspect
 from unittest import mock
 
 import pytest
@@ -12,6 +13,7 @@
     BoundAction,
     ResourceActionsClient,
 )
+from hcloud.core import ResourceClientBase
 
 from ..conftest import assert_bound_action1, assert_bound_action2
 
@@ -89,86 +91,198 @@ def test_wait_until_finished_max_retries(
         assert request_mock.call_count == 1
 
 
+def resource_clients_with_actions() -> list[str]:
+    """
+    Return the names of all resource clients that support actions.
+
+    Allows us to run TestResourceActionsClient against all resource clients that have an
+    resource actions client.
+    """
+    members = inspect.getmembers(
+        Client("TOKEN"),
+        predicate=lambda p: isinstance(p, ResourceClientBase) and hasattr(p, "actions"),
+    )
+    return [name for name, _, in members]
+
+
 class TestResourceActionsClient:
+    """
+    /<resource>/actions
+    /<resource>/actions/<id>
+    """
+
+    @pytest.fixture(params=resource_clients_with_actions())
+    def resource(self, request) -> str:
+        return request.param
+
     @pytest.fixture()
-    def actions_client(self, client: Client):
-        return ResourceActionsClient(client, resource="/resource")
+    def resource_client(self, client: Client, resource: str) -> ResourceActionsClient:
+        """
+        Extract the resource actions client from the client.
+        """
+        return getattr(client, resource).actions
 
     def test_get_by_id(
         self,
         request_mock: mock.MagicMock,
-        actions_client: ActionsClient,
+        resource_client: ResourceActionsClient,
+        resource: str,
         action_response,
     ):
         request_mock.return_value = action_response
 
-        action = actions_client.get_by_id(1)
+        action = resource_client.get_by_id(1)
 
         request_mock.assert_called_with(
             method="GET",
-            url="/resource/actions/1",
+            url=f"/{resource}/actions/1",
         )
 
-        assert_bound_action1(action, actions_client._parent.actions)
+        assert_bound_action1(action, resource_client._parent.actions)
 
     @pytest.mark.parametrize(
         "params",
         [
             {},
-            {"status": ["active"], "sort": ["status"], "page": 2, "per_page": 10},
+            {"status": ["running"], "sort": ["status"], "page": 2, "per_page": 10},
         ],
     )
     def test_get_list(
         self,
         request_mock: mock.MagicMock,
-        actions_client: ActionsClient,
+        resource_client: ResourceActionsClient,
+        resource: str,
         action_list_response,
         params,
     ):
         request_mock.return_value = action_list_response
 
-        result = actions_client.get_list(**params)
+        result = resource_client.get_list(**params)
 
         request_mock.assert_called_with(
             method="GET",
-            url="/resource/actions",
+            url=f"/{resource}/actions",
             params=params,
         )
 
         assert result.meta is not None
 
         actions = result.actions
         assert len(actions) == 2
-        assert_bound_action1(actions[0], actions_client._parent.actions)
-        assert_bound_action2(actions[1], actions_client._parent.actions)
+        assert_bound_action1(actions[0], resource_client._parent.actions)
+        assert_bound_action2(actions[1], resource_client._parent.actions)
 
-    @pytest.mark.parametrize("params", [{}, {"status": ["active"], "sort": ["status"]}])
+    @pytest.mark.parametrize(
+        "params",
+        [
+            {},
+            {"status": ["running"], "sort": ["status"]},
+        ],
+    )
     def test_get_all(
         self,
         request_mock: mock.MagicMock,
-        actions_client: ActionsClient,
+        resource_client: ResourceActionsClient,
+        resource: str,
+        action_list_response,
+        params,
+    ):
+        request_mock.return_value = action_list_response
+
+        actions = resource_client.get_all(**params)
+
+        request_mock.assert_called_with(
+            method="GET",
+            url=f"/{resource}/actions",
+            params={**params, "page": 1, "per_page": 50},
+        )
+
+        assert len(actions) == 2
+        assert_bound_action1(actions[0], resource_client._parent.actions)
+        assert_bound_action2(actions[1], resource_client._parent.actions)
+
+
+class TestResourceObjectActionsClient:
+    """
+    /<resource>/<id>/actions
+    """
+
+    @pytest.fixture(params=resource_clients_with_actions())
+    def resource(self, request):
+        if request.param == "primary_ips":
+            pytest.skip("not implemented yet")
+        return request.param
+
+    @pytest.fixture()
+    def resource_client(self, client: Client, resource: str) -> ResourceClientBase:
+        return getattr(client, resource)
+
+    @pytest.mark.parametrize(
+        "params",
+        [
+            {},
+            {"status": ["running"], "sort": ["status"], "page": 2, "per_page": 10},
+        ],
+    )
+    def test_get_actions_list(
+        self,
+        request_mock: mock.MagicMock,
+        resource_client: ResourceClientBase,
+        resource: str,
+        action_list_response,
+        params,
+    ):
+        request_mock.return_value = action_list_response
+
+        result = resource_client.get_actions_list(mock.MagicMock(id=1), **params)
+
+        request_mock.assert_called_with(
+            method="GET",
+            url=f"/{resource}/1/actions",
+            params=params,
+        )
+
+        assert result.meta is not None
+
+        actions = result.actions
+        assert len(actions) == 2
+        assert_bound_action1(actions[0], resource_client._parent.actions)
+        assert_bound_action2(actions[1], resource_client._parent.actions)
+
+    @pytest.mark.parametrize(
+        "params",
+        [
+            {},
+            {"status": ["running"], "sort": ["status"]},
+        ],
+    )
+    def test_get_actions(
+        self,
+        request_mock: mock.MagicMock,
+        resource_client: ResourceClientBase,
+        resource: str,
         action_list_response,
         params,
     ):
         request_mock.return_value = action_list_response
 
-        actions = actions_client.get_all(**params)
+        actions = resource_client.get_actions(mock.MagicMock(id=1), **params)
 
         request_mock.assert_called_with(
             method="GET",
-            url="/resource/actions",
+            url=f"/{resource}/1/actions",
             params={**params, "page": 1, "per_page": 50},
         )
 
         assert len(actions) == 2
-        assert_bound_action1(actions[0], actions_client._parent.actions)
-        assert_bound_action2(actions[1], actions_client._parent.actions)
+        assert_bound_action1(actions[0], resource_client._parent.actions)
+        assert_bound_action2(actions[1], resource_client._parent.actions)
 
 
 class TestActionsClient:
     @pytest.fixture()
-    def actions_client(self, client: Client):
-        return ActionsClient(client)
+    def actions_client(self, client: Client) -> ActionsClient:
+        return client.actions
 
     def test_get_by_id(
         self,
@@ -184,13 +298,13 @@ def test_get_by_id(
             method="GET",
             url="/actions/1",
         )
-        assert_bound_action1(action, actions_client._parent.actions)
+        assert_bound_action1(action, actions_client)
 
     @pytest.mark.parametrize(
         "params",
         [
             {},
-            {"status": ["active"], "sort": ["status"], "page": 2, "per_page": 10},
+            {"status": ["running"], "sort": ["status"], "page": 2, "per_page": 10},
         ],
     )
     def test_get_list(
@@ -215,10 +329,16 @@ def test_get_list(
 
         actions = result.actions
         assert len(actions) == 2
-        assert_bound_action1(actions[0], actions_client._parent.actions)
-        assert_bound_action2(actions[1], actions_client._parent.actions)
+        assert_bound_action1(actions[0], actions_client)
+        assert_bound_action2(actions[1], actions_client)
 
-    @pytest.mark.parametrize("params", [{}, {"status": ["active"], "sort": ["status"]}])
+    @pytest.mark.parametrize(
+        "params",
+        [
+            {},
+            {"status": ["running"], "sort": ["status"]},
+        ],
+    )
     def test_get_all(
         self,
         request_mock: mock.MagicMock,
@@ -238,5 +358,5 @@ def test_get_all(
         )
 
         assert len(actions) == 2
-        assert_bound_action1(actions[0], actions_client._parent.actions)
-        assert_bound_action2(actions[1], actions_client._parent.actions)
+        assert_bound_action1(actions[0], actions_client)
+        assert_bound_action2(actions[1], actions_client)