From e6918187568dbd01842d8d1d2c808ce16a894239 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 21 Apr 2024 13:54:28 +0200
Subject: Adding upstream version 18.2.2.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../mgr/mds_autoscaler/tests/test_autoscaler.py    | 88 ++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 src/pybind/mgr/mds_autoscaler/tests/test_autoscaler.py

(limited to 'src/pybind/mgr/mds_autoscaler/tests/test_autoscaler.py')

diff --git a/src/pybind/mgr/mds_autoscaler/tests/test_autoscaler.py b/src/pybind/mgr/mds_autoscaler/tests/test_autoscaler.py
new file mode 100644
index 000000000..2d6017d4a
--- /dev/null
+++ b/src/pybind/mgr/mds_autoscaler/tests/test_autoscaler.py
@@ -0,0 +1,88 @@
+import pytest
+from unittest import mock
+
+from ceph.deployment.service_spec import ServiceSpec, PlacementSpec
+from orchestrator import DaemonDescription, OrchResult, ServiceDescription
+
+try:
+    from typing import Any, List
+except ImportError:
+    pass
+
+from mds_autoscaler.module import MDSAutoscaler
+
+
+@pytest.fixture()
+def mds_autoscaler_module():
+
+    yield MDSAutoscaler('mds_autoscaler', 0, 0)
+
+
+class TestCephadm(object):
+
+    @mock.patch("mds_autoscaler.module.MDSAutoscaler.get")
+    @mock.patch("mds_autoscaler.module.MDSAutoscaler.list_daemons")
+    @mock.patch("mds_autoscaler.module.MDSAutoscaler.describe_service")
+    @mock.patch("mds_autoscaler.module.MDSAutoscaler.apply_mds")
+    def test_scale_up(self, _apply_mds, _describe_service, _list_daemons, _get, mds_autoscaler_module: MDSAutoscaler):
+        daemons = OrchResult(result=[
+            DaemonDescription(
+                hostname='myhost',
+                daemon_type='mds',
+                daemon_id='fs_name.myhost.a'
+            ),
+            DaemonDescription(
+                hostname='myhost',
+                daemon_type='mds',
+                daemon_id='fs_name.myhost.b'
+            ),
+        ])
+        _list_daemons.return_value = daemons
+
+        services = OrchResult(result=[
+            ServiceDescription(
+                spec=ServiceSpec(
+                    service_type='mds',
+                    service_id='fs_name',
+                    placement=PlacementSpec(
+                        count=2
+                    )
+                )
+            )
+        ])
+        _describe_service.return_value = services
+
+        apply = OrchResult(result='')
+        _apply_mds.return_value = apply
+
+        _get.return_value = {
+            'filesystems': [
+                {
+                    'mdsmap': {
+                        'fs_name': 'fs_name',
+                        'in': [
+                            {
+                                'name': 'mds.fs_name.myhost.a',
+                            }
+                        ],
+                        'standby_count_wanted': 2,
+                        'max_mds': 1
+                    }
+                }
+            ],
+            'standbys': [
+                {
+                    'name': 'mds.fs_name.myhost.b',
+                }
+            ],
+
+        }
+        mds_autoscaler_module.notify('fs_map', None)
+
+        _apply_mds.assert_called_with(ServiceSpec(
+            service_type='mds',
+            service_id='fs_name',
+            placement=PlacementSpec(
+                count=3
+            )
+        ))
-- 
cgit v1.2.3