Merge "Throw TypeError in repository enable/disable macros"

2 files changed, 16 insertions, 0 deletions

diff --git a/kolla/template/methods.py b/kolla/template/methods.py
index 66c5dcfa9..fb470a65f 100644
--- a/kolla/template/methods.py
+++ b/kolla/template/methods.py
@@ -88,6 +88,9 @@ def handle_repos(context, reponames, mode):
     else:
         raise KeyError
 
+    if not isinstance(reponames, list):
+        raise TypeError("First argument should be a list of repositories")
+
     repofile = os.path.dirname(os.path.realpath(__file__)) + '/repos.yaml'
     with open(repofile, 'r') as repos_file:
         repo_data = {}
diff --git a/kolla/tests/test_methods.py b/kolla/tests/test_methods.py
index d2f6eff9d..c33aba7f9 100644
--- a/kolla/tests/test_methods.py
+++ b/kolla/tests/test_methods.py
@@ -150,3 +150,16 @@ class MethodsTest(base.TestCase):
         result = methods.handle_repos(template_vars, ['grafana'], 'disable')
         expectCmd = ''
         self.assertEqual(expectCmd, result)
+
+    def test_handle_repos_string(self):
+        template_vars = {
+            'base_arch': 'x86_64',
+            'base_distro': 'debian',
+            'base_package_type': 'deb'
+        }
+
+        self.assertRaisesRegex(TypeError,
+                               r'First argument should be a list of '
+                               r'repositories',
+                               methods.handle_repos, template_vars, 'grafana',
+                               'disable')