root / test / py / ganeti.ovf_unittest.py @ 91c17910
History | View | Annotate | Download (30.1 kB)
1 |
#!/usr/bin/python
|
---|---|
2 |
#
|
3 |
|
4 |
# Copyright (C) 2011, 2012 Google Inc.
|
5 |
#
|
6 |
# This program is free software; you can redistribute it and/or modify
|
7 |
# it under the terms of the GNU General Public License as published by
|
8 |
# the Free Software Foundation; either version 2 of the License, or
|
9 |
# (at your option) any later version.
|
10 |
#
|
11 |
# This program is distributed in the hope that it will be useful, but
|
12 |
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
13 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
14 |
# General Public License for more details.
|
15 |
#
|
16 |
# You should have received a copy of the GNU General Public License
|
17 |
# along with this program; if not, write to the Free Software
|
18 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
19 |
# 02110-1301, USA.
|
20 |
|
21 |
|
22 |
"""Script for testing ganeti.ovf.
|
23 |
|
24 |
"""
|
25 |
|
26 |
import optparse |
27 |
import os |
28 |
import os.path |
29 |
import re |
30 |
import shutil |
31 |
import sys |
32 |
import tempfile |
33 |
import unittest |
34 |
|
35 |
try:
|
36 |
import xml.etree.ElementTree as ET |
37 |
except ImportError: |
38 |
import elementtree.ElementTree as ET |
39 |
|
40 |
from ganeti import constants |
41 |
from ganeti import errors |
42 |
from ganeti import ovf |
43 |
from ganeti import utils |
44 |
from ganeti import pathutils |
45 |
|
46 |
import testutils |
47 |
|
48 |
OUTPUT_DIR = "newdir"
|
49 |
|
50 |
GANETI_DISKS = { |
51 |
"disk_count": "1", |
52 |
"disk0_dump": "new_disk.raw", |
53 |
"disk0_size": "0", |
54 |
"disk0_ivname": "disk/0", |
55 |
} |
56 |
GANETI_NETWORKS = { |
57 |
"nic_count": "1", |
58 |
"nic0_mode": "bridged", |
59 |
"nic0_ip": "none", |
60 |
"nic0_mac": "aa:00:00:d8:2c:1e", |
61 |
"nic0_link": "xen-br0", |
62 |
"nic0_network": "auto", |
63 |
} |
64 |
GANETI_HYPERVISOR = { |
65 |
"hypervisor_name": "xen-pvm", |
66 |
"root-path": "/dev/sda", |
67 |
"kernel_args": "ro", |
68 |
} |
69 |
GANETI_OS = {"os_name": "lenny-image"} |
70 |
GANETI_BACKEND = { |
71 |
"vcpus": "1", |
72 |
"memory" : "2048", |
73 |
"auto_balance": "False", |
74 |
} |
75 |
GANETI_NAME = "ganeti-test-xen"
|
76 |
GANETI_TEMPLATE = "plain"
|
77 |
GANETI_TAGS = None
|
78 |
GANETI_VERSION = "0"
|
79 |
|
80 |
VIRTUALBOX_DISKS = { |
81 |
"disk_count": "2", |
82 |
"disk0_ivname": "disk/0", |
83 |
"disk0_dump": "new_disk.raw", |
84 |
"disk0_size": "0", |
85 |
"disk1_ivname": "disk/1", |
86 |
"disk1_dump": "second_disk.raw", |
87 |
"disk1_size": "0", |
88 |
} |
89 |
VIRTUALBOX_NETWORKS = { |
90 |
"nic_count": "1", |
91 |
"nic0_mode": "bridged", |
92 |
"nic0_ip": "none", |
93 |
"nic0_link": "auto", |
94 |
"nic0_mac": "auto", |
95 |
"nic0_network": "auto", |
96 |
} |
97 |
VIRTUALBOX_HYPERVISOR = {"hypervisor_name": "auto"} |
98 |
VIRTUALBOX_OS = {"os_name": None} |
99 |
VIRTUALBOX_BACKEND = { |
100 |
"vcpus": "1", |
101 |
"memory" : "2048", |
102 |
"auto_balance": "auto", |
103 |
} |
104 |
VIRTUALBOX_NAME = None
|
105 |
VIRTUALBOX_TEMPLATE = None
|
106 |
VIRTUALBOX_TAGS = None
|
107 |
VIRTUALBOX_VERSION = None
|
108 |
|
109 |
EMPTY_DISKS = {} |
110 |
EMPTY_NETWORKS = {} |
111 |
EMPTY_HYPERVISOR = {"hypervisor_name": "auto"} |
112 |
EMPTY_OS = {} |
113 |
EMPTY_BACKEND = { |
114 |
"vcpus": "auto", |
115 |
"memory" : "auto", |
116 |
"auto_balance": "auto", |
117 |
} |
118 |
EMPTY_NAME = None
|
119 |
EMPTY_TEMPLATE = None
|
120 |
EMPTY_TAGS = None
|
121 |
EMPTY_VERSION = None
|
122 |
|
123 |
CMDARGS_DISKS = { |
124 |
"disk_count": "1", |
125 |
"disk0_ivname": "disk/0", |
126 |
"disk0_dump": "disk0.raw", |
127 |
"disk0_size": "8", |
128 |
} |
129 |
CMDARGS_NETWORKS = { |
130 |
"nic0_link": "auto", |
131 |
"nic0_mode": "bridged", |
132 |
"nic0_ip": "none", |
133 |
"nic0_mac": "auto", |
134 |
"nic_count": "1", |
135 |
"nic0_network": "auto", |
136 |
} |
137 |
CMDARGS_HYPERVISOR = { |
138 |
"hypervisor_name": "xen-pvm" |
139 |
} |
140 |
CMDARGS_OS = {"os_name": "lenny-image"} |
141 |
CMDARGS_BACKEND = { |
142 |
"auto_balance": False, |
143 |
"vcpus": "1", |
144 |
"memory": "256", |
145 |
} |
146 |
CMDARGS_NAME = "test-instance"
|
147 |
CMDARGS_TEMPLATE = "plain"
|
148 |
CMDARGS_TAGS = "test-tag-1,test-tag-2"
|
149 |
|
150 |
ARGS_EMPTY = { |
151 |
"output_dir": None, |
152 |
"nics": [],
|
153 |
"disks": [],
|
154 |
"name": "test-instance", |
155 |
"ova_package": False, |
156 |
"ext_usage": False, |
157 |
"disk_format": "cow", |
158 |
"compression": False, |
159 |
} |
160 |
ARGS_EXPORT_DIR = dict(ARGS_EMPTY, **{
|
161 |
"output_dir": OUTPUT_DIR,
|
162 |
"name": None, |
163 |
"hypervisor": None, |
164 |
"os": None, |
165 |
"beparams": {},
|
166 |
"no_nics": False, |
167 |
"disk_template": None, |
168 |
"tags": None, |
169 |
}) |
170 |
ARGS_VBOX = dict(ARGS_EXPORT_DIR, **{
|
171 |
"output_dir": OUTPUT_DIR,
|
172 |
"name": "test-instance", |
173 |
"os": "lenny-image", |
174 |
"hypervisor": ("xen-pvm", {}), |
175 |
"osparams":{},
|
176 |
"disks": [],
|
177 |
}) |
178 |
ARGS_COMPLETE = dict(ARGS_VBOX, **{
|
179 |
"beparams": {"vcpus":"1", "memory":"256", "auto_balance": False}, |
180 |
"disks": [(0,{"size":"5mb"})], |
181 |
"nics": [("0",{"mode":"bridged"})], |
182 |
"disk_template": "plain", |
183 |
"tags": "test-tag-1,test-tag-2", |
184 |
}) |
185 |
ARGS_BROKEN = dict(ARGS_EXPORT_DIR , **{
|
186 |
"no_nics": True, |
187 |
"disk_template": "diskless", |
188 |
"name": "test-instance", |
189 |
"os": "lenny-image", |
190 |
"osparams": {},
|
191 |
}) |
192 |
|
193 |
EXP_ARGS_COMPRESSED = dict(ARGS_EXPORT_DIR, **{
|
194 |
"compression": True, |
195 |
}) |
196 |
|
197 |
EXP_DISKS_LIST = [ |
198 |
{ |
199 |
"format": "vmdk", |
200 |
"compression": "gzip", |
201 |
"virt-size": 90000, |
202 |
"real-size": 203, |
203 |
"path": "new_disk.cow.gz", |
204 |
}, |
205 |
{ |
206 |
"format": "cow", |
207 |
"virt-size": 15, |
208 |
"real-size": 15, |
209 |
"path": "new_disk.cow", |
210 |
}, |
211 |
] |
212 |
EXP_NETWORKS_LIST = [ |
213 |
{"mac": "aa:00:00:d8:2c:1e", "ip":"None", "link":"br0", |
214 |
"mode":"routed", "network": "test"}, |
215 |
] |
216 |
EXP_PARTIAL_GANETI_DICT = { |
217 |
"hypervisor": {"name": "xen-kvm"}, |
218 |
"os": {"name": "lenny-image"}, |
219 |
"auto_balance": "True", |
220 |
"version": "0", |
221 |
} |
222 |
EXP_GANETI_DICT = { |
223 |
"tags": None, |
224 |
"auto_balance": "False", |
225 |
"hypervisor": {
|
226 |
"root-path": "/dev/sda", |
227 |
"name": "xen-pvm", |
228 |
"kernel_args": "ro" |
229 |
}, |
230 |
"version": "0", |
231 |
"disk_template": None, |
232 |
"os": {"name": "lenny-image"} |
233 |
} |
234 |
EXP_NAME ="xen-dev-i1"
|
235 |
EXP_VCPUS = 1
|
236 |
EXP_MEMORY = 512
|
237 |
|
238 |
EXPORT_EMPTY = ("<Envelope xml:lang=\"en-US\" xmlns=\"http://schemas.dmtf.org/"
|
239 |
"ovf/envelope/1\" xmlns:gnt=\"http://ganeti\" xmlns:ovf=\""
|
240 |
"http://schemas.dmtf.org/ovf/envelope/1\" xmlns:rasd=\""
|
241 |
"http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/CIM_Resource"
|
242 |
"AllocationSettingData\" xmlns:vssd=\"http://schemas.dmtf.org"
|
243 |
"/wbem/wscim/1/cim-schema/2/CIM_VirtualSystemSettingData\""
|
244 |
" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" />")
|
245 |
EXPORT_DISKS_EMPTY = ("<References /><DiskSection><Info>Virtual disk"
|
246 |
" information</Info></DiskSection>")
|
247 |
EXPORT_DISKS = ("<References><File ovf:compression=\"gzip\" ovf:href=\"new_disk"
|
248 |
".cow.gz\" ovf:id=\"file0\" ovf:size=\"203\" /><File ovf:href="
|
249 |
"\"new_disk.cow\" ovf:id=\"file1\" ovf:size=\"15\" />"
|
250 |
"</References><DiskSection><Info>Virtual disk information"
|
251 |
"</Info><Disk ovf:capacity=\"90000\" ovf:diskId=\"disk0\" ovf"
|
252 |
":fileRef=\"file0\" ovf:format=\"http://www.vmware.com/"
|
253 |
"interfaces/specifications/vmdk.html#monolithicSparse\" /><Disk"
|
254 |
" ovf:capacity=\"15\" ovf:diskId=\"disk1\" ovf:fileRef"
|
255 |
"=\"file1\" ovf:format=\"http://www.gnome.org/~markmc/qcow"
|
256 |
"-image-format.html\" /></DiskSection>")
|
257 |
EXPORT_NETWORKS_EMPTY = ("<NetworkSection><Info>List of logical networks</Info>"
|
258 |
"</NetworkSection>")
|
259 |
EXPORT_NETWORKS = ("<NetworkSection><Info>List of logical networks</Info>"
|
260 |
"<Network ovf:name=\"routed0\" /></NetworkSection>")
|
261 |
EXPORT_GANETI_INCOMPLETE = ("<gnt:GanetiSection><gnt:Version>0</gnt:Version>"
|
262 |
"<gnt:AutoBalance>True</gnt:AutoBalance><gnt:"
|
263 |
"OperatingSystem><gnt:Name>lenny-image</gnt:Name>"
|
264 |
"<gnt:Parameters /></gnt:OperatingSystem><gnt:"
|
265 |
"Hypervisor><gnt:Name>xen-kvm</gnt:Name><gnt:"
|
266 |
"Parameters /></gnt:Hypervisor><gnt:Network><gnt:"
|
267 |
"Nic ovf:name=\"routed0\"><gnt:Mode>routed</gnt:"
|
268 |
"Mode><gnt:MACAddress>aa:00:00:d8:2c:1e</gnt:"
|
269 |
"MACAddress><gnt:IPAddress>None</gnt:IPAddress>"
|
270 |
"<gnt:Link>br0</gnt:Link><gnt:Net>test</gnt:Net>"
|
271 |
"</gnt:Nic></gnt:Network></gnt:GanetiSection>")
|
272 |
EXPORT_GANETI = ("<gnt:GanetiSection><gnt:Version>0</gnt:Version><gnt:"
|
273 |
"AutoBalance>False</gnt:AutoBalance><gnt:OperatingSystem>"
|
274 |
"<gnt:Name>lenny-image</gnt:Name><gnt:Parameters /></gnt:"
|
275 |
"OperatingSystem><gnt:Hypervisor><gnt:Name>xen-pvm</gnt:Name>"
|
276 |
"<gnt:Parameters><gnt:root-path>/dev/sda</gnt:root-path><gnt:"
|
277 |
"kernel_args>ro</gnt:kernel_args></gnt:Parameters></gnt:"
|
278 |
"Hypervisor><gnt:Network><gnt:Nic ovf:name=\"routed0\"><gnt:"
|
279 |
"Mode>routed</gnt:Mode><gnt:MACAddress>aa:00:00:d8:2c:1e</gnt:"
|
280 |
"MACAddress><gnt:IPAddress>None</gnt:IPAddress><gnt:Link>br0"
|
281 |
"</gnt:Link><gnt:Net>test</gnt:Net></gnt:Nic></gnt:Network>"
|
282 |
"</gnt:GanetiSection>")
|
283 |
EXPORT_SYSTEM = ("<References><File ovf:compression=\"gzip\" ovf:href=\"new_"
|
284 |
"disk.cow.gz\" ovf:id=\"file0\" ovf:size=\"203\" /><File ovf:"
|
285 |
"href=\"new_disk.cow\" ovf:id=\"file1\" ovf:size=\"15\" />"
|
286 |
"</References><DiskSection><Info>Virtual disk information"
|
287 |
"</Info><Disk ovf:capacity=\"90000\" ovf:diskId=\"disk0\""
|
288 |
" ovf:fileRef=\"file0\" ovf:format=\"http://www.vmware.com"
|
289 |
"/interfaces/specifications/vmdk.html#monolithicSparse\" />"
|
290 |
"<Disk ovf:capacity=\"15\" ovf:diskId=\"disk1\" ovf:fileRef"
|
291 |
"=\"file1\" ovf:format=\"http://www.gnome.org/~markmc/qcow"
|
292 |
"-image-format.html\" /></DiskSection><NetworkSection><Info>"
|
293 |
"List of logical networks</Info><Network ovf:name=\"routed0\""
|
294 |
" /></NetworkSection><VirtualSystem ovf:id=\"xen-dev-i1\">"
|
295 |
"<Info>A virtual machine</Info><Name>xen-dev-i1</Name>"
|
296 |
"<OperatingSystemSection ovf:id=\"0\"><Info>Installed guest"
|
297 |
" operating system</Info></OperatingSystemSection><Virtual"
|
298 |
"HardwareSection><Info>Virtual hardware requirements</Info>"
|
299 |
"<System><vssd:ElementName>Virtual Hardware Family"
|
300 |
"</vssd:ElementName><vssd:InstanceID>0</vssd:InstanceID><vssd:"
|
301 |
"VirtualSystemIdentifier>xen-dev-i1</vssd:VirtualSystem"
|
302 |
"Identifier><vssd:VirtualSystemType>ganeti-ovf</vssd:Virtual"
|
303 |
"SystemType></System><Item><rasd:ElementName>1 virtual CPU(s)"
|
304 |
"</rasd:ElementName><rasd:InstanceID>1</rasd:InstanceID><rasd:"
|
305 |
"ResourceType>3</rasd:ResourceType><rasd:VirtualQuantity>1"
|
306 |
"</rasd:VirtualQuantity></Item><Item><rasd:AllocationUnits>"
|
307 |
"byte * 2^20</rasd:AllocationUnits><rasd:ElementName>512MB of"
|
308 |
" memory</rasd:ElementName><rasd:InstanceID>2</rasd:"
|
309 |
"InstanceID><rasd:ResourceType>4</rasd:ResourceType><rasd:"
|
310 |
"VirtualQuantity>512</rasd:VirtualQuantity></Item><Item>"
|
311 |
"<rasd:Address>0</rasd:Address><rasd:ElementName>scsi"
|
312 |
"_controller0</rasd:ElementName><rasd:InstanceID>3"
|
313 |
"</rasd:InstanceID><rasd:ResourceSubType>lsilogic</rasd"
|
314 |
":ResourceSubType><rasd:ResourceType>6</rasd:ResourceType>"
|
315 |
"</Item><Item><rasd:ElementName>disk0</rasd:ElementName><rasd"
|
316 |
":HostResource>ovf:/disk/disk0</rasd:HostResource><rasd"
|
317 |
":InstanceID>4</rasd:InstanceID><rasd:Parent>3</rasd:Parent>"
|
318 |
"<rasd:ResourceType>17</rasd:ResourceType></Item><Item><rasd:"
|
319 |
"ElementName>disk1</rasd:ElementName><rasd:HostResource>ovf:/"
|
320 |
"disk/disk1</rasd:HostResource><rasd:InstanceID>5</rasd"
|
321 |
":InstanceID><rasd:Parent>3</rasd:Parent><rasd:ResourceType>17"
|
322 |
"</rasd:ResourceType></Item><Item><rasd:Address>aa:00"
|
323 |
":00:d8:2c:1e</rasd:Address><rasd:Connection>routed0</rasd"
|
324 |
":Connection><rasd:ElementName>routed0</rasd:ElementName><rasd"
|
325 |
":InstanceID>6</rasd:InstanceID><rasd:ResourceType>10</rasd"
|
326 |
":ResourceType></Item></VirtualHardwareSection>"
|
327 |
"</VirtualSystem>")
|
328 |
|
329 |
|
330 |
def _GetArgs(args, with_name=False): |
331 |
options = optparse.Values() |
332 |
needed = args |
333 |
if with_name:
|
334 |
needed["name"] = "test-instance" |
335 |
options._update_loose(needed) |
336 |
return options
|
337 |
|
338 |
|
339 |
OPTS_EMPTY = _GetArgs(ARGS_EMPTY) |
340 |
OPTS_EXPORT_NO_NAME = _GetArgs(ARGS_EXPORT_DIR) |
341 |
OPTS_EXPORT = _GetArgs(ARGS_EXPORT_DIR, with_name=True)
|
342 |
|
343 |
EXP_OPTS = OPTS_EXPORT_NO_NAME |
344 |
EXP_OPTS_COMPRESSED = _GetArgs(EXP_ARGS_COMPRESSED) |
345 |
|
346 |
OPTS_VBOX = _GetArgs(ARGS_VBOX) |
347 |
OPTS_COMPLETE = _GetArgs(ARGS_COMPLETE) |
348 |
OPTS_NONIC_NODISK = _GetArgs(ARGS_BROKEN) |
349 |
|
350 |
|
351 |
def _GetFullFilename(file_name): |
352 |
file_path = "%s/test/data/ovfdata/%s" % (testutils.GetSourceDir(),
|
353 |
file_name) |
354 |
file_path = os.path.abspath(file_path) |
355 |
return file_path
|
356 |
|
357 |
|
358 |
class BetterUnitTest(unittest.TestCase): |
359 |
def assertRaisesRegexp(self, exception, regexp_val, function, *args): |
360 |
try:
|
361 |
function(*args) |
362 |
self.fail("Expected raising %s" % exception) |
363 |
except exception, err:
|
364 |
regexp = re.compile(regexp_val) |
365 |
if re.search(regexp, str(err)) == None: |
366 |
self.fail("Expected matching '%s', got '%s'" % |
367 |
(regexp_val, str(err)))
|
368 |
|
369 |
|
370 |
class TestOVFImporter(BetterUnitTest): |
371 |
def setUp(self): |
372 |
self.non_existing_file = _GetFullFilename("not_the_file.ovf") |
373 |
self.ganeti_ovf = _GetFullFilename("ganeti.ovf") |
374 |
self.virtualbox_ovf = _GetFullFilename("virtualbox.ovf") |
375 |
self.ova_package = _GetFullFilename("ova.ova") |
376 |
self.empty_ovf = _GetFullFilename("empty.ovf") |
377 |
self.wrong_extension = _GetFullFilename("wrong_extension.ovd") |
378 |
self.wrong_ova_archive = _GetFullFilename("wrong_ova.ova") |
379 |
self.no_ovf_in_ova = _GetFullFilename("no_ovf.ova") |
380 |
self.importer = None |
381 |
|
382 |
def tearDown(self): |
383 |
if self.importer: |
384 |
self.importer.Cleanup()
|
385 |
del_dir = os.path.abspath(OUTPUT_DIR) |
386 |
try:
|
387 |
shutil.rmtree(del_dir) |
388 |
except OSError: |
389 |
pass
|
390 |
|
391 |
def testFileDoesNotExistError(self): |
392 |
self.assertRaisesRegexp(errors.OpPrereqError, "does not exist", |
393 |
ovf.OVFImporter, self.non_existing_file, None) |
394 |
|
395 |
def testWrongInputFileExtensionError(self): |
396 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
397 |
"Unknown file extension", ovf.OVFImporter,
|
398 |
self.wrong_extension, None) |
399 |
|
400 |
def testOVAUnpackingDirectories(self): |
401 |
self.importer = ovf.OVFImporter(self.ova_package, OPTS_EMPTY) |
402 |
self.assertTrue(self.importer.input_dir != None) |
403 |
self.assertEquals(self.importer.output_dir , pathutils.EXPORT_DIR) |
404 |
self.assertTrue(self.importer.temp_dir != None) |
405 |
|
406 |
def testOVFUnpackingDirectories(self): |
407 |
self.importer = ovf.OVFImporter(self.virtualbox_ovf, |
408 |
OPTS_EMPTY) |
409 |
self.assertEquals(self.importer.input_dir , _GetFullFilename("")) |
410 |
self.assertEquals(self.importer.output_dir , pathutils.EXPORT_DIR) |
411 |
self.assertEquals(self.importer.temp_dir , None) |
412 |
|
413 |
def testOVFSetOutputDirDirectories(self): |
414 |
self.importer = ovf.OVFImporter(self.ganeti_ovf, OPTS_EXPORT) |
415 |
self.assertEquals(self.importer.input_dir , _GetFullFilename("")) |
416 |
self.assertTrue(OUTPUT_DIR in self.importer.output_dir) |
417 |
self.assertEquals(self.importer.temp_dir , None) |
418 |
|
419 |
def testWrongOVAArchiveError(self): |
420 |
self.assertRaisesRegexp(errors.OpPrereqError, "not a proper tar", |
421 |
ovf.OVFImporter, self.wrong_ova_archive, None) |
422 |
|
423 |
def testNoOVFFileInOVAPackageError(self): |
424 |
self.assertRaisesRegexp(errors.OpPrereqError, "No .ovf file", |
425 |
ovf.OVFImporter, self.no_ovf_in_ova, None) |
426 |
|
427 |
def testParseGanetiOvf(self): |
428 |
self.importer = ovf.OVFImporter(self.ganeti_ovf, OPTS_EXPORT_NO_NAME) |
429 |
self.importer.Parse()
|
430 |
self.assertTrue("%s/ganeti-test-xen" % OUTPUT_DIR in |
431 |
self.importer.output_dir)
|
432 |
self.assertEqual(self.importer.results_disk, GANETI_DISKS) |
433 |
self.assertEqual(self.importer.results_network, GANETI_NETWORKS) |
434 |
self.assertEqual(self.importer.results_hypervisor, GANETI_HYPERVISOR) |
435 |
self.assertEqual(self.importer.results_os, GANETI_OS) |
436 |
self.assertEqual(self.importer.results_backend, GANETI_BACKEND) |
437 |
self.assertEqual(self.importer.results_name, GANETI_NAME) |
438 |
self.assertEqual(self.importer.results_template, GANETI_TEMPLATE) |
439 |
self.assertEqual(self.importer.results_tags, GANETI_TAGS) |
440 |
self.assertEqual(self.importer.results_version, GANETI_VERSION) |
441 |
|
442 |
def testParseVirtualboxOvf(self): |
443 |
self.importer = ovf.OVFImporter(self.virtualbox_ovf, OPTS_VBOX) |
444 |
self.importer.Parse()
|
445 |
self.assertTrue("%s/test-instance" % OUTPUT_DIR in self.importer.output_dir) |
446 |
self.assertEquals(self.importer.results_disk, VIRTUALBOX_DISKS) |
447 |
self.assertEquals(self.importer.results_network, VIRTUALBOX_NETWORKS) |
448 |
self.assertEquals(self.importer.results_hypervisor, CMDARGS_HYPERVISOR) |
449 |
self.assertEquals(self.importer.results_os, CMDARGS_OS) |
450 |
self.assertEquals(self.importer.results_backend, VIRTUALBOX_BACKEND) |
451 |
self.assertEquals(self.importer.results_name, CMDARGS_NAME) |
452 |
self.assertEquals(self.importer.results_template, VIRTUALBOX_TEMPLATE) |
453 |
self.assertEqual(self.importer.results_tags, VIRTUALBOX_TAGS) |
454 |
self.assertEqual(self.importer.results_version, constants.EXPORT_VERSION) |
455 |
|
456 |
def testParseEmptyOvf(self): |
457 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
458 |
self.importer.Parse()
|
459 |
self.assertTrue("%s/test-instance" % OUTPUT_DIR in self.importer.output_dir) |
460 |
self.assertEquals(self.importer.results_disk, CMDARGS_DISKS) |
461 |
self.assertEquals(self.importer.results_network, CMDARGS_NETWORKS) |
462 |
self.assertEquals(self.importer.results_hypervisor, CMDARGS_HYPERVISOR) |
463 |
self.assertEquals(self.importer.results_os, CMDARGS_OS) |
464 |
self.assertEquals(self.importer.results_backend, CMDARGS_BACKEND) |
465 |
self.assertEquals(self.importer.results_name, CMDARGS_NAME) |
466 |
self.assertEquals(self.importer.results_template, CMDARGS_TEMPLATE) |
467 |
self.assertEqual(self.importer.results_tags, CMDARGS_TAGS) |
468 |
self.assertEqual(self.importer.results_version, constants.EXPORT_VERSION) |
469 |
|
470 |
def testParseNameOptions(self): |
471 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
472 |
results = self.importer._ParseNameOptions()
|
473 |
self.assertEquals(results, CMDARGS_NAME)
|
474 |
|
475 |
def testParseHypervisorOptions(self): |
476 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
477 |
results = self.importer._ParseHypervisorOptions()
|
478 |
self.assertEquals(results, CMDARGS_HYPERVISOR)
|
479 |
|
480 |
def testParseOSOptions(self): |
481 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
482 |
results = self.importer._ParseOSOptions()
|
483 |
self.assertEquals(results, CMDARGS_OS)
|
484 |
|
485 |
def testParseBackendOptions(self): |
486 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
487 |
results = self.importer._ParseBackendOptions()
|
488 |
self.assertEquals(results, CMDARGS_BACKEND)
|
489 |
|
490 |
def testParseTags(self): |
491 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
492 |
results = self.importer._ParseTags()
|
493 |
self.assertEquals(results, CMDARGS_TAGS)
|
494 |
|
495 |
def testParseNicOptions(self): |
496 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
497 |
results = self.importer._ParseNicOptions()
|
498 |
self.assertEquals(results, CMDARGS_NETWORKS)
|
499 |
|
500 |
def testParseDiskOptionsFromGanetiOVF(self): |
501 |
self.importer = ovf.OVFImporter(self.ganeti_ovf, OPTS_EXPORT) |
502 |
os.mkdir(OUTPUT_DIR) |
503 |
results = self.importer._GetDiskInfo()
|
504 |
self.assertEquals(results, GANETI_DISKS)
|
505 |
|
506 |
def testParseTemplateOptions(self): |
507 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
508 |
results = self.importer._ParseTemplateOptions()
|
509 |
self.assertEquals(results, GANETI_TEMPLATE)
|
510 |
|
511 |
def testParseDiskOptionsFromCmdLine(self): |
512 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_COMPLETE) |
513 |
os.mkdir(OUTPUT_DIR) |
514 |
results = self.importer._ParseDiskOptions()
|
515 |
self.assertEquals(results, CMDARGS_DISKS)
|
516 |
|
517 |
def testGetDiskFormat(self): |
518 |
self.importer = ovf.OVFImporter(self.ganeti_ovf, OPTS_EXPORT) |
519 |
disks_list = self.importer.ovf_reader.GetDisksNames()
|
520 |
results = [self.importer._GetDiskQemuInfo("%s/%s" % |
521 |
(self.importer.input_dir, path), "file format: (\S+)") |
522 |
for (path, _) in disks_list] |
523 |
self.assertEqual(results, ["vmdk"]) |
524 |
|
525 |
def testNoInstanceNameOVF(self): |
526 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_EXPORT_NO_NAME) |
527 |
self.assertRaisesRegexp(errors.OpPrereqError, "Name of instance", |
528 |
self.importer.Parse)
|
529 |
|
530 |
def testErrorNoOSNameOVF(self): |
531 |
self.importer = ovf.OVFImporter(self.virtualbox_ovf, OPTS_EXPORT) |
532 |
self.assertRaisesRegexp(errors.OpPrereqError, "OS name", |
533 |
self.importer.Parse)
|
534 |
|
535 |
def testErrorNoDiskAndNoNetwork(self): |
536 |
self.importer = ovf.OVFImporter(self.empty_ovf, OPTS_NONIC_NODISK) |
537 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
538 |
"Either disk specification or network"
|
539 |
" description", self.importer.Parse) |
540 |
|
541 |
|
542 |
class TestOVFExporter(BetterUnitTest): |
543 |
def setUp(self): |
544 |
self.exporter = None |
545 |
self.wrong_config_file = _GetFullFilename("wrong_config.ini") |
546 |
self.unsafe_path_to_disk = _GetFullFilename("unsafe_path.ini") |
547 |
self.disk_image_not_exist = _GetFullFilename("no_disk.ini") |
548 |
self.empty_config = _GetFullFilename("empty.ini") |
549 |
self.standard_export = _GetFullFilename("config.ini") |
550 |
self.wrong_network_mode = self.disk_image_not_exist |
551 |
self.no_memory = self.disk_image_not_exist |
552 |
self.no_vcpus = self.disk_image_not_exist |
553 |
self.no_os = _GetFullFilename("no_os.ini") |
554 |
self.no_hypervisor = self.disk_image_not_exist |
555 |
|
556 |
def tearDown(self): |
557 |
if self.exporter: |
558 |
self.exporter.Cleanup()
|
559 |
del_dir = os.path.abspath(OUTPUT_DIR) |
560 |
try:
|
561 |
shutil.rmtree(del_dir) |
562 |
except OSError: |
563 |
pass
|
564 |
|
565 |
def testErrorWrongConfigFile(self): |
566 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
567 |
"Error when trying to read", ovf.OVFExporter,
|
568 |
self.wrong_config_file, EXP_OPTS)
|
569 |
|
570 |
def testErrorPathToTheDiskIncorrect(self): |
571 |
self.exporter = ovf.OVFExporter(self.unsafe_path_to_disk, EXP_OPTS) |
572 |
self.assertRaisesRegexp(errors.OpPrereqError, "contains a directory name", |
573 |
self.exporter._ParseDisks)
|
574 |
|
575 |
def testErrorDiskImageNotExist(self): |
576 |
self.exporter = ovf.OVFExporter(self.disk_image_not_exist, EXP_OPTS) |
577 |
self.assertRaisesRegexp(errors.OpPrereqError, "Disk image does not exist", |
578 |
self.exporter._ParseDisks)
|
579 |
|
580 |
def testParseNetworks(self): |
581 |
self.exporter = ovf.OVFExporter(self.standard_export, EXP_OPTS) |
582 |
results = self.exporter._ParseNetworks()
|
583 |
self.assertEqual(results, EXP_NETWORKS_LIST)
|
584 |
|
585 |
def testErrorWrongNetworkMode(self): |
586 |
self.exporter = ovf.OVFExporter(self.wrong_network_mode, EXP_OPTS) |
587 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
588 |
"Network mode nic not recognized", self.exporter._ParseNetworks) |
589 |
|
590 |
def testParseVCPusMem(self): |
591 |
self.exporter = ovf.OVFExporter(self.standard_export, EXP_OPTS) |
592 |
vcpus = self.exporter._ParseVCPUs()
|
593 |
memory = self.exporter._ParseMemory()
|
594 |
self.assertEqual(vcpus, EXP_VCPUS)
|
595 |
self.assertEqual(memory, EXP_MEMORY)
|
596 |
|
597 |
def testErrorNoVCPUs(self): |
598 |
self.exporter = ovf.OVFExporter(self.no_vcpus, EXP_OPTS) |
599 |
self.assertRaisesRegexp(errors.OpPrereqError, "No CPU information found", |
600 |
self.exporter._ParseVCPUs)
|
601 |
|
602 |
def testErrorNoMemory(self): |
603 |
self.exporter = ovf.OVFExporter(self.no_memory, EXP_OPTS) |
604 |
self.assertRaisesRegexp(errors.OpPrereqError, "No memory information found", |
605 |
self.exporter._ParseMemory)
|
606 |
|
607 |
def testParseGaneti(self): |
608 |
self.exporter = ovf.OVFExporter(self.standard_export, EXP_OPTS) |
609 |
results = self.exporter._ParseGaneti()
|
610 |
self.assertEqual(results, EXP_GANETI_DICT)
|
611 |
|
612 |
def testErrorNoHypervisor(self): |
613 |
self.exporter = ovf.OVFExporter(self.no_hypervisor, EXP_OPTS) |
614 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
615 |
"No hypervisor information found", self.exporter._ParseGaneti) |
616 |
|
617 |
def testErrorNoOS(self): |
618 |
self.exporter = ovf.OVFExporter(self.no_os, EXP_OPTS) |
619 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
620 |
"No operating system information found", self.exporter._ParseGaneti) |
621 |
|
622 |
def testErrorParseNoInstanceName(self): |
623 |
self.exporter = ovf.OVFExporter(self.empty_config, EXP_OPTS) |
624 |
self.assertRaisesRegexp(errors.OpPrereqError, "No instance name found", |
625 |
self.exporter.Parse)
|
626 |
|
627 |
|
628 |
class TestOVFReader(BetterUnitTest): |
629 |
def setUp(self): |
630 |
self.wrong_xml_file = _GetFullFilename("wrong_xml.ovf") |
631 |
self.ganeti_ovf = _GetFullFilename("ganeti.ovf") |
632 |
self.virtualbox_ovf = _GetFullFilename("virtualbox.ovf") |
633 |
self.corrupted_ovf = _GetFullFilename("corrupted_resources.ovf") |
634 |
self.wrong_manifest_ovf = _GetFullFilename("wrong_manifest.ovf") |
635 |
self.no_disk_in_ref_ovf = _GetFullFilename("no_disk_in_ref.ovf") |
636 |
self.empty_ovf = _GetFullFilename("empty.ovf") |
637 |
self.compressed_disk = _GetFullFilename("gzip_disk.ovf") |
638 |
|
639 |
def tearDown(self): |
640 |
pass
|
641 |
|
642 |
def testXMLParsingError(self): |
643 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
644 |
"Error while reading .ovf", ovf.OVFReader, self.wrong_xml_file) |
645 |
|
646 |
def testFileInResourcesDoesNotExistError(self): |
647 |
self.assertRaisesRegexp(errors.OpPrereqError, "does not exist", |
648 |
ovf.OVFReader, self.corrupted_ovf)
|
649 |
|
650 |
def testWrongManifestChecksumError(self): |
651 |
reader = ovf.OVFReader(self.wrong_manifest_ovf)
|
652 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
653 |
"does not match the value in manifest file", reader.VerifyManifest)
|
654 |
|
655 |
def testGoodManifestChecksum(self): |
656 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
657 |
self.assertEqual(reader.VerifyManifest(), None) |
658 |
|
659 |
def testGetDisksNamesOVFCorruptedError(self): |
660 |
reader = ovf.OVFReader(self.no_disk_in_ref_ovf)
|
661 |
self.assertRaisesRegexp(errors.OpPrereqError,
|
662 |
"not found in references", reader.GetDisksNames)
|
663 |
|
664 |
def testGetDisksNamesVirtualbox(self): |
665 |
reader = ovf.OVFReader(self.virtualbox_ovf)
|
666 |
disk_names = reader.GetDisksNames() |
667 |
expected_names = [ |
668 |
("new_disk.vmdk", None) , |
669 |
("second_disk.vmdk", None), |
670 |
] |
671 |
self.assertEqual(sorted(disk_names), sorted(expected_names)) |
672 |
|
673 |
def testGetDisksNamesEmpty(self): |
674 |
reader = ovf.OVFReader(self.empty_ovf)
|
675 |
disk_names = reader.GetDisksNames() |
676 |
self.assertEqual(disk_names, [])
|
677 |
|
678 |
def testGetDisksNamesCompressed(self): |
679 |
reader = ovf.OVFReader(self.compressed_disk)
|
680 |
disk_names = reader.GetDisksNames() |
681 |
self.assertEqual(disk_names, [("compr_disk.vmdk.gz", "gzip")]) |
682 |
|
683 |
def testGetNetworkDataGaneti(self): |
684 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
685 |
networks = reader.GetNetworkData() |
686 |
self.assertEqual(networks, GANETI_NETWORKS)
|
687 |
|
688 |
def testGetNetworkDataVirtualbox(self): |
689 |
reader = ovf.OVFReader(self.virtualbox_ovf)
|
690 |
networks = reader.GetNetworkData() |
691 |
self.assertEqual(networks, VIRTUALBOX_NETWORKS)
|
692 |
|
693 |
def testGetNetworkDataEmpty(self): |
694 |
reader = ovf.OVFReader(self.empty_ovf)
|
695 |
networks = reader.GetNetworkData() |
696 |
self.assertEqual(networks, EMPTY_NETWORKS)
|
697 |
|
698 |
def testGetHypervisorDataGaneti(self): |
699 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
700 |
hypervisor = reader.GetHypervisorData() |
701 |
self.assertEqual(hypervisor, GANETI_HYPERVISOR)
|
702 |
|
703 |
def testGetHypervisorDataEmptyOvf(self): |
704 |
reader = ovf.OVFReader(self.empty_ovf)
|
705 |
hypervisor = reader.GetHypervisorData() |
706 |
self.assertEqual(hypervisor, EMPTY_HYPERVISOR)
|
707 |
|
708 |
def testGetOSDataGaneti(self): |
709 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
710 |
osys = reader.GetOSData() |
711 |
self.assertEqual(osys, GANETI_OS)
|
712 |
|
713 |
def testGetOSDataEmptyOvf(self): |
714 |
reader = ovf.OVFReader(self.empty_ovf)
|
715 |
osys = reader.GetOSData() |
716 |
self.assertEqual(osys, EMPTY_OS)
|
717 |
|
718 |
def testGetBackendDataGaneti(self): |
719 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
720 |
backend = reader.GetBackendData() |
721 |
self.assertEqual(backend, GANETI_BACKEND)
|
722 |
|
723 |
def testGetBackendDataVirtualbox(self): |
724 |
reader = ovf.OVFReader(self.virtualbox_ovf)
|
725 |
backend = reader.GetBackendData() |
726 |
self.assertEqual(backend, VIRTUALBOX_BACKEND)
|
727 |
|
728 |
def testGetBackendDataEmptyOvf(self): |
729 |
reader = ovf.OVFReader(self.empty_ovf)
|
730 |
backend = reader.GetBackendData() |
731 |
self.assertEqual(backend, EMPTY_BACKEND)
|
732 |
|
733 |
def testGetInstanceNameGaneti(self): |
734 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
735 |
name = reader.GetInstanceName() |
736 |
self.assertEqual(name, GANETI_NAME)
|
737 |
|
738 |
def testGetInstanceNameDataEmptyOvf(self): |
739 |
reader = ovf.OVFReader(self.empty_ovf)
|
740 |
name = reader.GetInstanceName() |
741 |
self.assertEqual(name, EMPTY_NAME)
|
742 |
|
743 |
def testGetDiskTemplateGaneti(self): |
744 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
745 |
name = reader.GetDiskTemplate() |
746 |
self.assertEqual(name, GANETI_TEMPLATE)
|
747 |
|
748 |
def testGetDiskTemplateEmpty(self): |
749 |
reader = ovf.OVFReader(self.empty_ovf)
|
750 |
name = reader.GetDiskTemplate() |
751 |
self.assertEqual(name, EMPTY_TEMPLATE)
|
752 |
|
753 |
def testGetTagsGaneti(self): |
754 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
755 |
tags = reader.GetTagsData() |
756 |
self.assertEqual(tags, GANETI_TAGS)
|
757 |
|
758 |
def testGetTagsEmpty(self): |
759 |
reader = ovf.OVFReader(self.empty_ovf)
|
760 |
tags = reader.GetTagsData() |
761 |
self.assertEqual(tags, EMPTY_TAGS)
|
762 |
|
763 |
def testGetVersionGaneti(self): |
764 |
reader = ovf.OVFReader(self.ganeti_ovf)
|
765 |
version = reader.GetVersionData() |
766 |
self.assertEqual(version, GANETI_VERSION)
|
767 |
|
768 |
def testGetVersionEmpty(self): |
769 |
reader = ovf.OVFReader(self.empty_ovf)
|
770 |
version = reader.GetVersionData() |
771 |
self.assertEqual(version, EMPTY_VERSION)
|
772 |
|
773 |
|
774 |
class TestOVFWriter(BetterUnitTest): |
775 |
def setUp(self): |
776 |
self.writer = ovf.OVFWriter(True) |
777 |
|
778 |
def tearDown(self): |
779 |
pass
|
780 |
|
781 |
def testOVFWriterInit(self): |
782 |
result = ET.tostring(self.writer.tree)
|
783 |
self.assertTrue(EXPORT_EMPTY in result) |
784 |
|
785 |
def testSaveDisksDataEmpty(self): |
786 |
self.writer.SaveDisksData([])
|
787 |
result = ET.tostring(self.writer.tree)
|
788 |
self.assertTrue(EXPORT_DISKS_EMPTY in result) |
789 |
|
790 |
def testSaveDisksData(self): |
791 |
self.writer.SaveDisksData(EXP_DISKS_LIST)
|
792 |
result = ET.tostring(self.writer.tree)
|
793 |
self.assertTrue(EXPORT_DISKS in result) |
794 |
|
795 |
def testSaveNetworkDataEmpty(self): |
796 |
self.writer.SaveNetworksData([])
|
797 |
result = ET.tostring(self.writer.tree)
|
798 |
self.assertTrue(EXPORT_NETWORKS_EMPTY in result) |
799 |
|
800 |
def testSaveNetworksData(self): |
801 |
self.writer.SaveNetworksData(EXP_NETWORKS_LIST)
|
802 |
result = ET.tostring(self.writer.tree)
|
803 |
self.assertTrue(EXPORT_NETWORKS in result) |
804 |
|
805 |
def testSaveGanetiDataIncomplete(self): |
806 |
self.writer.SaveGanetiData(EXP_PARTIAL_GANETI_DICT, EXP_NETWORKS_LIST)
|
807 |
result = ET.tostring(self.writer.tree)
|
808 |
self.assertTrue(EXPORT_GANETI_INCOMPLETE in result) |
809 |
|
810 |
def testSaveGanetiDataComplete(self): |
811 |
self.writer.SaveGanetiData(EXP_GANETI_DICT, EXP_NETWORKS_LIST)
|
812 |
result = ET.tostring(self.writer.tree)
|
813 |
self.assertTrue(EXPORT_GANETI in result) |
814 |
|
815 |
def testSaveVirtualSystem(self): |
816 |
self.writer.SaveDisksData(EXP_DISKS_LIST)
|
817 |
self.writer.SaveNetworksData(EXP_NETWORKS_LIST)
|
818 |
self.writer.SaveVirtualSystemData(EXP_NAME, EXP_VCPUS, EXP_MEMORY)
|
819 |
result = ET.tostring(self.writer.tree)
|
820 |
self.assertTrue(EXPORT_SYSTEM in result) |
821 |
|
822 |
|
823 |
if __name__ == "__main__": |
824 |
testutils.GanetiTestProgram() |