import 15.2.4

[ceph.git] / ceph / src / pybind / mgr / dashboard / tests / test_rgw_client.py

diff --git a/ceph/src/pybind/mgr/dashboard/tests/test_rgw_client.py b/ceph/src/pybind/mgr/dashboard/tests/test_rgw_client.py
index 7e153a2eefc6671639287888012b65fcbd8cda38..d69ab2f48df5c87bd6b1cfe67e8b814c35ec29c5 100644 (file)
--- a/ceph/src/pybind/mgr/dashboard/tests/test_rgw_client.py
+++ b/ceph/src/pybind/mgr/dashboard/tests/test_rgw_client.py
@@ -3,9 +3,9 @@
 import unittest
 
 try:
-    from mock import patch
-except ImportError:
     from unittest.mock import patch
+except ImportError:
+    from mock import patch  # type: ignore
 
 from ..services.rgw_client import RgwClient, _parse_frontend_config
 from ..settings import Settings
@@ -124,6 +124,23 @@ class RgwClientTest(unittest.TestCase, KVStoreMockMixin):
         }
         self.assertEqual(expected_result, instance.get_placement_targets())
 
+    @patch.object(RgwClient, '_get_realms_info')
+    def test_get_realms(self, realms_info):
+        realms_info.side_effect = [
+            {
+                'default_info': '51de8373-bc24-4f74-a9b7-8e9ef4cb71f7',
+                'realms': [
+                    'realm1',
+                    'realm2'
+                ]
+            },
+            {}
+        ]
+        instance = RgwClient.admin_instance()
+
+        self.assertEqual(['realm1', 'realm2'], instance.get_realms())
+        self.assertEqual([], instance.get_realms())
+
 
 class RgwClientHelperTest(unittest.TestCase):
     def test_parse_frontend_config_1(self):