68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from cloudinit import util
|
|
|
|
from cloudinit.config import cc_yum_add_repo
|
|
|
|
from tests.unittests import helpers
|
|
|
|
import logging
|
|
|
|
from StringIO import StringIO
|
|
|
|
import configobj
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class TestConfig(helpers.FilesystemMockingTestCase):
|
|
def setUp(self):
|
|
super(TestConfig, self).setUp()
|
|
self.tmp = self.makeDir(prefix="unittest_")
|
|
|
|
def test_bad_config(self):
|
|
cfg = {
|
|
'yum_repos': {
|
|
'epel-testing': {
|
|
'name': 'Extra Packages for Enterprise Linux 5 - Testing',
|
|
# Missing this should cause the repo not to be written
|
|
#'baseurl': 'http://blah.org/pub/epel/testing/5/$basearch',
|
|
'enabled': False,
|
|
'gpgcheck': True,
|
|
'gpgkey': 'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-EPEL',
|
|
'failovermethod': 'priority',
|
|
},
|
|
},
|
|
}
|
|
self.patchUtils(self.tmp)
|
|
cc_yum_add_repo.handle('yum_add_repo', cfg, None, LOG, [])
|
|
self.assertRaises(IOError, util.load_file,
|
|
"/etc/yum.repos.d/epel_testing.repo")
|
|
|
|
def test_write_config(self):
|
|
cfg = {
|
|
'yum_repos': {
|
|
'epel-testing': {
|
|
'name': 'Extra Packages for Enterprise Linux 5 - Testing',
|
|
'baseurl': 'http://blah.org/pub/epel/testing/5/$basearch',
|
|
'enabled': False,
|
|
'gpgcheck': True,
|
|
'gpgkey': 'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-EPEL',
|
|
'failovermethod': 'priority',
|
|
},
|
|
},
|
|
}
|
|
self.patchUtils(self.tmp)
|
|
cc_yum_add_repo.handle('yum_add_repo', cfg, None, LOG, [])
|
|
contents = util.load_file("/etc/yum.repos.d/epel_testing.repo")
|
|
contents = configobj.ConfigObj(StringIO(contents))
|
|
expected = {
|
|
'epel_testing': {
|
|
'name': 'Extra Packages for Enterprise Linux 5 - Testing',
|
|
'failovermethod': 'priority',
|
|
'gpgkey': 'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-EPEL',
|
|
'enabled': '0',
|
|
'baseurl': 'http://blah.org/pub/epel/testing/5/$basearch',
|
|
'gpgcheck': '1',
|
|
}
|
|
}
|
|
self.assertEquals(expected, dict(contents))
|