1 files changed, 129 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/tests/test_rgw.py b/src/pybind/mgr/dashboard/tests/test_rgw.py
new file mode 100644
index 00000000..9f586be2
--- /dev/null
+++ b/src/pybind/mgr/dashboard/tests/test_rgw.py
@@ -0,0 +1,129 @@
+import mock
+
+from .. import mgr
+from ..controllers.rgw import Rgw, RgwUser
+from . import ControllerTestCase  # pylint: disable=no-name-in-module
+
+
+class RgwControllerTestCase(ControllerTestCase):
+    @classmethod
+    def setup_server(cls):
+        Rgw._cp_config['tools.authenticate.on'] = False  # pylint: disable=protected-access
+        cls.setup_controllers([Rgw], '/test')
+
+    def test_status_no_service(self):
+        mgr.list_servers.return_value = []
+        self._get('/test/api/rgw/status')
+        self.assertStatus(200)
+        self.assertJsonBody({'available': False, 'message': 'No RGW service is running.'})
+
+
+class RgwUserControllerTestCase(ControllerTestCase):
+    @classmethod
+    def setup_server(cls):
+        RgwUser._cp_config['tools.authenticate.on'] = False  # pylint: disable=protected-access
+        cls.setup_controllers([RgwUser], '/test')
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    def test_user_list(self, mock_proxy):
+        mock_proxy.side_effect = [{
+            'count': 3,
+            'keys': ['test1', 'test2', 'test3'],
+            'truncated': False
+        }]
+        self._get('/test/api/rgw/user')
+        self.assertStatus(200)
+        mock_proxy.assert_has_calls([
+            mock.call('GET', 'user?list', {})
+        ])
+        self.assertJsonBody(['test1', 'test2', 'test3'])
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    def test_user_list_marker(self, mock_proxy):
+        mock_proxy.side_effect = [{
+            'count': 3,
+            'keys': ['test1', 'test2', 'test3'],
+            'marker': 'foo:bar',
+            'truncated': True
+        }, {
+            'count': 1,
+            'keys': ['admin'],
+            'truncated': False
+        }]
+        self._get('/test/api/rgw/user')
+        self.assertStatus(200)
+        mock_proxy.assert_has_calls([
+            mock.call('GET', 'user?list', {}),
+            mock.call('GET', 'user?list', {'marker': 'foo:bar'})
+        ])
+        self.assertJsonBody(['test1', 'test2', 'test3', 'admin'])
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    def test_user_list_duplicate_marker(self, mock_proxy):
+        mock_proxy.side_effect = [{
+            'count': 3,
+            'keys': ['test1', 'test2', 'test3'],
+            'marker': 'foo:bar',
+            'truncated': True
+        }, {
+            'count': 3,
+            'keys': ['test4', 'test5', 'test6'],
+            'marker': 'foo:bar',
+            'truncated': True
+        }, {
+            'count': 1,
+            'keys': ['admin'],
+            'truncated': False
+        }]
+        self._get('/test/api/rgw/user')
+        self.assertStatus(500)
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    def test_user_list_invalid_marker(self, mock_proxy):
+        mock_proxy.side_effect = [{
+            'count': 3,
+            'keys': ['test1', 'test2', 'test3'],
+            'marker': 'foo:bar',
+            'truncated': True
+        }, {
+            'count': 3,
+            'keys': ['test4', 'test5', 'test6'],
+            'marker': '',
+            'truncated': True
+        }, {
+            'count': 1,
+            'keys': ['admin'],
+            'truncated': False
+        }]
+        self._get('/test/api/rgw/user')
+        self.assertStatus(500)
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    @mock.patch.object(RgwUser, '_keys_allowed')
+    def test_user_get_with_keys(self, keys_allowed, mock_proxy):
+        keys_allowed.return_value = True
+        mock_proxy.return_value = {
+            'tenant': '',
+            'user_id': 'my_user_id',
+            'keys': [],
+            'swift_keys': []
+        }
+        self._get('/test/api/rgw/user/testuser')
+        self.assertStatus(200)
+        self.assertInJsonBody('keys')
+        self.assertInJsonBody('swift_keys')
+
+    @mock.patch('dashboard.controllers.rgw.RgwRESTController.proxy')
+    @mock.patch.object(RgwUser, '_keys_allowed')
+    def test_user_get_without_keys(self, keys_allowed, mock_proxy):
+        keys_allowed.return_value = False
+        mock_proxy.return_value = {
+            'tenant': '',
+            'user_id': 'my_user_id',
+            'keys': [],
+            'swift_keys': []
+        }
+        self._get('/test/api/rgw/user/testuser')
+        self.assertStatus(200)
+        self.assertNotIn('keys', self.jsonBody())
+        self.assertNotIn('swift_keys', self.jsonBody())