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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
|
# This file is part of curtin. See LICENSE file for copyright and license info.
""" test_apt_source
Collection of tests for the apt configuration features
"""
import textwrap
from . import VMBaseClass
from .releases import base_vm_classes as relbase
from unittest import SkipTest
from curtin import util
from curtin.config import load_config
class TestAptSrcAbs(VMBaseClass):
"""TestAptSrcAbs - Basic tests for apt features of curtin"""
test_type = 'config'
interactive = False
extra_disks = []
fstab_expected = {}
disk_to_check = []
extra_collect_scripts = [textwrap.dedent("""
cd OUTPUT_COLLECT_D
apt-key list "F430BBA5" > keyid-F430BBA5
apt-key list "0165013E" > keyppa-0165013E
apt-key list "F470A0AC" > keylongid-F470A0AC
apt-key list "8280B242" > keyraw-8280B242
ls -laF /etc/apt/sources.list.d/ > sources.list.d
cp /etc/apt/sources.list.d/curtin-dev-ppa.list .
cp /etc/apt/sources.list.d/my-repo2.list .
cp /etc/apt/sources.list.d/my-repo4.list .
cp /etc/apt/sources.list.d/curtin-dev-ubuntu-test-archive-*.list .
find /etc/apt/sources.list.d/ -maxdepth 1 -name "*ignore*" | wc -l > ic
apt-config dump | grep Retries > aptconf
cp /etc/apt/sources.list sources.list
cp /etc/cloud/cloud.cfg.d/curtin-preserve-sources.cfg .
exit 0
""")]
mirror = "http://us.archive.ubuntu.com/ubuntu"
secmirror = "http://security.ubuntu.com/ubuntu"
def test_output_files_exist(self):
"""test_output_files_exist - Check if all output files exist"""
self.output_files_exist(
["fstab", "ic", "keyid-F430BBA5", "keylongid-F470A0AC",
"keyraw-8280B242", "keyppa-0165013E", "aptconf", "sources.list",
"curtin-dev-ppa.list", "my-repo2.list", "my-repo4.list"])
self.output_files_exist(
["curtin-dev-ubuntu-test-archive-%s.list" % self.release])
def test_keys_imported(self):
"""test_keys_imported - Check if all keys are imported correctly"""
self.check_file_regex("keyid-F430BBA5",
r"Launchpad PPA for Ubuntu Screen Profile")
self.check_file_regex("keylongid-F470A0AC",
r"Ryan Harper")
self.check_file_regex("keyppa-0165013E",
r"Launchpad PPA for curtin developers")
self.check_file_regex("keyraw-8280B242",
r"Christian Ehrhardt")
def test_preserve_source(self):
"""test_preserve_source - no clobbering sources.list by cloud-init"""
self.output_files_exist(["curtin-preserve-sources.cfg"])
# For earlier than xenial 'apt_preserve_sources_list' is expected
self.assertEqual(
{'apt': {'preserve_sources_list': True}},
load_config(self.collect_path("curtin-preserve-sources.cfg")))
def test_source_files(self):
"""test_source_files - Check generated .lists for correct content"""
# hard coded deb lines
self.check_file_strippedline("curtin-dev-ppa.list",
("deb http://ppa.launchpad.net/curtin-dev"
"/test-archive/ubuntu xenial main"))
self.check_file_strippedline("my-repo4.list",
("deb http://ppa.launchpad.net/curtin-dev"
"/test-archive/ubuntu xenial main"))
# mirror and release replacement in deb line
self.check_file_strippedline("my-repo2.list", "deb %s %s multiverse" %
(self.mirror, self.release))
# auto creation by apt-add-repository
self.check_file_regex("curtin-dev-ubuntu-test-archive-%s.list" %
self.release,
(r"http://ppa.launchpad.net/"
r"curtin-dev/test-archive/ubuntu"
r" %s main" % self.release))
def test_ignore_count(self):
"""test_ignore_count - Check for files that should not be created"""
self.check_file_strippedline("ic", "0")
def test_apt_conf(self):
"""test_apt_conf - Check if the selected apt conf was set"""
self.check_file_strippedline("aptconf", 'Acquire::Retries "3";')
class TestAptSrcCustom(TestAptSrcAbs):
"""TestAptSrcNormal - tests valid in the custom sources.list case"""
conf_file = "examples/tests/apt_source_custom.yaml"
def test_custom_source_list(self):
"""test_custom_source_list - Check custom sources with replacement"""
# check that all replacements happened
self.check_file_strippedline("sources.list",
"deb %s %s main restricted" %
(self.mirror, self.release))
self.check_file_strippedline("sources.list",
"deb-src %s %s main restricted" %
(self.mirror, self.release))
self.check_file_strippedline("sources.list",
"deb %s %s universe restricted" %
(self.mirror, self.release))
self.check_file_strippedline("sources.list",
"deb %s %s-security multiverse" %
(self.secmirror, self.release))
# check for something that guarantees us to come from our test
self.check_file_strippedline("sources.list",
"# nice line to check in test")
class TestAptSrcPreserve(TestAptSrcAbs):
"""TestAptSrcPreserve - tests valid in the preserved sources.list case"""
conf_file = "examples/tests/apt_source_preserve.yaml"
boot_cloudconf = None
def test_preserved_source_list(self):
"""test_preserved_source_list - Check sources to be preserved as-is"""
# curtin didn't touch it, so we should find what curtin set as default
self.check_file_regex("sources.list",
r"this file is written by cloud-init")
# overwrite inherited check to match situation here
def test_preserve_source(self):
"""test_preserve_source - check apt_preserve_sources_list not set"""
self.output_files_dont_exist(["curtin-preserve-sources.cfg"])
class TestAptSrcModify(TestAptSrcAbs):
"""TestAptSrcModify - tests modifying sources.list"""
conf_file = "examples/tests/apt_source_modify.yaml"
def test_modified_source_list(self):
"""test_modified_source_list - Check sources with replacement"""
# we set us.archive which is non default, check for that
# this will catch if a target ever changes the expected defaults we
# have to replace in case there is no custom template
self.check_file_regex("sources.list",
r"us.archive.ubuntu.com")
self.check_file_regex("sources.list",
r"security.ubuntu.com")
class TestAptSrcDisablePockets(TestAptSrcAbs):
"""TestAptSrcDisablePockets - tests disabling a suite in sources.list"""
conf_file = "examples/tests/apt_source_modify_disable_suite.yaml"
def test_disabled_suite(self):
"""test_disabled_suite - Check if suites were disabled"""
# two not disabled
self.check_file_regex("sources.list",
r"deb.*us.archive.ubuntu.com")
self.check_file_regex("sources.list",
r"deb.*security.ubuntu.com")
# updates disabled
self.check_file_regex("sources.list",
r"# suite disabled by curtin:.*-updates")
class TestAptSrcModifyArches(TestAptSrcModify):
"""TestAptSrcModify - tests modifying sources.list with per arch mirror"""
# same test, just different yaml to specify the mirrors per arch
conf_file = "examples/tests/apt_source_modify_arches.yaml"
class TestAptSrcSearch(TestAptSrcAbs):
"""TestAptSrcSearch - tests checking a list of mirror options"""
conf_file = "examples/tests/apt_source_search.yaml"
def test_mirror_search(self):
"""test_mirror_search
Check searching through a mirror list
This is checked in the test (late) intentionally.
No matter if resolution worked or failed it shouldn't fail
fatally (python error and trace).
We just can't rely on the content to be found in that case
so we skip the check then."""
res1 = util.is_resolvable_url("http://does.not.exist/ubuntu")
res2 = util.is_resolvable_url("http://does.also.not.exist/ubuntu")
res3 = util.is_resolvable_url("http://us.archive.ubuntu.com/ubuntu")
res4 = util.is_resolvable_url("http://security.ubuntu.com/ubuntu")
if res1 or res2 or not res3 or not res4:
raise SkipTest(("Name resolution not as required"
"(%s, %s, %s, %s)" % (res1, res2, res3, res4)))
self.check_file_regex("sources.list",
r"us.archive.ubuntu.com")
self.check_file_regex("sources.list",
r"security.ubuntu.com")
class XenialTestAptSrcCustom(relbase.xenial, TestAptSrcCustom):
""" XenialTestAptSrcCustom
apt feature Test for Xenial with a custom template
"""
__test__ = True
class XenialTestAptSrcPreserve(relbase.xenial, TestAptSrcPreserve):
""" XenialTestAptSrcPreserve
apt feature Test for Xenial with apt_preserve_sources_list enabled
"""
__test__ = True
class XenialTestAptSrcModify(relbase.xenial, TestAptSrcModify):
""" XenialTestAptSrcModify
apt feature Test for Xenial modifying the sources.list of the image
"""
__test__ = True
class XenialTestAptSrcSearch(relbase.xenial, TestAptSrcSearch):
""" XenialTestAptSrcModify
apt feature Test for Xenial searching for mirrors
"""
__test__ = True
class XenialTestAptSrcModifyArches(relbase.xenial, TestAptSrcModifyArches):
""" XenialTestAptSrcModifyArches
apt feature Test for Xenial checking per arch mirror specification
"""
__test__ = True
class XenialTestAptSrcDisablePockets(relbase.xenial, TestAptSrcDisablePockets):
""" XenialTestAptSrcDisablePockets
apt feature Test for Xenial disabling a suite
"""
__test__ = True
# vi: ts=4 expandtab syntax=python
|
