root / snf-cyclades-app / synnefo / logic / tests / servers.py @ 406968dc
History | View | Annotate | Download (11 kB)
1 | 0208770b | Christos Stavrakakis | # vim: set fileencoding=utf-8 :
|
---|---|---|---|
2 | 0208770b | Christos Stavrakakis | # Copyright 2013 GRNET S.A. All rights reserved.
|
3 | 0208770b | Christos Stavrakakis | #
|
4 | 0208770b | Christos Stavrakakis | # Redistribution and use in source and binary forms, with or without
|
5 | 0208770b | Christos Stavrakakis | # modification, are permitted provided that the following conditions
|
6 | 0208770b | Christos Stavrakakis | # are met:
|
7 | 0208770b | Christos Stavrakakis | #
|
8 | 0208770b | Christos Stavrakakis | # 1. Redistributions of source code must retain the above copyright
|
9 | 0208770b | Christos Stavrakakis | # notice, this list of conditions and the following disclaimer.
|
10 | 0208770b | Christos Stavrakakis | #
|
11 | 0208770b | Christos Stavrakakis | # 2. Redistributions in binary form must reproduce the above copyright
|
12 | 0208770b | Christos Stavrakakis | # notice, this list of conditions and the following disclaimer in the
|
13 | 0208770b | Christos Stavrakakis | # documentation and/or other materials provided with the distribution.
|
14 | 0208770b | Christos Stavrakakis | #
|
15 | 0208770b | Christos Stavrakakis | # THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
|
16 | 0208770b | Christos Stavrakakis | # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
17 | 0208770b | Christos Stavrakakis | # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
18 | 0208770b | Christos Stavrakakis | # ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
|
19 | 0208770b | Christos Stavrakakis | # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
20 | 0208770b | Christos Stavrakakis | # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
21 | 0208770b | Christos Stavrakakis | # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
22 | 0208770b | Christos Stavrakakis | # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
23 | 0208770b | Christos Stavrakakis | # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
24 | 0208770b | Christos Stavrakakis | # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
25 | 0208770b | Christos Stavrakakis | # SUCH DAMAGE.
|
26 | 0208770b | Christos Stavrakakis | #
|
27 | 0208770b | Christos Stavrakakis | # The views and conclusions contained in the software and documentation are
|
28 | 0208770b | Christos Stavrakakis | # those of the authors and should not be interpreted as representing official
|
29 | 0208770b | Christos Stavrakakis | # policies, either expressed or implied, of GRNET S.A.
|
30 | 0208770b | Christos Stavrakakis | |
31 | 0208770b | Christos Stavrakakis | # Provides automated tests for logic module
|
32 | 562bf712 | Christos Stavrakakis | from django.test import TransactionTestCase |
33 | 0208770b | Christos Stavrakakis | #from snf_django.utils.testing import mocked_quotaholder
|
34 | 0208770b | Christos Stavrakakis | from synnefo.logic import servers |
35 | fb540e3a | Giorgos Korfiatis | from synnefo import quotas |
36 | 562bf712 | Christos Stavrakakis | from synnefo.db import models_factory as mfactory, models |
37 | 0208770b | Christos Stavrakakis | from mock import patch |
38 | 0208770b | Christos Stavrakakis | |
39 | 0208770b | Christos Stavrakakis | from snf_django.lib.api import faults |
40 | 0208770b | Christos Stavrakakis | from snf_django.utils.testing import mocked_quotaholder, override_settings |
41 | 728e8923 | Christos Stavrakakis | from django.conf import settings |
42 | 562bf712 | Christos Stavrakakis | from copy import deepcopy |
43 | 0208770b | Christos Stavrakakis | |
44 | 0208770b | Christos Stavrakakis | |
45 | 0208770b | Christos Stavrakakis | @patch("synnefo.logic.rapi_pool.GanetiRapiClient") |
46 | 562bf712 | Christos Stavrakakis | class ServerCreationTest(TransactionTestCase): |
47 | 0208770b | Christos Stavrakakis | def test_create(self, mrapi): |
48 | 0208770b | Christos Stavrakakis | flavor = mfactory.FlavorFactory() |
49 | 0208770b | Christos Stavrakakis | kwargs = { |
50 | 0208770b | Christos Stavrakakis | "userid": "test", |
51 | 0208770b | Christos Stavrakakis | "name": "test_vm", |
52 | 0208770b | Christos Stavrakakis | "password": "1234", |
53 | 0208770b | Christos Stavrakakis | "flavor": flavor,
|
54 | 0208770b | Christos Stavrakakis | "image": {"id": "foo", "backend_id": "foo", "format": "diskdump", |
55 | 0208770b | Christos Stavrakakis | "metadata": "{}"}, |
56 | 0208770b | Christos Stavrakakis | "metadata": {"foo": "bar"}, |
57 | 0208770b | Christos Stavrakakis | "personality": [],
|
58 | 0208770b | Christos Stavrakakis | } |
59 | 562bf712 | Christos Stavrakakis | # no backend!
|
60 | 562bf712 | Christos Stavrakakis | mfactory.BackendFactory(offline=True)
|
61 | 562bf712 | Christos Stavrakakis | self.assertRaises(faults.ServiceUnavailable, servers.create, **kwargs)
|
62 | 562bf712 | Christos Stavrakakis | self.assertEqual(models.VirtualMachine.objects.count(), 0) |
63 | 0208770b | Christos Stavrakakis | |
64 | 3aecadc8 | Christos Stavrakakis | mfactory.IPv4SubnetFactory(network__public=True)
|
65 | 3aecadc8 | Christos Stavrakakis | mfactory.IPv6SubnetFactory(network__public=True)
|
66 | 3aecadc8 | Christos Stavrakakis | mfactory.BackendFactory() |
67 | 562bf712 | Christos Stavrakakis | |
68 | 562bf712 | Christos Stavrakakis | # error in nics
|
69 | 562bf712 | Christos Stavrakakis | req = deepcopy(kwargs) |
70 | 99988465 | Christos Stavrakakis | req["networks"] = [{"uuid": 42}] |
71 | 562bf712 | Christos Stavrakakis | self.assertRaises(faults.ItemNotFound, servers.create, **req)
|
72 | 562bf712 | Christos Stavrakakis | self.assertEqual(models.VirtualMachine.objects.count(), 0) |
73 | 562bf712 | Christos Stavrakakis | |
74 | 562bf712 | Christos Stavrakakis | # error in enqueue. check the vm is deleted and resources released
|
75 | 562bf712 | Christos Stavrakakis | mrapi().CreateInstance.side_effect = Exception("ganeti is down") |
76 | 562bf712 | Christos Stavrakakis | with mocked_quotaholder():
|
77 | 9dcfad23 | Christos Stavrakakis | servers.create(**kwargs) |
78 | 562bf712 | Christos Stavrakakis | vm = models.VirtualMachine.objects.get() |
79 | 9dcfad23 | Christos Stavrakakis | self.assertFalse(vm.deleted)
|
80 | 9dcfad23 | Christos Stavrakakis | self.assertEqual(vm.operstate, "ERROR") |
81 | 3aecadc8 | Christos Stavrakakis | self.assertEqual(len(vm.nics.all()), 2) |
82 | 9dcfad23 | Christos Stavrakakis | for nic in vm.nics.all(): |
83 | 9dcfad23 | Christos Stavrakakis | self.assertEqual(nic.state, "ERROR") |
84 | 562bf712 | Christos Stavrakakis | |
85 | 562bf712 | Christos Stavrakakis | |
86 | 562bf712 | Christos Stavrakakis | @patch("synnefo.logic.rapi_pool.GanetiRapiClient") |
87 | 562bf712 | Christos Stavrakakis | class ServerTest(TransactionTestCase): |
88 | 0208770b | Christos Stavrakakis | def test_connect_network(self, mrapi): |
89 | 0208770b | Christos Stavrakakis | # Common connect
|
90 | 710b1c43 | Christos Stavrakakis | for dhcp in [True, False]: |
91 | 710b1c43 | Christos Stavrakakis | subnet = mfactory.IPv4SubnetFactory(network__flavor="CUSTOM",
|
92 | 710b1c43 | Christos Stavrakakis | cidr="192.168.2.0/24",
|
93 | 710b1c43 | Christos Stavrakakis | gateway="192.168.2.1",
|
94 | 3b2984dc | Christos Stavrakakis | dhcp=dhcp) |
95 | 710b1c43 | Christos Stavrakakis | net = subnet.network |
96 | 710b1c43 | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STARTED")
|
97 | 710b1c43 | Christos Stavrakakis | mfactory.BackendNetworkFactory(network=net, backend=vm.backend) |
98 | 710b1c43 | Christos Stavrakakis | mrapi().ModifyInstance.return_value = 42
|
99 | e8234183 | Christos Stavrakakis | with override_settings(settings, GANETI_USE_HOTPLUG=True): |
100 | e8234183 | Christos Stavrakakis | servers.connect(vm, net) |
101 | 3b2984dc | Christos Stavrakakis | pool = net.get_ip_pools(locked=False)[0] |
102 | 710b1c43 | Christos Stavrakakis | self.assertFalse(pool.is_available("192.168.2.2")) |
103 | 710b1c43 | Christos Stavrakakis | args, kwargs = mrapi().ModifyInstance.call_args |
104 | 710b1c43 | Christos Stavrakakis | nics = kwargs["nics"][0] |
105 | 710b1c43 | Christos Stavrakakis | self.assertEqual(kwargs["instance"], vm.backend_vm_id) |
106 | 710b1c43 | Christos Stavrakakis | self.assertEqual(nics[0], "add") |
107 | 710b1c43 | Christos Stavrakakis | self.assertEqual(nics[1], "-1") |
108 | 710b1c43 | Christos Stavrakakis | self.assertEqual(nics[2]["ip"], "192.168.2.2") |
109 | 710b1c43 | Christos Stavrakakis | self.assertEqual(nics[2]["network"], net.backend_id) |
110 | 0208770b | Christos Stavrakakis | |
111 | 0208770b | Christos Stavrakakis | # Test connect to IPv6 only network
|
112 | 9599e997 | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STARTED")
|
113 | 8764d304 | Christos Stavrakakis | subnet = mfactory.IPv6SubnetFactory(cidr="2000::/64",
|
114 | 8764d304 | Christos Stavrakakis | gateway="2000::1")
|
115 | 8764d304 | Christos Stavrakakis | net = subnet.network |
116 | 0208770b | Christos Stavrakakis | mfactory.BackendNetworkFactory(network=net, backend=vm.backend) |
117 | e8234183 | Christos Stavrakakis | with override_settings(settings, GANETI_USE_HOTPLUG=True): |
118 | e8234183 | Christos Stavrakakis | servers.connect(vm, net) |
119 | 0208770b | Christos Stavrakakis | args, kwargs = mrapi().ModifyInstance.call_args |
120 | 0208770b | Christos Stavrakakis | nics = kwargs["nics"][0] |
121 | 6488097c | Christos Stavrakakis | self.assertEqual(kwargs["instance"], vm.backend_vm_id) |
122 | 0208770b | Christos Stavrakakis | self.assertEqual(nics[0], "add") |
123 | a1baa42b | Christos Stavrakakis | self.assertEqual(nics[1], "-1") |
124 | a1baa42b | Christos Stavrakakis | self.assertEqual(nics[2]["ip"], None) |
125 | a1baa42b | Christos Stavrakakis | self.assertEqual(nics[2]["network"], net.backend_id) |
126 | 0208770b | Christos Stavrakakis | |
127 | 0208770b | Christos Stavrakakis | |
128 | 0208770b | Christos Stavrakakis | @patch("synnefo.logic.rapi_pool.GanetiRapiClient") |
129 | 562bf712 | Christos Stavrakakis | class ServerCommandTest(TransactionTestCase): |
130 | 0208770b | Christos Stavrakakis | def test_pending_task(self, mrapi): |
131 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(task="REBOOT", task_job_id=1) |
132 | 0208770b | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.start, vm)
|
133 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(task="BUILD", task_job_id=1) |
134 | 0208770b | Christos Stavrakakis | self.assertRaises(faults.BuildInProgress, servers.start, vm)
|
135 | 0208770b | Christos Stavrakakis | # Assert always succeeds
|
136 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(task="BUILD", task_job_id=1) |
137 | 0208770b | Christos Stavrakakis | mrapi().DeleteInstance.return_value = 1
|
138 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
139 | 0208770b | Christos Stavrakakis | servers.destroy(vm) |
140 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(task="REBOOT", task_job_id=1) |
141 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
142 | 0208770b | Christos Stavrakakis | servers.destroy(vm) |
143 | 0208770b | Christos Stavrakakis | |
144 | 0208770b | Christos Stavrakakis | def test_deleted_vm(self, mrapi): |
145 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(deleted=True)
|
146 | 0208770b | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.start, vm)
|
147 | 0208770b | Christos Stavrakakis | |
148 | 0208770b | Christos Stavrakakis | def test_invalid_operstate_for_action(self, mrapi): |
149 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STARTED")
|
150 | 0208770b | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.start, vm)
|
151 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STOPPED")
|
152 | 0208770b | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.stop, vm)
|
153 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STARTED")
|
154 | 64bca363 | Giorgos Korfiatis | flavor = mfactory.FlavorFactory() |
155 | 64bca363 | Giorgos Korfiatis | self.assertRaises(faults.BadRequest, servers.resize, vm, flavor)
|
156 | 9599e997 | Christos Stavrakakis | # Check that connect/disconnect is allowed only in STOPPED vms
|
157 | 9599e997 | Christos Stavrakakis | # if hotplug is disabled.
|
158 | 9599e997 | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STARTED")
|
159 | 9599e997 | Christos Stavrakakis | network = mfactory.NetworkFactory(state="ACTIVE")
|
160 | 9599e997 | Christos Stavrakakis | with override_settings(settings, GANETI_USE_HOTPLUG=False): |
161 | 9599e997 | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.connect, vm, network)
|
162 | 9599e997 | Christos Stavrakakis | self.assertRaises(faults.BadRequest, servers.disconnect, vm,
|
163 | 9599e997 | Christos Stavrakakis | network) |
164 | 0208770b | Christos Stavrakakis | #test valid
|
165 | 9599e997 | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STOPPED")
|
166 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.return_value = 1
|
167 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
168 | 0208770b | Christos Stavrakakis | servers.start(vm) |
169 | 0208770b | Christos Stavrakakis | vm.task = None
|
170 | 0208770b | Christos Stavrakakis | vm.task_job_id = None
|
171 | 0208770b | Christos Stavrakakis | vm.save() |
172 | fb540e3a | Giorgos Korfiatis | with mocked_quotaholder():
|
173 | fb540e3a | Giorgos Korfiatis | quotas.accept_serial(vm.serial) |
174 | 0208770b | Christos Stavrakakis | mrapi().RebootInstance.return_value = 1
|
175 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
176 | 0208770b | Christos Stavrakakis | servers.reboot(vm, "HARD")
|
177 | 0208770b | Christos Stavrakakis | |
178 | 0208770b | Christos Stavrakakis | def test_commission(self, mrapi): |
179 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory(operstate="STOPPED")
|
180 | 0208770b | Christos Stavrakakis | # Still pending
|
181 | 0208770b | Christos Stavrakakis | vm.serial = mfactory.QuotaHolderSerialFactory(serial=200,
|
182 | 0208770b | Christos Stavrakakis | resolved=False,
|
183 | 0208770b | Christos Stavrakakis | pending=True)
|
184 | 0208770b | Christos Stavrakakis | serial = vm.serial |
185 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.return_value = 1
|
186 | 0208770b | Christos Stavrakakis | with mocked_quotaholder() as m: |
187 | fb540e3a | Giorgos Korfiatis | with self.assertRaises(quotas.ResolveError): |
188 | fb540e3a | Giorgos Korfiatis | servers.start(vm) |
189 | 0208770b | Christos Stavrakakis | # Not pending, rejct
|
190 | 0208770b | Christos Stavrakakis | vm.task = None
|
191 | 0208770b | Christos Stavrakakis | vm.serial = mfactory.QuotaHolderSerialFactory(serial=400,
|
192 | 0208770b | Christos Stavrakakis | resolved=False,
|
193 | 0208770b | Christos Stavrakakis | pending=False,
|
194 | 0208770b | Christos Stavrakakis | accept=False)
|
195 | 0208770b | Christos Stavrakakis | serial = vm.serial |
196 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.return_value = 1
|
197 | 0208770b | Christos Stavrakakis | with mocked_quotaholder() as m: |
198 | 0208770b | Christos Stavrakakis | servers.start(vm) |
199 | e407f159 | Ilias Tsitsimpis | m.resolve_commissions.assert_called_once_with([], |
200 | 0208770b | Christos Stavrakakis | [serial.serial]) |
201 | 0208770b | Christos Stavrakakis | self.assertTrue(m.issue_one_commission.called)
|
202 | 0208770b | Christos Stavrakakis | # Not pending, accept
|
203 | 0208770b | Christos Stavrakakis | vm.task = None
|
204 | 0208770b | Christos Stavrakakis | vm.serial = mfactory.QuotaHolderSerialFactory(serial=600,
|
205 | 0208770b | Christos Stavrakakis | resolved=False,
|
206 | 0208770b | Christos Stavrakakis | pending=False,
|
207 | 0208770b | Christos Stavrakakis | accept=True)
|
208 | 0208770b | Christos Stavrakakis | serial = vm.serial |
209 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.return_value = 1
|
210 | 0208770b | Christos Stavrakakis | with mocked_quotaholder() as m: |
211 | 0208770b | Christos Stavrakakis | servers.start(vm) |
212 | e407f159 | Ilias Tsitsimpis | m.resolve_commissions.assert_called_once_with([serial.serial], |
213 | 0208770b | Christos Stavrakakis | []) |
214 | 0208770b | Christos Stavrakakis | self.assertTrue(m.issue_one_commission.called)
|
215 | 0208770b | Christos Stavrakakis | |
216 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.side_effect = ValueError
|
217 | 0208770b | Christos Stavrakakis | vm.task = None
|
218 | 0208770b | Christos Stavrakakis | vm.serial = None
|
219 | 0208770b | Christos Stavrakakis | # Test reject if Ganeti erro
|
220 | 0208770b | Christos Stavrakakis | with mocked_quotaholder() as m: |
221 | 0208770b | Christos Stavrakakis | try:
|
222 | 0208770b | Christos Stavrakakis | servers.start(vm) |
223 | 0208770b | Christos Stavrakakis | except:
|
224 | 0208770b | Christos Stavrakakis | m.resolve_commissions\ |
225 | e407f159 | Ilias Tsitsimpis | .assert_called_once_with([], [vm.serial.serial]) |
226 | 0208770b | Christos Stavrakakis | |
227 | 0208770b | Christos Stavrakakis | def test_task_after(self, mrapi): |
228 | 0208770b | Christos Stavrakakis | return
|
229 | 0208770b | Christos Stavrakakis | vm = mfactory.VirtualMachineFactory() |
230 | 0208770b | Christos Stavrakakis | mrapi().StartupInstance.return_value = 1
|
231 | 0208770b | Christos Stavrakakis | mrapi().ShutdownInstance.return_value = 2
|
232 | 0208770b | Christos Stavrakakis | mrapi().RebootInstance.return_value = 2
|
233 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
234 | 0208770b | Christos Stavrakakis | vm.task = None
|
235 | 0208770b | Christos Stavrakakis | vm.operstate = "STOPPED"
|
236 | 0208770b | Christos Stavrakakis | servers.start(vm) |
237 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task, "START") |
238 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task_job_id, 1) |
239 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
240 | 0208770b | Christos Stavrakakis | vm.task = None
|
241 | 0208770b | Christos Stavrakakis | vm.operstate = "STARTED"
|
242 | 0208770b | Christos Stavrakakis | servers.stop(vm) |
243 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task, "STOP") |
244 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task_job_id, 2) |
245 | 0208770b | Christos Stavrakakis | with mocked_quotaholder():
|
246 | 0208770b | Christos Stavrakakis | vm.task = None
|
247 | 0208770b | Christos Stavrakakis | servers.reboot(vm) |
248 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task, "REBOOT") |
249 | 0208770b | Christos Stavrakakis | self.assertEqual(vm.task_job_id, 3) |