Throw TypeError in repository enable/disable macros

Currently we do not validate the type of the reponames argument to the
enable_extra_repos and disable_extra_repos macros. It should be a list,
but we merged a call where the argument was a string, which interpreted
as a list of single characters ('e', 'p', 'e', 'l'), and silently
ignored.

This change validates that the reponames argument is a list to avoid
this issue.

Change-Id: I497c8b2c1dc548e678698474d3604664cf8eeb07
This commit is contained in:
Mark Goddard 2020-02-26 12:52:44 +00:00
parent 1fe8012ce2
commit 5e7979f31d
2 changed files with 16 additions and 0 deletions

View File

@ -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 = {}

View File

@ -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')