add volume tests

digitalocean · Oct 1, 2023 · 2f0b847 · 2f0b847
1 parent 30f4ff3
commit 2f0b847
Showing 1 changed file with 160 additions and 0 deletions.
diff --git a/tests/mocked/test_block_storage.py b/tests/mocked/test_block_storage.py
@@ -330,6 +330,61 @@ def test_block_storage_snapshots_create(mock_client: Client, mock_client_url):
 
     assert create_resp == expected
 
+
+@responses.activate
+def test_volume_actions_get(mock_client: Client, mock_client_url):
+    """Tests to retrieve the status of a volume action"""
+    expected = {
+            "action": {
+                "id": 72531856,
+                "status": "completed",
+                "type": "attach_volume",
+                "started_at": "2020-11-12T17:51:03Z",
+                "completed_at": "2020-11-12T17:51:14Z",
+                "resource_type": "volume",
+                "region": {
+                "name": "New York 1",
+                "slug": "nyc1",
+                "sizes": [
+                    "s-1vcpu-1gb",
+                    "s-1vcpu-2gb",
+                    "s-1vcpu-3gb",
+                    "s-2vcpu-2gb",
+                    "s-3vcpu-1gb",
+                    "s-2vcpu-4gb",
+                    "s-4vcpu-8gb",
+                    "s-6vcpu-16gb",
+                    "s-8vcpu-32gb",
+                    "s-12vcpu-48gb",
+                    "s-16vcpu-64gb",
+                    "s-20vcpu-96gb",
+                    "s-24vcpu-128gb",
+                    "s-32vcpu-192gb"
+                ],
+                "features": [
+                    "private_networking",
+                    "backups",
+                    "ipv6",
+                    "metadata"
+                ],
+                "available": True
+                },
+                "region_slug": "nyc1"
+            }
+        }
+
+    responses.add(
+        responses.GET,
+        f"{mock_client_url}/v2/volumes/7724db7c/actions/72531856",
+        json=expected,
+        status=200,
+    )
+
+    get_resp = mock_client.volume_actions.get(volume_id="7724db7c", action_id="72531856")
+
+    assert get_resp == expected
+
+
 @responses.activate
 def test_volume_actions_list(mock_client: Client, mock_client_url):
     """Tests retrieving all actions that have been executed on a volume"""
@@ -388,3 +443,108 @@ def test_volume_actions_list(mock_client: Client, mock_client_url):
     get_resp = mock_client.volume_actions.list(volume_id="7724db7c")
 
     assert get_resp == expected
+
+
+@responses.activate
+def test_volume_actions_post(mock_client: Client, mock_client_url):
+    """Tests to initiate an action on a block storage volume by Name"""
+    expected = {
+                "action": {
+                    "id": 72531856,
+                    "status": "completed",
+                    "type": "attach_volume",
+                    "started_at": "2020-11-12T17:51:03Z",
+                    "completed_at": "2020-11-12T17:51:14Z",
+                    "resource_type": "volume",
+                    "region": {
+                    "name": "New York 1",
+                    "slug": "nyc1",
+                    "sizes": [
+                        "s-1vcpu-1gb",
+                        "s-1vcpu-2gb",
+                        "s-1vcpu-3gb",
+                        "s-2vcpu-2gb",
+                        "s-3vcpu-1gb",
+                        "s-2vcpu-4gb",
+                        "s-4vcpu-8gb",
+                        "s-6vcpu-16gb",
+                        "s-8vcpu-32gb",
+                        "s-12vcpu-48gb",
+                        "s-16vcpu-64gb",
+                        "s-20vcpu-96gb",
+                        "s-24vcpu-128gb",
+                        "s-32vcpu-192gb"
+                    ],
+                    "features": [
+                        "private_networking",
+                        "backups",
+                        "ipv6",
+                        "metadata"
+                    ],
+                    "available": True
+                    },
+                    "region_slug": "nyc1"
+                }
+            }
+
+    post_body={
+            "type": "attach",
+            "volume_name": "example",
+            "droplet_id": 11612190,
+            "region": "nyc1",
+            "tags": [
+            "aninterestingtag"
+            ]
+    }
+
+    responses.add(
+        responses.POST,
+        f"{mock_client_url}/v2/volumes/actions",
+        json=expected,
+        status=202,
+    )
+
+    create_resp = mock_client.volume_actions.post(
+        body=post_body
+    )
+
+    assert create_resp == expected
+
+
+@responses.activate
+def test_volume_actions_post_by_id(mock_client: Client, mock_client_url):
+    """Tests to initiate an action on a block storage volume by Id"""
+    expected = {
+        "action": {
+        "id": 72531856,
+        "status": "completed",
+        "type": "attach_volume",
+        "started_at": "2020-11-12T17:51:03Z",
+        "completed_at": "2020-11-12T17:51:14Z",
+        "resource_type": "volume",
+        "region": {},
+        "region_slug": "nyc1"
+        }
+    }
+
+    post_body = {
+        "type": "attach",
+        "droplet_id": 11612190,
+        "region": "nyc1",
+        "tags": [
+            "aninterestingtag"
+        ]
+    }
+
+    responses.add(
+        responses.POST,
+        f"{mock_client_url}/v2/volumes/7724db7c/actions",
+        json=expected,
+        status=202,
+    )
+
+    create_resp = mock_client.volume_actions.post_by_id(
+        volume_id="7724db7c", body=post_body
+    )
+
+    assert create_resp == expected