def test_apt_src_ppa_tri(self):
"""Test adding three ppa's"""
params = self._get_default_params()
cfg1 = {'source': 'ppa:smoser/cloud-init-test',
'filename': self.aptlistfile}
cfg2 = {'source': 'ppa:smoser/cloud-init-test2',
'filename': self.aptlistfile2}
cfg3 = {'source': 'ppa:smoser/cloud-init-test3',
'filename': self.aptlistfile3}
# default matcher needed for ppa
matcher = re.compile(r'^[\w-]+:\w').search
with mock.patch.object(util, 'subp') as mockobj:
cc_apt_configure.add_apt_sources([cfg1, cfg2, cfg3], params,
aa_repo_match=matcher)
calls = [call(['add-apt-repository', 'ppa:smoser/cloud-init-test']),
call(['add-apt-repository', 'ppa:smoser/cloud-init-test2']),
call(['add-apt-repository', 'ppa:smoser/cloud-init-test3'])]
mockobj.assert_has_calls(calls, any_order=True)
# adding ppa should ignore all filenames (uses add-apt-repository)
self.assertFalse(os.path.isfile(self.aptlistfile))
self.assertFalse(os.path.isfile(self.aptlistfile2))
self.assertFalse(os.path.isfile(self.aptlistfile3))
def apt_src_keyid(self, filename, cfg, keynum):
"""apt_src_keyid
Test specification of a source + keyid
"""
params = self._get_default_params()
with mock.patch.object(util, 'subp',
return_value=('fakekey 1234', '')) as mockobj:
cc_apt_configure.add_apt_sources(cfg, params)
# check if it added the right ammount of keys
calls = []
for _ in range(keynum):
calls.append(call(('apt-key', 'add', '-'), 'fakekey 1234'))
mockobj.assert_has_calls(calls, any_order=True)
self.assertTrue(os.path.isfile(filename))
contents = load_tfile_or_url(filename)
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_src_keyidonly(self):
"""Test specification of a keyid without source"""
params = self._get_default_params()
cfg = {'keyid': "03683F77",
'filename': self.aptlistfile}
with mock.patch.object(util, 'subp',
return_value=('fakekey 1212', '')) as mockobj:
cc_apt_configure.add_apt_sources([cfg], params)
mockobj.assert_called_with(('apt-key', 'add', '-'), 'fakekey 1212')
# filename should be ignored on key only
self.assertFalse(os.path.isfile(self.aptlistfile))
def test_apt_src_keyonly(self):
"""Test specifying key without source"""
params = self._get_default_params()
cfg = {'key': "fakekey 4242",
'filename': self.aptlistfile}
with mock.patch.object(util, 'subp') as mockobj:
cc_apt_configure.add_apt_sources([cfg], params)
mockobj.assert_called_once_with(('apt-key', 'add', '-'),
'fakekey 4242')
# filename should be ignored on key only
self.assertFalse(os.path.isfile(self.aptlistfile))
def apt_src_replacement(self, filename, cfg):
"""apt_src_replace
Test Autoreplacement of MIRROR and RELEASE in source specs
"""
params = self._get_default_params()
cc_apt_configure.add_apt_sources(cfg, params)
self.assertTrue(os.path.isfile(filename))
contents = load_tfile_or_url(filename)
self.assertTrue(re.search(r"%s %s %s %s\n" %
("deb", params['MIRROR'], params['RELEASE'],
"multiverse"),
contents, flags=re.IGNORECASE))
def apt_src_basic(self, filename, cfg):
"""apt_src_basic
Test Fix deb source string, has to overwrite mirror conf in params
"""
params = self._get_default_params()
cc_apt_configure.add_apt_sources(cfg, params)
self.assertTrue(os.path.isfile(filename))
contents = load_tfile_or_url(filename)
self.assertTrue(re.search(r"%s %s %s %s\n" %
("deb", "http://archive.ubuntu.com/ubuntu",
"karmic-backports",
"main universe multiverse restricted"),
contents, flags=re.IGNORECASE))
def test_apt_src_ppa(self):
"""Test adding a ppa"""
params = self._get_default_params()
cfg = {'source': 'ppa:smoser/cloud-init-test',
'filename': self.aptlistfile}
# default matcher needed for ppa
matcher = re.compile(r'^[\w-]+:\w').search
with mock.patch.object(util, 'subp') as mockobj:
cc_apt_configure.add_apt_sources([cfg], params,
aa_repo_match=matcher)
mockobj.assert_called_once_with(['add-apt-repository',
'ppa:smoser/cloud-init-test'])
# adding ppa should ignore filename (uses add-apt-repository)
self.assertFalse(os.path.isfile(self.aptlistfile))
def apt_src_keyid_real(self, cfg, expectedkey):
"""apt_src_keyid_real
Test specification of a keyid without source including
up to addition of the key (add_apt_key_raw mocked to keep the
environment as is)
"""
params = self._get_default_params()
with mock.patch.object(cc_apt_configure, 'add_apt_key_raw') as mockkey:
with mock.patch.object(gpg, 'get_key_by_id',
return_value=expectedkey) as mockgetkey:
cc_apt_configure.add_apt_sources([cfg], params)
mockgetkey.assert_called_with(cfg['keyid'],
cfg.get('keyserver',
'keyserver.ubuntu.com'))
mockkey.assert_called_with(expectedkey)
# filename should be ignored on key only
self.assertFalse(os.path.isfile(self.aptlistfile))
def apt_src_key(self, filename, cfg):
"""apt_src_key
Test specification of a source + key
"""
params = self._get_default_params()
with mock.patch.object(util, 'subp') as mockobj:
cc_apt_configure.add_apt_sources([cfg], params)
mockobj.assert_called_with(('apt-key', 'add', '-'), 'fakekey 4321')
self.assertTrue(os.path.isfile(filename))
contents = load_tfile_or_url(filename)
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 _add_apt_sources(*args, **kwargs):
with mock.patch.object(cc_apt_configure, 'update_packages'):
cc_apt_configure.add_apt_sources(*args, **kwargs)
def _add_apt_sources(*args, **kwargs):
with mock.patch.object(cc_apt_configure, 'update_packages'):
cc_apt_configure.add_apt_sources(*args, **kwargs)
