1 files changed, 41 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/tests/test_daemon.py b/src/pybind/mgr/dashboard/tests/test_daemon.py
new file mode 100644
index 000000000..2008c8630
--- /dev/null
+++ b/src/pybind/mgr/dashboard/tests/test_daemon.py
@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+from ..controllers._version import APIVersion
+from ..controllers.daemon import Daemon
+from ..tests import ControllerTestCase, patch_orch
+
+
+class DaemonTest(ControllerTestCase):
+
+    URL_DAEMON = '/api/daemon'
+
+    @classmethod
+    def setup_server(cls):
+        cls.setup_controllers([Daemon])
+
+    def test_daemon_action(self):
+        msg = "Scheduled to stop crash.b78cd1164a1b on host 'hostname'"
+
+        with patch_orch(True) as fake_client:
+            fake_client.daemons.action.return_value = msg
+            payload = {
+                'action': 'restart',
+                'container_image': None
+            }
+            self._put(f'{self.URL_DAEMON}/crash.b78cd1164a1b', payload, version=APIVersion(0, 1))
+            self.assertJsonBody(msg)
+            self.assertStatus(200)
+
+    def test_daemon_invalid_action(self):
+        payload = {
+            'action': 'invalid',
+            'container_image': None
+        }
+        with patch_orch(True):
+            self._put(f'{self.URL_DAEMON}/crash.b78cd1164a1b', payload, version=APIVersion(0, 1))
+            self.assertJsonBody({
+                'detail': 'Daemon action "invalid" is either not valid or not supported.',
+                'code': 'invalid_daemon_action',
+                'component': None
+            })
+            self.assertStatus(400)