From 44d2f42eae7de8fe16ec7cc9fe22bd1751926438 Mon Sep 17 00:00:00 2001
From: Cristian Matiut <cmatiut@cloudbasesolutions.com>
Date: Thu, 14 Dec 2023 12:47:27 +0200
Subject: [PATCH] Add tests for `coriolis.api.v1.replica_actions` module

---
 coriolis/tests/api/v1/test_replica_actions.py | 105 ++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 coriolis/tests/api/v1/test_replica_actions.py

diff --git a/coriolis/tests/api/v1/test_replica_actions.py b/coriolis/tests/api/v1/test_replica_actions.py
new file mode 100644
index 000000000..0b8d7eeb3
--- /dev/null
+++ b/coriolis/tests/api/v1/test_replica_actions.py
@@ -0,0 +1,105 @@
+# Copyright 2023 Cloudbase Solutions Srl
+# All Rights Reserved.
+
+from unittest import mock
+
+from webob import exc
+
+from coriolis.api.v1 import replica_actions
+from coriolis.api.v1.views import replica_tasks_execution_view
+from coriolis import exception
+from coriolis.replicas import api
+from coriolis.tests import test_base
+from coriolis.tests import testutils
+
+
+class ReplicaActionsControllerTestCase(test_base.CoriolisBaseTestCase):
+    """Test suite for the Coriolis Replica Actions v1 API"""
+
+    def setUp(self):
+        super(ReplicaActionsControllerTestCase, self).setUp()
+        self.replica_actions = replica_actions.ReplicaActionsController()
+
+    @mock.patch.object(replica_tasks_execution_view, 'single')
+    @mock.patch.object(api.API, 'delete_disks')
+    def test_delete_disks(
+        self,
+        mock_delete_disks,
+        mock_single
+    ):
+        mock_req = mock.Mock()
+        mock_context = mock.Mock()
+        mock_req.environ = {'coriolis.context': mock_context}
+        id = mock.sentinel.id
+        body = mock.sentinel.body
+
+        result = testutils.get_wrapped_function(
+            self.replica_actions._delete_disks)(
+                mock_req,
+                id,
+                body
+        )
+
+        self.assertEqual(
+            mock_single.return_value,
+            result
+        )
+
+        mock_context.can.assert_called_once_with(
+            "migration:replicas:delete_disks")
+        mock_delete_disks.assert_called_once_with(mock_context, id)
+        mock_single.assert_called_once_with(mock_delete_disks.return_value)
+
+    @mock.patch.object(replica_tasks_execution_view, 'single')
+    @mock.patch.object(api.API, 'delete_disks')
+    def test_delete_disks_not_found(
+        self,
+        mock_delete_disks,
+        mock_single
+    ):
+        mock_req = mock.Mock()
+        mock_context = mock.Mock()
+        mock_req.environ = {'coriolis.context': mock_context}
+        id = mock.sentinel.id
+        body = mock.sentinel.body
+        mock_delete_disks.side_effect = exception.NotFound()
+
+        self.assertRaises(
+            exc.HTTPNotFound,
+            testutils.get_wrapped_function(self.replica_actions._delete_disks),
+            req=mock_req,
+            id=id,
+            body=body
+        )
+
+        mock_context.can.assert_called_once_with(
+            "migration:replicas:delete_disks")
+        mock_delete_disks.assert_called_once_with(mock_context, id)
+        mock_single.assert_not_called()
+    
+    @mock.patch.object(replica_tasks_execution_view, 'single')
+    @mock.patch.object(api.API, 'delete_disks')
+    def test_delete_disks_invalid_parameter_value(
+        self,
+        mock_delete_disks,
+        mock_single
+    ):
+        mock_req = mock.Mock()
+        mock_context = mock.Mock()
+        mock_req.environ = {'coriolis.context': mock_context}
+        id = mock.sentinel.id
+        body = mock.sentinel.body
+        mock_delete_disks.side_effect = exception.InvalidParameterValue("err")
+
+        self.assertRaises(
+            exc.HTTPNotFound,
+            testutils.get_wrapped_function(self.replica_actions._delete_disks),
+            req=mock_req,
+            id=id,
+            body=body
+        )
+
+        mock_context.can.assert_called_once_with(
+            "migration:replicas:delete_disks")
+        mock_delete_disks.assert_called_once_with(mock_context, id)
+        mock_single.assert_not_called()