1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# This file is part of cloud-init. See LICENSE file for license information.
from cloudinit.config import cc_yum_add_repo
from cloudinit import util
from .. import helpers
try:
from configparser import ConfigParser
except ImportError:
from ConfigParser import ConfigParser
import logging
import shutil
from six import StringIO
import tempfile
LOG = logging.getLogger(__name__)
class TestConfig(helpers.FilesystemMockingTestCase):
def setUp(self):
super(TestConfig, self).setUp()
self.tmp = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self.tmp)
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/$barch',
'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")
parser = ConfigParser()
parser.readfp(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',
}
}
for section in expected:
self.assertTrue(parser.has_section(section),
"Contains section {}".format(section))
for k, v in expected[section].items():
self.assertEqual(parser.get(section, k), v)
def test_write_config_array(self):
cfg = {
'yum_repos': {
'puppetlabs-products': {
'name': 'Puppet Labs Products El 6 - $basearch',
'baseurl':
'http://yum.puppetlabs.com/el/6/products/$basearch',
'gpgkey': [
'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-puppetlabs',
'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-puppet',
],
'enabled': True,
'gpgcheck': True,
}
}
}
self.patchUtils(self.tmp)
cc_yum_add_repo.handle('yum_add_repo', cfg, None, LOG, [])
contents = util.load_file("/etc/yum.repos.d/puppetlabs_products.repo")
parser = ConfigParser()
parser.readfp(StringIO(contents))
expected = {
'puppetlabs_products': {
'name': 'Puppet Labs Products El 6 - $basearch',
'baseurl': 'http://yum.puppetlabs.com/el/6/products/$basearch',
'gpgkey': 'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-puppetlabs\n'
'file:///etc/pki/rpm-gpg/RPM-GPG-KEY-puppet',
'enabled': '1',
'gpgcheck': '1',
}
}
for section in expected:
self.assertTrue(parser.has_section(section),
"Contains section {}".format(section))
for k, v in expected[section].items():
self.assertEqual(parser.get(section, k), v)
# vi: ts=4 expandtab
|
