1 files changed, 73 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/tests/test_home.py b/src/pybind/mgr/dashboard/tests/test_home.py
new file mode 100644
index 000000000..883c6cc6a
--- /dev/null
+++ b/src/pybind/mgr/dashboard/tests/test_home.py
@@ -0,0 +1,73 @@
+
+import logging
+import os
+
+try:
+    import mock
+except ImportError:
+    import unittest.mock as mock
+
+from .. import mgr
+from ..controllers.home import HomeController, LanguageMixin
+from ..tests import ControllerTestCase, FakeFsMixin
+
+logger = logging.getLogger()
+
+
+class HomeTest(ControllerTestCase, FakeFsMixin):
+    @classmethod
+    def setup_server(cls):
+        frontend_path = mgr.get_frontend_path()
+        cls.fs.reset()
+        cls.fs.create_dir(frontend_path)
+        cls.fs.create_file(
+            os.path.join(frontend_path, '..', 'package.json'),
+            contents='{"config":{"locale": "en"}}')
+        with mock.patch(cls.builtins_open, new=cls.f_open),\
+                mock.patch('os.listdir', new=cls.f_os.listdir):
+            lang = LanguageMixin()
+            cls.fs.create_file(
+                os.path.join(lang.DEFAULT_LANGUAGE_PATH, 'index.html'),
+                contents='<!doctype html><html lang="en"><body></body></html>')
+            cls.setup_controllers([HomeController])
+
+    @mock.patch(FakeFsMixin.builtins_open, new=FakeFsMixin.f_open)
+    @mock.patch('os.stat', new=FakeFsMixin.f_os.stat)
+    @mock.patch('os.listdir', new=FakeFsMixin.f_os.listdir)
+    def test_home_default_lang(self):
+        self._get('/')
+        self.assertStatus(200)
+        logger.info(self.body)
+        self.assertIn('<html lang="en">', self.body.decode('utf-8'))
+
+    @mock.patch(FakeFsMixin.builtins_open, new=FakeFsMixin.f_open)
+    @mock.patch('os.stat', new=FakeFsMixin.f_os.stat)
+    @mock.patch('os.listdir', new=FakeFsMixin.f_os.listdir)
+    def test_home_uplevel_check(self):
+        self._get('/../../../../../../etc/shadow')
+        self.assertStatus(403)
+
+    @mock.patch(FakeFsMixin.builtins_open, new=FakeFsMixin.f_open)
+    @mock.patch('os.stat', new=FakeFsMixin.f_os.stat)
+    @mock.patch('os.listdir', new=FakeFsMixin.f_os.listdir)
+    def test_home_en(self):
+        self._get('/', headers=[('Accept-Language', 'en-US')])
+        self.assertStatus(200)
+        logger.info(self.body)
+        self.assertIn('<html lang="en">', self.body.decode('utf-8'))
+
+    @mock.patch(FakeFsMixin.builtins_open, new=FakeFsMixin.f_open)
+    @mock.patch('os.stat', new=FakeFsMixin.f_os.stat)
+    @mock.patch('os.listdir', new=FakeFsMixin.f_os.listdir)
+    def test_home_non_supported_lang(self):
+        self._get('/', headers=[('Accept-Language', 'NO-NO')])
+        self.assertStatus(200)
+        logger.info(self.body)
+        self.assertIn('<html lang="en">', self.body.decode('utf-8'))
+
+    @mock.patch(FakeFsMixin.builtins_open, new=FakeFsMixin.f_open)
+    @mock.patch('os.stat', new=FakeFsMixin.f_os.stat)
+    @mock.patch('os.listdir', new=FakeFsMixin.f_os.listdir)
+    def test_home_multiple_subtags_lang(self):
+        self._get('/', headers=[('Accept-Language', 'zh-Hans-CN')])
+        self.assertStatus(200)