root / test / py / cfgupgrade_unittest.py @ fa166f67
History | View | Annotate | Download (15.2 kB)
1 | aefbe369 | Michael Hanselmann | #!/usr/bin/python
|
---|---|---|---|
2 | aefbe369 | Michael Hanselmann | #
|
3 | aefbe369 | Michael Hanselmann | |
4 | 1709435e | Bernardo Dal Seno | # Copyright (C) 2010, 2012, 2013 Google Inc.
|
5 | aefbe369 | Michael Hanselmann | #
|
6 | aefbe369 | Michael Hanselmann | # This program is free software; you can redistribute it and/or modify
|
7 | aefbe369 | Michael Hanselmann | # it under the terms of the GNU General Public License as published by
|
8 | aefbe369 | Michael Hanselmann | # the Free Software Foundation; either version 2 of the License, or
|
9 | aefbe369 | Michael Hanselmann | # (at your option) any later version.
|
10 | aefbe369 | Michael Hanselmann | #
|
11 | aefbe369 | Michael Hanselmann | # This program is distributed in the hope that it will be useful, but
|
12 | aefbe369 | Michael Hanselmann | # WITHOUT ANY WARRANTY; without even the implied warranty of
|
13 | aefbe369 | Michael Hanselmann | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
14 | aefbe369 | Michael Hanselmann | # General Public License for more details.
|
15 | aefbe369 | Michael Hanselmann | #
|
16 | aefbe369 | Michael Hanselmann | # You should have received a copy of the GNU General Public License
|
17 | aefbe369 | Michael Hanselmann | # along with this program; if not, write to the Free Software
|
18 | aefbe369 | Michael Hanselmann | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
19 | aefbe369 | Michael Hanselmann | # 02110-1301, USA.
|
20 | aefbe369 | Michael Hanselmann | |
21 | aefbe369 | Michael Hanselmann | |
22 | aefbe369 | Michael Hanselmann | """Script for testing tools/cfgupgrade"""
|
23 | aefbe369 | Michael Hanselmann | |
24 | aefbe369 | Michael Hanselmann | import os |
25 | aefbe369 | Michael Hanselmann | import sys |
26 | aefbe369 | Michael Hanselmann | import unittest |
27 | aefbe369 | Michael Hanselmann | import shutil |
28 | aefbe369 | Michael Hanselmann | import tempfile |
29 | aefbe369 | Michael Hanselmann | import operator |
30 | aefbe369 | Michael Hanselmann | |
31 | aefbe369 | Michael Hanselmann | from ganeti import constants |
32 | aefbe369 | Michael Hanselmann | from ganeti import utils |
33 | aefbe369 | Michael Hanselmann | from ganeti import errors |
34 | aefbe369 | Michael Hanselmann | from ganeti import serializer |
35 | 011974df | Michael Hanselmann | from ganeti import netutils |
36 | aefbe369 | Michael Hanselmann | |
37 | aefbe369 | Michael Hanselmann | import testutils |
38 | aefbe369 | Michael Hanselmann | |
39 | aefbe369 | Michael Hanselmann | |
40 | 1709435e | Bernardo Dal Seno | def _RunUpgrade(path, dry_run, no_verify, ignore_hostname=True, |
41 | 1709435e | Bernardo Dal Seno | downgrade=False):
|
42 | aefbe369 | Michael Hanselmann | cmd = [sys.executable, "%s/tools/cfgupgrade" % testutils.GetSourceDir(),
|
43 | 7939f60c | Michael Hanselmann | "--debug", "--force", "--path=%s" % path, "--confdir=%s" % path] |
44 | 011974df | Michael Hanselmann | |
45 | 011974df | Michael Hanselmann | if ignore_hostname:
|
46 | 011974df | Michael Hanselmann | cmd.append("--ignore-hostname")
|
47 | aefbe369 | Michael Hanselmann | if dry_run:
|
48 | aefbe369 | Michael Hanselmann | cmd.append("--dry-run")
|
49 | aefbe369 | Michael Hanselmann | if no_verify:
|
50 | aefbe369 | Michael Hanselmann | cmd.append("--no-verify")
|
51 | 1709435e | Bernardo Dal Seno | if downgrade:
|
52 | 1709435e | Bernardo Dal Seno | cmd.append("--downgrade")
|
53 | aefbe369 | Michael Hanselmann | |
54 | aefbe369 | Michael Hanselmann | result = utils.RunCmd(cmd, cwd=os.getcwd()) |
55 | aefbe369 | Michael Hanselmann | if result.failed:
|
56 | aefbe369 | Michael Hanselmann | raise Exception("cfgupgrade failed: %s, output %r" % |
57 | aefbe369 | Michael Hanselmann | (result.fail_reason, result.output)) |
58 | aefbe369 | Michael Hanselmann | |
59 | aefbe369 | Michael Hanselmann | |
60 | aefbe369 | Michael Hanselmann | class TestCfgupgrade(unittest.TestCase): |
61 | aefbe369 | Michael Hanselmann | def setUp(self): |
62 | aefbe369 | Michael Hanselmann | self.tmpdir = tempfile.mkdtemp()
|
63 | aefbe369 | Michael Hanselmann | |
64 | aefbe369 | Michael Hanselmann | self.config_path = utils.PathJoin(self.tmpdir, "config.data") |
65 | aefbe369 | Michael Hanselmann | self.noded_cert_path = utils.PathJoin(self.tmpdir, "server.pem") |
66 | aefbe369 | Michael Hanselmann | self.rapi_cert_path = utils.PathJoin(self.tmpdir, "rapi.pem") |
67 | fdd9ac5b | Michael Hanselmann | self.rapi_users_path = utils.PathJoin(self.tmpdir, "rapi", "users") |
68 | fdd9ac5b | Michael Hanselmann | self.rapi_users_path_pre24 = utils.PathJoin(self.tmpdir, "rapi_users") |
69 | aefbe369 | Michael Hanselmann | self.known_hosts_path = utils.PathJoin(self.tmpdir, "known_hosts") |
70 | aefbe369 | Michael Hanselmann | self.confd_hmac_path = utils.PathJoin(self.tmpdir, "hmac.key") |
71 | aefbe369 | Michael Hanselmann | self.cds_path = utils.PathJoin(self.tmpdir, "cluster-domain-secret") |
72 | 011974df | Michael Hanselmann | self.ss_master_node_path = utils.PathJoin(self.tmpdir, "ssconf_master_node") |
73 | 7939f60c | Michael Hanselmann | self.file_storage_paths = utils.PathJoin(self.tmpdir, "file-storage-paths") |
74 | aefbe369 | Michael Hanselmann | |
75 | aefbe369 | Michael Hanselmann | def tearDown(self): |
76 | aefbe369 | Michael Hanselmann | shutil.rmtree(self.tmpdir)
|
77 | aefbe369 | Michael Hanselmann | |
78 | aefbe369 | Michael Hanselmann | def _LoadConfig(self): |
79 | aefbe369 | Michael Hanselmann | return serializer.LoadJson(utils.ReadFile(self.config_path)) |
80 | aefbe369 | Michael Hanselmann | |
81 | aefbe369 | Michael Hanselmann | def _CreateValidConfigDir(self): |
82 | aefbe369 | Michael Hanselmann | utils.WriteFile(self.noded_cert_path, data="") |
83 | aefbe369 | Michael Hanselmann | utils.WriteFile(self.known_hosts_path, data="") |
84 | 011974df | Michael Hanselmann | utils.WriteFile(self.ss_master_node_path,
|
85 | 011974df | Michael Hanselmann | data="node.has.another.name.example.net")
|
86 | aefbe369 | Michael Hanselmann | |
87 | aefbe369 | Michael Hanselmann | def testNoConfigDir(self): |
88 | aefbe369 | Michael Hanselmann | self.assertFalse(utils.ListVisibleFiles(self.tmpdir)) |
89 | aefbe369 | Michael Hanselmann | self.assertRaises(Exception, _RunUpgrade, self.tmpdir, False, True) |
90 | aefbe369 | Michael Hanselmann | self.assertRaises(Exception, _RunUpgrade, self.tmpdir, True, True) |
91 | aefbe369 | Michael Hanselmann | |
92 | 011974df | Michael Hanselmann | def testWrongHostname(self): |
93 | 011974df | Michael Hanselmann | self._CreateValidConfigDir()
|
94 | 011974df | Michael Hanselmann | |
95 | 011974df | Michael Hanselmann | utils.WriteFile(self.config_path, data=serializer.DumpJson({
|
96 | 011974df | Michael Hanselmann | "version": constants.CONFIG_VERSION,
|
97 | 011974df | Michael Hanselmann | "cluster": {},
|
98 | a19d8cd5 | Iustin Pop | "instances": {},
|
99 | 6e8091f9 | Dimitris Aragiorgis | "nodegroups": {},
|
100 | 011974df | Michael Hanselmann | })) |
101 | 011974df | Michael Hanselmann | |
102 | 011974df | Michael Hanselmann | hostname = netutils.GetHostname().name |
103 | 011974df | Michael Hanselmann | assert hostname != utils.ReadOneLineFile(self.ss_master_node_path) |
104 | 011974df | Michael Hanselmann | |
105 | 011974df | Michael Hanselmann | self.assertRaises(Exception, _RunUpgrade, self.tmpdir, False, True, |
106 | 011974df | Michael Hanselmann | ignore_hostname=False)
|
107 | 011974df | Michael Hanselmann | |
108 | 011974df | Michael Hanselmann | def testCorrectHostname(self): |
109 | 011974df | Michael Hanselmann | self._CreateValidConfigDir()
|
110 | 011974df | Michael Hanselmann | |
111 | 011974df | Michael Hanselmann | utils.WriteFile(self.config_path, data=serializer.DumpJson({
|
112 | 011974df | Michael Hanselmann | "version": constants.CONFIG_VERSION,
|
113 | 011974df | Michael Hanselmann | "cluster": {},
|
114 | a19d8cd5 | Iustin Pop | "instances": {},
|
115 | 6e8091f9 | Dimitris Aragiorgis | "nodegroups": {},
|
116 | 011974df | Michael Hanselmann | })) |
117 | 011974df | Michael Hanselmann | |
118 | 011974df | Michael Hanselmann | utils.WriteFile(self.ss_master_node_path,
|
119 | 011974df | Michael Hanselmann | data="%s\n" % netutils.GetHostname().name)
|
120 | 011974df | Michael Hanselmann | |
121 | 011974df | Michael Hanselmann | _RunUpgrade(self.tmpdir, False, True, ignore_hostname=False) |
122 | 011974df | Michael Hanselmann | |
123 | aefbe369 | Michael Hanselmann | def testInconsistentConfig(self): |
124 | aefbe369 | Michael Hanselmann | self._CreateValidConfigDir()
|
125 | aefbe369 | Michael Hanselmann | # There should be no "config_version"
|
126 | aefbe369 | Michael Hanselmann | cfg = { |
127 | aefbe369 | Michael Hanselmann | "version": 0, |
128 | aefbe369 | Michael Hanselmann | "cluster": {
|
129 | aefbe369 | Michael Hanselmann | "config_version": 0, |
130 | aefbe369 | Michael Hanselmann | }, |
131 | a19d8cd5 | Iustin Pop | "instances": {},
|
132 | 6e8091f9 | Dimitris Aragiorgis | "nodegroups": {},
|
133 | aefbe369 | Michael Hanselmann | } |
134 | aefbe369 | Michael Hanselmann | utils.WriteFile(self.config_path, data=serializer.DumpJson(cfg))
|
135 | aefbe369 | Michael Hanselmann | self.assertRaises(Exception, _RunUpgrade, self.tmpdir, False, True) |
136 | aefbe369 | Michael Hanselmann | |
137 | aefbe369 | Michael Hanselmann | def testInvalidConfig(self): |
138 | aefbe369 | Michael Hanselmann | self._CreateValidConfigDir()
|
139 | aefbe369 | Michael Hanselmann | # Missing version from config
|
140 | aefbe369 | Michael Hanselmann | utils.WriteFile(self.config_path, data=serializer.DumpJson({}))
|
141 | aefbe369 | Michael Hanselmann | self.assertRaises(Exception, _RunUpgrade, self.tmpdir, False, True) |
142 | aefbe369 | Michael Hanselmann | |
143 | 7939f60c | Michael Hanselmann | def _TestSimpleUpgrade(self, from_version, dry_run, |
144 | 7939f60c | Michael Hanselmann | file_storage_dir=None,
|
145 | 7939f60c | Michael Hanselmann | shared_file_storage_dir=None):
|
146 | 7939f60c | Michael Hanselmann | cluster = {} |
147 | 7939f60c | Michael Hanselmann | |
148 | 7939f60c | Michael Hanselmann | if file_storage_dir:
|
149 | 7939f60c | Michael Hanselmann | cluster["file_storage_dir"] = file_storage_dir
|
150 | 7939f60c | Michael Hanselmann | if shared_file_storage_dir:
|
151 | 7939f60c | Michael Hanselmann | cluster["shared_file_storage_dir"] = shared_file_storage_dir
|
152 | 7939f60c | Michael Hanselmann | |
153 | aefbe369 | Michael Hanselmann | cfg = { |
154 | aefbe369 | Michael Hanselmann | "version": from_version,
|
155 | 7939f60c | Michael Hanselmann | "cluster": cluster,
|
156 | a19d8cd5 | Iustin Pop | "instances": {},
|
157 | 6e8091f9 | Dimitris Aragiorgis | "nodegroups": {},
|
158 | aefbe369 | Michael Hanselmann | } |
159 | fa166f67 | Bernardo Dal Seno | self._TestUpgradeFromData(cfg, dry_run)
|
160 | fa166f67 | Bernardo Dal Seno | |
161 | fa166f67 | Bernardo Dal Seno | def _TestUpgradeFromData(self, cfg, dry_run): |
162 | fa166f67 | Bernardo Dal Seno | assert "version" in cfg |
163 | fa166f67 | Bernardo Dal Seno | from_version = cfg["version"]
|
164 | aefbe369 | Michael Hanselmann | self._CreateValidConfigDir()
|
165 | aefbe369 | Michael Hanselmann | utils.WriteFile(self.config_path, data=serializer.DumpJson(cfg))
|
166 | aefbe369 | Michael Hanselmann | |
167 | aefbe369 | Michael Hanselmann | self.assertFalse(os.path.isfile(self.rapi_cert_path)) |
168 | aefbe369 | Michael Hanselmann | self.assertFalse(os.path.isfile(self.confd_hmac_path)) |
169 | aefbe369 | Michael Hanselmann | self.assertFalse(os.path.isfile(self.cds_path)) |
170 | aefbe369 | Michael Hanselmann | |
171 | aefbe369 | Michael Hanselmann | _RunUpgrade(self.tmpdir, dry_run, True) |
172 | aefbe369 | Michael Hanselmann | |
173 | aefbe369 | Michael Hanselmann | if dry_run:
|
174 | aefbe369 | Michael Hanselmann | expversion = from_version |
175 | aefbe369 | Michael Hanselmann | checkfn = operator.not_ |
176 | aefbe369 | Michael Hanselmann | else:
|
177 | aefbe369 | Michael Hanselmann | expversion = constants.CONFIG_VERSION |
178 | aefbe369 | Michael Hanselmann | checkfn = operator.truth |
179 | aefbe369 | Michael Hanselmann | |
180 | aefbe369 | Michael Hanselmann | self.assert_(checkfn(os.path.isfile(self.rapi_cert_path))) |
181 | aefbe369 | Michael Hanselmann | self.assert_(checkfn(os.path.isfile(self.confd_hmac_path))) |
182 | aefbe369 | Michael Hanselmann | self.assert_(checkfn(os.path.isfile(self.cds_path))) |
183 | aefbe369 | Michael Hanselmann | |
184 | aefbe369 | Michael Hanselmann | newcfg = self._LoadConfig()
|
185 | aefbe369 | Michael Hanselmann | self.assertEqual(newcfg["version"], expversion) |
186 | aefbe369 | Michael Hanselmann | |
187 | fdd9ac5b | Michael Hanselmann | def testRapiUsers(self): |
188 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
189 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
190 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(os.path.dirname(self.rapi_users_path))) |
191 | fdd9ac5b | Michael Hanselmann | |
192 | fdd9ac5b | Michael Hanselmann | utils.WriteFile(self.rapi_users_path_pre24, data="some user\n") |
193 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), False) |
194 | fdd9ac5b | Michael Hanselmann | |
195 | 87c80992 | Michael Hanselmann | self.assertTrue(os.path.isdir(os.path.dirname(self.rapi_users_path))) |
196 | fdd9ac5b | Michael Hanselmann | self.assert_(os.path.islink(self.rapi_users_path_pre24)) |
197 | fdd9ac5b | Michael Hanselmann | self.assert_(os.path.isfile(self.rapi_users_path)) |
198 | 87c80992 | Michael Hanselmann | self.assertEqual(os.readlink(self.rapi_users_path_pre24), |
199 | 87c80992 | Michael Hanselmann | self.rapi_users_path)
|
200 | fdd9ac5b | Michael Hanselmann | for path in [self.rapi_users_path, self.rapi_users_path_pre24]: |
201 | fdd9ac5b | Michael Hanselmann | self.assertEqual(utils.ReadFile(path), "some user\n") |
202 | fdd9ac5b | Michael Hanselmann | |
203 | fdd9ac5b | Michael Hanselmann | def testRapiUsers24AndAbove(self): |
204 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
205 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
206 | fdd9ac5b | Michael Hanselmann | |
207 | fdd9ac5b | Michael Hanselmann | os.mkdir(os.path.dirname(self.rapi_users_path))
|
208 | fdd9ac5b | Michael Hanselmann | utils.WriteFile(self.rapi_users_path, data="other user\n") |
209 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), False) |
210 | fdd9ac5b | Michael Hanselmann | |
211 | fdd9ac5b | Michael Hanselmann | self.assert_(os.path.islink(self.rapi_users_path_pre24)) |
212 | fdd9ac5b | Michael Hanselmann | self.assert_(os.path.isfile(self.rapi_users_path)) |
213 | 87c80992 | Michael Hanselmann | self.assertEqual(os.readlink(self.rapi_users_path_pre24), |
214 | 87c80992 | Michael Hanselmann | self.rapi_users_path)
|
215 | fdd9ac5b | Michael Hanselmann | for path in [self.rapi_users_path, self.rapi_users_path_pre24]: |
216 | fdd9ac5b | Michael Hanselmann | self.assertEqual(utils.ReadFile(path), "other user\n") |
217 | fdd9ac5b | Michael Hanselmann | |
218 | fdd9ac5b | Michael Hanselmann | def testRapiUsersExistingSymlink(self): |
219 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
220 | fdd9ac5b | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
221 | fdd9ac5b | Michael Hanselmann | |
222 | 87c80992 | Michael Hanselmann | os.mkdir(os.path.dirname(self.rapi_users_path))
|
223 | fdd9ac5b | Michael Hanselmann | os.symlink(self.rapi_users_path, self.rapi_users_path_pre24) |
224 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path, data="hello world\n") |
225 | fdd9ac5b | Michael Hanselmann | |
226 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 2, 0), False) |
227 | fdd9ac5b | Michael Hanselmann | |
228 | 87c80992 | Michael Hanselmann | self.assert_(os.path.isfile(self.rapi_users_path) and |
229 | 87c80992 | Michael Hanselmann | not os.path.islink(self.rapi_users_path)) |
230 | fdd9ac5b | Michael Hanselmann | self.assert_(os.path.islink(self.rapi_users_path_pre24)) |
231 | 87c80992 | Michael Hanselmann | self.assertEqual(os.readlink(self.rapi_users_path_pre24), |
232 | 87c80992 | Michael Hanselmann | self.rapi_users_path)
|
233 | 87c80992 | Michael Hanselmann | for path in [self.rapi_users_path, self.rapi_users_path_pre24]: |
234 | 87c80992 | Michael Hanselmann | self.assertEqual(utils.ReadFile(path), "hello world\n") |
235 | 87c80992 | Michael Hanselmann | |
236 | 87c80992 | Michael Hanselmann | def testRapiUsersExistingTarget(self): |
237 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
238 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
239 | 87c80992 | Michael Hanselmann | |
240 | 87c80992 | Michael Hanselmann | os.mkdir(os.path.dirname(self.rapi_users_path))
|
241 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path, data="other user\n") |
242 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path_pre24, data="hello world\n") |
243 | 87c80992 | Michael Hanselmann | |
244 | 87c80992 | Michael Hanselmann | self.assertRaises(Exception, self._TestSimpleUpgrade, |
245 | 87c80992 | Michael Hanselmann | constants.BuildVersion(2, 2, 0), False) |
246 | 87c80992 | Michael Hanselmann | |
247 | 87c80992 | Michael Hanselmann | for path in [self.rapi_users_path, self.rapi_users_path_pre24]: |
248 | 87c80992 | Michael Hanselmann | self.assert_(os.path.isfile(path) and not os.path.islink(path)) |
249 | 87c80992 | Michael Hanselmann | self.assertEqual(utils.ReadFile(self.rapi_users_path), "other user\n") |
250 | 87c80992 | Michael Hanselmann | self.assertEqual(utils.ReadFile(self.rapi_users_path_pre24), |
251 | 87c80992 | Michael Hanselmann | "hello world\n")
|
252 | 87c80992 | Michael Hanselmann | |
253 | 87c80992 | Michael Hanselmann | def testRapiUsersDryRun(self): |
254 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
255 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
256 | 87c80992 | Michael Hanselmann | |
257 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path_pre24, data="some user\n") |
258 | 87c80992 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), True) |
259 | 87c80992 | Michael Hanselmann | |
260 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.isdir(os.path.dirname(self.rapi_users_path))) |
261 | 87c80992 | Michael Hanselmann | self.assertTrue(os.path.isfile(self.rapi_users_path_pre24) and |
262 | 87c80992 | Michael Hanselmann | not os.path.islink(self.rapi_users_path_pre24)) |
263 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
264 | 87c80992 | Michael Hanselmann | |
265 | 87c80992 | Michael Hanselmann | def testRapiUsers24AndAboveDryRun(self): |
266 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
267 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
268 | 87c80992 | Michael Hanselmann | |
269 | 87c80992 | Michael Hanselmann | os.mkdir(os.path.dirname(self.rapi_users_path))
|
270 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path, data="other user\n") |
271 | 87c80992 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), True) |
272 | 87c80992 | Michael Hanselmann | |
273 | 87c80992 | Michael Hanselmann | self.assertTrue(os.path.isfile(self.rapi_users_path) and |
274 | 87c80992 | Michael Hanselmann | not os.path.islink(self.rapi_users_path)) |
275 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
276 | 87c80992 | Michael Hanselmann | self.assertEqual(utils.ReadFile(self.rapi_users_path), "other user\n") |
277 | 87c80992 | Michael Hanselmann | |
278 | 87c80992 | Michael Hanselmann | def testRapiUsersExistingSymlinkDryRun(self): |
279 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path)) |
280 | 87c80992 | Michael Hanselmann | self.assertFalse(os.path.exists(self.rapi_users_path_pre24)) |
281 | 87c80992 | Michael Hanselmann | |
282 | 87c80992 | Michael Hanselmann | os.mkdir(os.path.dirname(self.rapi_users_path))
|
283 | 87c80992 | Michael Hanselmann | os.symlink(self.rapi_users_path, self.rapi_users_path_pre24) |
284 | 87c80992 | Michael Hanselmann | utils.WriteFile(self.rapi_users_path, data="hello world\n") |
285 | 87c80992 | Michael Hanselmann | |
286 | 87c80992 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 2, 0), True) |
287 | 87c80992 | Michael Hanselmann | |
288 | 87c80992 | Michael Hanselmann | self.assertTrue(os.path.islink(self.rapi_users_path_pre24)) |
289 | 87c80992 | Michael Hanselmann | self.assertTrue(os.path.isfile(self.rapi_users_path) and |
290 | 87c80992 | Michael Hanselmann | not os.path.islink(self.rapi_users_path)) |
291 | 87c80992 | Michael Hanselmann | self.assertEqual(os.readlink(self.rapi_users_path_pre24), |
292 | 87c80992 | Michael Hanselmann | self.rapi_users_path)
|
293 | fdd9ac5b | Michael Hanselmann | for path in [self.rapi_users_path, self.rapi_users_path_pre24]: |
294 | fdd9ac5b | Michael Hanselmann | self.assertEqual(utils.ReadFile(path), "hello world\n") |
295 | fdd9ac5b | Michael Hanselmann | |
296 | 7939f60c | Michael Hanselmann | def testFileStoragePathsDryRun(self): |
297 | 7939f60c | Michael Hanselmann | self.assertFalse(os.path.exists(self.file_storage_paths)) |
298 | 7939f60c | Michael Hanselmann | |
299 | 7939f60c | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 6, 0), True, |
300 | 7939f60c | Michael Hanselmann | file_storage_dir=self.tmpdir,
|
301 | 7939f60c | Michael Hanselmann | shared_file_storage_dir="/tmp")
|
302 | 7939f60c | Michael Hanselmann | |
303 | 7939f60c | Michael Hanselmann | self.assertFalse(os.path.exists(self.file_storage_paths)) |
304 | 7939f60c | Michael Hanselmann | |
305 | 7939f60c | Michael Hanselmann | def testFileStoragePathsBoth(self): |
306 | 7939f60c | Michael Hanselmann | self.assertFalse(os.path.exists(self.file_storage_paths)) |
307 | 7939f60c | Michael Hanselmann | |
308 | 7939f60c | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 6, 0), False, |
309 | 7939f60c | Michael Hanselmann | file_storage_dir=self.tmpdir,
|
310 | 7939f60c | Michael Hanselmann | shared_file_storage_dir="/tmp")
|
311 | 7939f60c | Michael Hanselmann | |
312 | 7939f60c | Michael Hanselmann | lines = utils.ReadFile(self.file_storage_paths).splitlines()
|
313 | 7939f60c | Michael Hanselmann | self.assertTrue(lines.pop(0).startswith("# ")) |
314 | 7939f60c | Michael Hanselmann | self.assertTrue(lines.pop(0).startswith("# cfgupgrade")) |
315 | 7939f60c | Michael Hanselmann | self.assertEqual(lines.pop(0), self.tmpdir) |
316 | 7939f60c | Michael Hanselmann | self.assertEqual(lines.pop(0), "/tmp") |
317 | 7939f60c | Michael Hanselmann | self.assertFalse(lines)
|
318 | 7939f60c | Michael Hanselmann | self.assertEqual(os.stat(self.file_storage_paths).st_mode & 0777, |
319 | 7939f60c | Michael Hanselmann | 0600, msg="Wrong permissions") |
320 | 7939f60c | Michael Hanselmann | |
321 | 7939f60c | Michael Hanselmann | def testFileStoragePathsSharedOnly(self): |
322 | 7939f60c | Michael Hanselmann | self.assertFalse(os.path.exists(self.file_storage_paths)) |
323 | 7939f60c | Michael Hanselmann | |
324 | 7939f60c | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 5, 0), False, |
325 | 7939f60c | Michael Hanselmann | file_storage_dir=None,
|
326 | 7939f60c | Michael Hanselmann | shared_file_storage_dir=self.tmpdir)
|
327 | 7939f60c | Michael Hanselmann | |
328 | 7939f60c | Michael Hanselmann | lines = utils.ReadFile(self.file_storage_paths).splitlines()
|
329 | 7939f60c | Michael Hanselmann | self.assertTrue(lines.pop(0).startswith("# ")) |
330 | 7939f60c | Michael Hanselmann | self.assertTrue(lines.pop(0).startswith("# cfgupgrade")) |
331 | 7939f60c | Michael Hanselmann | self.assertEqual(lines.pop(0), self.tmpdir) |
332 | 7939f60c | Michael Hanselmann | self.assertFalse(lines)
|
333 | 7939f60c | Michael Hanselmann | |
334 | aefbe369 | Michael Hanselmann | def testUpgradeFrom_2_0(self): |
335 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 0, 0), False) |
336 | aefbe369 | Michael Hanselmann | |
337 | aefbe369 | Michael Hanselmann | def testUpgradeFrom_2_1(self): |
338 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 1, 0), False) |
339 | aefbe369 | Michael Hanselmann | |
340 | fdd9ac5b | Michael Hanselmann | def testUpgradeFrom_2_2(self): |
341 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 2, 0), False) |
342 | fdd9ac5b | Michael Hanselmann | |
343 | fdd9ac5b | Michael Hanselmann | def testUpgradeFrom_2_3(self): |
344 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), False) |
345 | fdd9ac5b | Michael Hanselmann | |
346 | a91f69c4 | Michael Hanselmann | def testUpgradeFrom_2_4(self): |
347 | a91f69c4 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 4, 0), False) |
348 | a91f69c4 | Michael Hanselmann | |
349 | 9f604ab8 | Michael Hanselmann | def testUpgradeFrom_2_5(self): |
350 | 9f604ab8 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 5, 0), False) |
351 | 9f604ab8 | Michael Hanselmann | |
352 | 93fd9bb1 | Iustin Pop | def testUpgradeFrom_2_6(self): |
353 | 93fd9bb1 | Iustin Pop | self._TestSimpleUpgrade(constants.BuildVersion(2, 6, 0), False) |
354 | 93fd9bb1 | Iustin Pop | |
355 | aefbe369 | Michael Hanselmann | def testUpgradeCurrent(self): |
356 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.CONFIG_VERSION, False) |
357 | aefbe369 | Michael Hanselmann | |
358 | fa166f67 | Bernardo Dal Seno | def _RunDowngradeUpgrade(self): |
359 | 1709435e | Bernardo Dal Seno | oldconf = self._LoadConfig()
|
360 | 1709435e | Bernardo Dal Seno | _RunUpgrade(self.tmpdir, False, True, downgrade=True) |
361 | 1709435e | Bernardo Dal Seno | _RunUpgrade(self.tmpdir, False, True) |
362 | 1709435e | Bernardo Dal Seno | newconf = self._LoadConfig()
|
363 | 1709435e | Bernardo Dal Seno | self.assertEqual(oldconf, newconf)
|
364 | 1709435e | Bernardo Dal Seno | |
365 | fa166f67 | Bernardo Dal Seno | def testDowngrade(self): |
366 | 1709435e | Bernardo Dal Seno | self._TestSimpleUpgrade(constants.CONFIG_VERSION, False) |
367 | fa166f67 | Bernardo Dal Seno | self._RunDowngradeUpgrade()
|
368 | fa166f67 | Bernardo Dal Seno | |
369 | fa166f67 | Bernardo Dal Seno | def _RunDowngradeTwice(self): |
370 | fa166f67 | Bernardo Dal Seno | """Make sure that downgrade is idempotent."""
|
371 | 1709435e | Bernardo Dal Seno | _RunUpgrade(self.tmpdir, False, True, downgrade=True) |
372 | 1709435e | Bernardo Dal Seno | oldconf = self._LoadConfig()
|
373 | 1709435e | Bernardo Dal Seno | _RunUpgrade(self.tmpdir, False, True, downgrade=True) |
374 | 1709435e | Bernardo Dal Seno | newconf = self._LoadConfig()
|
375 | 1709435e | Bernardo Dal Seno | self.assertEqual(oldconf, newconf)
|
376 | 1709435e | Bernardo Dal Seno | |
377 | fa166f67 | Bernardo Dal Seno | def testDowngradeTwice(self): |
378 | fa166f67 | Bernardo Dal Seno | self._TestSimpleUpgrade(constants.CONFIG_VERSION, False) |
379 | fa166f67 | Bernardo Dal Seno | self._RunDowngradeTwice()
|
380 | fa166f67 | Bernardo Dal Seno | |
381 | aefbe369 | Michael Hanselmann | def testUpgradeDryRunFrom_2_0(self): |
382 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 0, 0), True) |
383 | aefbe369 | Michael Hanselmann | |
384 | aefbe369 | Michael Hanselmann | def testUpgradeDryRunFrom_2_1(self): |
385 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 1, 0), True) |
386 | aefbe369 | Michael Hanselmann | |
387 | fdd9ac5b | Michael Hanselmann | def testUpgradeDryRunFrom_2_2(self): |
388 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 2, 0), True) |
389 | fdd9ac5b | Michael Hanselmann | |
390 | fdd9ac5b | Michael Hanselmann | def testUpgradeDryRunFrom_2_3(self): |
391 | fdd9ac5b | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 3, 0), True) |
392 | fdd9ac5b | Michael Hanselmann | |
393 | a91f69c4 | Michael Hanselmann | def testUpgradeDryRunFrom_2_4(self): |
394 | a91f69c4 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 4, 0), True) |
395 | a91f69c4 | Michael Hanselmann | |
396 | 9f604ab8 | Michael Hanselmann | def testUpgradeDryRunFrom_2_5(self): |
397 | 9f604ab8 | Michael Hanselmann | self._TestSimpleUpgrade(constants.BuildVersion(2, 5, 0), True) |
398 | 9f604ab8 | Michael Hanselmann | |
399 | 93fd9bb1 | Iustin Pop | def testUpgradeDryRunFrom_2_6(self): |
400 | 93fd9bb1 | Iustin Pop | self._TestSimpleUpgrade(constants.BuildVersion(2, 6, 0), True) |
401 | 93fd9bb1 | Iustin Pop | |
402 | aefbe369 | Michael Hanselmann | def testUpgradeCurrentDryRun(self): |
403 | aefbe369 | Michael Hanselmann | self._TestSimpleUpgrade(constants.CONFIG_VERSION, True) |
404 | aefbe369 | Michael Hanselmann | |
405 | 1709435e | Bernardo Dal Seno | def testDowngradeDryRun(self): |
406 | 1709435e | Bernardo Dal Seno | self._TestSimpleUpgrade(constants.CONFIG_VERSION, False) |
407 | 1709435e | Bernardo Dal Seno | oldconf = self._LoadConfig()
|
408 | 1709435e | Bernardo Dal Seno | _RunUpgrade(self.tmpdir, True, True, downgrade=True) |
409 | 1709435e | Bernardo Dal Seno | newconf = self._LoadConfig()
|
410 | 1709435e | Bernardo Dal Seno | self.assertEqual(oldconf["version"], newconf["version"]) |
411 | aefbe369 | Michael Hanselmann | |
412 | aefbe369 | Michael Hanselmann | if __name__ == "__main__": |
413 | aefbe369 | Michael Hanselmann | testutils.GanetiTestProgram() |