1 import unittest
2
3 import VMBuilder
4 import VMBuilder.plugins
5 from VMBuilder.exception import VMBuilderUserError
6
14
16 class TestHypervisor():
17 arg = 'test'
18
19 VMBuilder.register_hypervisor(TestHypervisor)
20 self.assertEqual(TestHypervisor, VMBuilder.get_hypervisor('test'))
21
23 class Plugin(object):
24 priority = 10
25
26 class PluginA(Plugin):
27 pass
28
29 class PluginB(Plugin):
30 priority = 5
31
32 class PluginC(Plugin):
33 priority = 15
34
35 saved_plugins = getattr(VMBuilder, plugin_attr_name)
36 setattr(VMBuilder, plugin_attr_name, [])
37 register_function(PluginA)
38 register_function(PluginB)
39 register_function(PluginC)
40 self.assertEqual(getattr(VMBuilder, plugin_attr_name)[0], PluginB)
41 self.assertEqual(getattr(VMBuilder, plugin_attr_name)[1], PluginA)
42 self.assertEqual(getattr(VMBuilder, plugin_attr_name)[2], PluginC)
43 setattr(VMBuilder, plugin_attr_name, saved_plugins)
44
47
50
53
56
61
63 info = VMBuilder.get_version_info()
64 self.assertTrue('major' in info)
65 self.assertTrue('minor' in info)
66 self.assertTrue('micro' in info)
67 self.assertTrue('revno' in info)
68