summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorChristian Ehrhardt <christian.ehrhardt@canonical.com>2016-05-11 16:52:17 +0200
committerChristian Ehrhardt <christian.ehrhardt@canonical.com>2016-05-11 16:52:17 +0200
commita9cd544d7de1af90f6c5cf9df43135b530acb308 (patch)
treece63159b4ad5ddaa99154464b23a28eefe47f660
parent0a35ba5b8d85077a865a137887aa8cfd76405eb2 (diff)
downloadvyos-cloud-init-a9cd544d7de1af90f6c5cf9df43135b530acb308.tar.gz
vyos-cloud-init-a9cd544d7de1af90f6c5cf9df43135b530acb308.zip
split test_apt_source_key into one for key and one for keyid
-rw-r--r--tests/unittests/test_handler/test_handler_apt_source.py32
1 files changed, 30 insertions, 2 deletions
diff --git a/tests/unittests/test_handler/test_handler_apt_source.py b/tests/unittests/test_handler/test_handler_apt_source.py
index 849f23c9..a7db0fa6 100644
--- a/tests/unittests/test_handler/test_handler_apt_source.py
+++ b/tests/unittests/test_handler/test_handler_apt_source.py
@@ -107,6 +107,34 @@ class TestAptSourceConfig(TestCase):
"multiverse"),
contents, flags=re.IGNORECASE))
+ def test_apt_source_keyid(self):
+ """ test_apt_source_keyid
+ Test specification of a source + keyid
+ """
+ params = self._get_default_params()
+ cfg = {'source': ('deb '
+ 'http://ppa.launchpad.net/'
+ 'smoser/cloud-init-test/ubuntu'
+ ' xenial main'),
+ 'keyid': "03683F77",
+ 'filename': self.aptlistfile}
+
+ with mock.patch.object(util, 'subp',
+ return_value=('fakekey 1234', '')) as mockobj:
+ cc_apt_configure.add_sources([cfg], params)
+
+ mockobj.assert_called_with(('apt-key', 'add', '-'), 'fakekey 1234')
+
+ self.assertTrue(os.path.isfile(self.aptlistfile))
+
+ contents = load_tfile_or_url(self.aptlistfile)
+ self.assertTrue(re.search(r"%s %s %s %s\n" %
+ ("deb",
+ ('http://ppa.launchpad.net/smoser/'
+ 'cloud-init-test/ubuntu'),
+ "xenial", "main"),
+ contents, flags=re.IGNORECASE))
+
def test_apt_source_key(self):
""" test_apt_source_key
Test specification of a source + key
@@ -116,14 +144,14 @@ class TestAptSourceConfig(TestCase):
'http://ppa.launchpad.net/'
'smoser/cloud-init-test/ubuntu'
' xenial main'),
- 'keyid': "03683F77",
+ 'key': "fakekey 4321",
'filename': self.aptlistfile}
with mock.patch.object(util, 'subp',
return_value=('fakekey 1234', '')) as mockobj:
cc_apt_configure.add_sources([cfg], params)
- mockobj.assert_called_with(('apt-key', 'add', '-'), 'fakekey 1234')
+ mockobj.assert_called_with(('apt-key', 'add', '-'), 'fakekey 4321')
self.assertTrue(os.path.isfile(self.aptlistfile))