root / snf-cyclades-app / synnefo / api / test / servers.py @ 039e3e61
History | View | Annotate | Download (16.3 kB)
1 |
# Copyright 2012 GRNET S.A. All rights reserved.
|
---|---|
2 |
#
|
3 |
# Redistribution and use in source and binary forms, with or
|
4 |
# without modification, are permitted provided that the following
|
5 |
# conditions are met:
|
6 |
#
|
7 |
# 1. Redistributions of source code must retain the above
|
8 |
# copyright notice, this list of conditions and the following
|
9 |
# disclaimer.
|
10 |
#
|
11 |
# 2. Redistributions in binary form must reproduce the above
|
12 |
# copyright notice, this list of conditions and the following
|
13 |
# disclaimer in the documentation and/or other materials
|
14 |
# provided with the distribution.
|
15 |
#
|
16 |
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 |
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 |
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 |
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 |
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 |
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 |
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 |
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 |
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 |
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 |
# POSSIBILITY OF SUCH DAMAGE.
|
28 |
#
|
29 |
# The views and conclusions contained in the software and
|
30 |
# documentation are those of the authors and should not be
|
31 |
# interpreted as representing official policies, either expressed
|
32 |
# or implied, of GRNET S.A.
|
33 |
|
34 |
import json |
35 |
|
36 |
from synnefo.api.tests import BaseAPITest |
37 |
from synnefo.db.models import VirtualMachine, VirtualMachineMetadata |
38 |
from synnefo.db import models_factory as mfactory |
39 |
from synnefo.logic.utils import get_rsapi_state |
40 |
|
41 |
from mock import patch |
42 |
|
43 |
|
44 |
class ServerAPITest(BaseAPITest): |
45 |
def setUp(self): |
46 |
self.user1 = 'user1' |
47 |
self.user2 = 'user2' |
48 |
self.vm1 = mfactory.VirtualMachineFactory(userid=self.user1) |
49 |
self.vm2 = mfactory.VirtualMachineFactory(userid=self.user2) |
50 |
self.vm3 = mfactory.VirtualMachineFactory(deleted=True, |
51 |
userid=self.user1)
|
52 |
self.vm4 = mfactory.VirtualMachineFactory(userid=self.user2) |
53 |
|
54 |
def test_server_list_1(self): |
55 |
"""Test if the expected list of servers is returned."""
|
56 |
response = self.get('/api/v1.1/servers') |
57 |
self.assertSuccess(response)
|
58 |
servers = json.loads(response.content)['servers']['values'] |
59 |
self.assertEqual(servers, [])
|
60 |
|
61 |
def test_server_list_2(self): |
62 |
"""Test if the expected list of servers is returned."""
|
63 |
response = self.get('/api/v1.1/servers', self.user1) |
64 |
self.assertSuccess(response)
|
65 |
servers = json.loads(response.content)['servers']['values'] |
66 |
db_server = self.vm1
|
67 |
self.assertEqual(servers, [{'name': db_server.name, |
68 |
'id': db_server.id}])
|
69 |
|
70 |
def test_server_list_detail(self): |
71 |
"""Test if the servers list details are returned."""
|
72 |
user = self.user2
|
73 |
user_vms = {self.vm2.id: self.vm2, |
74 |
self.vm4.id: self.vm4} |
75 |
|
76 |
response = self.get('/api/v1.1/servers/detail', user) |
77 |
servers = json.loads(response.content)['servers']['values'] |
78 |
self.assertEqual(len(servers), len(user_vms)) |
79 |
for api_vm in servers: |
80 |
db_vm = user_vms[api_vm['id']]
|
81 |
self.assertEqual(api_vm['flavorRef'], db_vm.flavor.id) |
82 |
self.assertEqual(api_vm['hostId'], db_vm.hostid) |
83 |
self.assertEqual(api_vm['id'], db_vm.id) |
84 |
self.assertEqual(api_vm['imageRef'], db_vm.imageid) |
85 |
self.assertEqual(api_vm['name'], db_vm.name) |
86 |
self.assertEqual(api_vm['status'], get_rsapi_state(db_vm)) |
87 |
self.assertSuccess(response)
|
88 |
|
89 |
def test_server_detail(self): |
90 |
"""Test if a server details are returned."""
|
91 |
db_vm = self.vm2
|
92 |
user = self.vm2.userid
|
93 |
net = mfactory.NetworkFactory() |
94 |
nic = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net)
|
95 |
|
96 |
db_vm_meta = mfactory.VirtualMachineMetadataFactory(vm=db_vm) |
97 |
|
98 |
response = self.get('/api/v1.1/servers/%d' % db_vm.id, user) |
99 |
server = json.loads(response.content)['server']
|
100 |
|
101 |
self.assertEqual(server['flavorRef'], db_vm.flavor.id) |
102 |
self.assertEqual(server['hostId'], db_vm.hostid) |
103 |
self.assertEqual(server['id'], db_vm.id) |
104 |
self.assertEqual(server['imageRef'], db_vm.imageid) |
105 |
self.assertEqual(server['name'], db_vm.name) |
106 |
self.assertEqual(server['status'], get_rsapi_state(db_vm)) |
107 |
api_nic = server['attachments']['values'][0] |
108 |
self.assertEqual(api_nic['network_id'], str(net.id)) |
109 |
self.assertEqual(api_nic['mac_address'], nic.mac) |
110 |
self.assertEqual(api_nic['firewallProfile'], nic.firewall_profile) |
111 |
self.assertEqual(api_nic['ipv4'], nic.ipv4) |
112 |
self.assertEqual(api_nic['ipv6'], nic.ipv6) |
113 |
self.assertEqual(api_nic['id'], 'nic-%s-%s' % (db_vm.id, nic.index)) |
114 |
|
115 |
metadata = server['metadata']['values'] |
116 |
self.assertEqual(len(metadata), 1) |
117 |
self.assertEqual(metadata[db_vm_meta.meta_key], db_vm_meta.meta_value)
|
118 |
self.assertSuccess(response)
|
119 |
|
120 |
def test_noauthorized(self): |
121 |
"""Test 404 for detail of other user vm"""
|
122 |
db_vm = self.vm2
|
123 |
|
124 |
response = self.get('/api/v1.1/servers/%d' % db_vm.id, 'wrong_user') |
125 |
self.assertItemNotFound(response)
|
126 |
|
127 |
def test_wrong_server(self): |
128 |
"""Test 404 response if server does not exist."""
|
129 |
response = self.get('/api/v1.1/servers/%d' % 5000) |
130 |
self.assertItemNotFound(response)
|
131 |
|
132 |
def test_create_server_empty(self): |
133 |
"""Test if the create server call returns a 400 badRequest if
|
134 |
no attributes are specified."""
|
135 |
|
136 |
response = self.post('/api/v1.1/servers', params={}) |
137 |
self.assertBadRequest(response)
|
138 |
|
139 |
def test_rename_server(self): |
140 |
vm = self.vm2
|
141 |
request = {'server': {'name': 'new_name'}} |
142 |
response = self.put('/api/v1.1/servers/%d' % vm.id, vm.userid, |
143 |
json.dumps(request), 'json')
|
144 |
self.assertSuccess(response)
|
145 |
self.assertEqual(VirtualMachine.objects.get(id=vm.id).name, "new_name") |
146 |
|
147 |
|
148 |
@patch('synnefo.api.util.get_image') |
149 |
@patch('synnefo.logic.rapi_pool.GanetiRapiClient') |
150 |
class ServerCreateAPITest(BaseAPITest): |
151 |
def test_create_server(self, mrapi, mimage): |
152 |
"""Test if the create server call returns the expected response
|
153 |
if a valid request has been speficied."""
|
154 |
mimage.return_value = {'location': 'pithos://foo', |
155 |
'disk_format': 'diskdump'} |
156 |
mrapi().CreateInstance.return_value = 12
|
157 |
flavor = mfactory.FlavorFactory() |
158 |
# Create public network and backend
|
159 |
network = mfactory.NetworkFactory(public=True)
|
160 |
backend = mfactory.BackendFactory() |
161 |
mfactory.BackendNetworkFactory(network=network, backend=backend) |
162 |
|
163 |
request = { |
164 |
"server": {
|
165 |
"name": "new-server-test", |
166 |
"userid": "test_user", |
167 |
"imageRef": 1, |
168 |
"flavorRef": flavor.id,
|
169 |
"metadata": {
|
170 |
"My Server Name": "Apache1" |
171 |
}, |
172 |
"personality": []
|
173 |
} |
174 |
} |
175 |
response = self.post('/api/v1.1/servers', 'test_user', |
176 |
json.dumps(request), 'json')
|
177 |
self.assertEqual(response.status_code, 202) |
178 |
mrapi().CreateInstance.assert_called_once() |
179 |
|
180 |
api_server = json.loads(response.content)['server']
|
181 |
self.assertEqual(api_server['status'], "BUILD") |
182 |
self.assertEqual(api_server['progress'], 0) |
183 |
self.assertEqual(api_server['metadata']['values'], |
184 |
{"My Server Name": "Apache1"}) |
185 |
self.assertTrue('adminPass' in api_server) |
186 |
|
187 |
db_vm = VirtualMachine.objects.get(userid='test_user')
|
188 |
self.assertEqual(api_server['name'], db_vm.name) |
189 |
self.assertEqual(api_server['status'], db_vm.operstate) |
190 |
|
191 |
def test_create_server_no_flavor(self, mrapi, mimage): |
192 |
request = { |
193 |
"server": {
|
194 |
"name": "new-server-test", |
195 |
"userid": "test_user", |
196 |
"imageRef": 1, |
197 |
"flavorRef": 42, |
198 |
"metadata": {
|
199 |
"My Server Name": "Apache1" |
200 |
}, |
201 |
"personality": []
|
202 |
} |
203 |
} |
204 |
response = self.post('/api/v1.1/servers', 'test_user', |
205 |
json.dumps(request), 'json')
|
206 |
self.assertItemNotFound(response)
|
207 |
|
208 |
|
209 |
@patch('synnefo.logic.rapi_pool.GanetiRapiClient') |
210 |
class ServerDestroyAPITest(BaseAPITest): |
211 |
def test_delete_server(self, mrapi): |
212 |
vm = mfactory.VirtualMachineFactory() |
213 |
response = self.delete('/api/v1.1/servers/%d' % vm.id, vm.userid) |
214 |
self.assertEqual(response.status_code, 204) |
215 |
mrapi().DeleteInstance.assert_called_once() |
216 |
|
217 |
def test_non_existing_delete_server(self, mrapi): |
218 |
vm = mfactory.VirtualMachineFactory() |
219 |
response = self.delete('/api/v1.1/servers/%d' % 42, vm.userid) |
220 |
self.assertItemNotFound(response)
|
221 |
self.assertFalse(mrapi.mock_calls)
|
222 |
|
223 |
|
224 |
class ServerMetadataAPITest(BaseAPITest): |
225 |
def setUp(self): |
226 |
self.vm = mfactory.VirtualMachineFactory()
|
227 |
self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm) |
228 |
|
229 |
def test_get_metadata(self): |
230 |
vm = self.vm
|
231 |
create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
|
232 |
metadata = [create_meta(), create_meta(), create_meta()] |
233 |
response = self.get('/api/v1.1/servers/%d/meta' % vm.id, vm.userid) |
234 |
self.assertTrue(response.status_code in [200, 203]) |
235 |
api_metadata = json.loads(response.content)['metadata']['values'] |
236 |
self.assertEqual(len(api_metadata), len(metadata) + 1) |
237 |
for db_m in metadata: |
238 |
self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
|
239 |
|
240 |
request = {'metadata':
|
241 |
{'foo': 'bar'}, |
242 |
metadata[0].meta_key: 'bar2' |
243 |
} |
244 |
response = self.post('/api/v1.1/servers/%d/meta' % vm.id, vm.userid, |
245 |
json.dumps(request), 'json')
|
246 |
metadata2 = VirtualMachineMetadata.objects.filter(vm=vm) |
247 |
response = self.get('/api/v1.1/servers/%d/meta' % vm.id, vm.userid) |
248 |
self.assertTrue(response.status_code in [200, 203]) |
249 |
api_metadata2 = json.loads(response.content)['metadata']['values'] |
250 |
self.assertTrue('foo' in api_metadata2.keys()) |
251 |
self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2') |
252 |
self.assertEqual(len(api_metadata2), len(metadata2)) |
253 |
for db_m in metadata2: |
254 |
self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
|
255 |
|
256 |
# Create new meta
|
257 |
request = {'meta': {'foo2': 'bar2'}} |
258 |
response = self.put('/api/v1.1/servers/%d/meta/foo2' % vm.id, |
259 |
vm.userid, json.dumps(request), 'json')
|
260 |
|
261 |
# Get the new meta
|
262 |
response = self.get('/api/v1.1/servers/%d/meta/foo2' % vm.id, |
263 |
vm.userid) |
264 |
meta = json.loads(response.content)['meta']
|
265 |
self.assertEqual(meta['foo2'], 'bar2') |
266 |
|
267 |
# Delete the new meta
|
268 |
response = self.delete('/api/v1.1/servers/%d/meta/foo2' % vm.id, |
269 |
vm.userid) |
270 |
self.assertEqual(response.status_code, 204) |
271 |
|
272 |
# Try to get the deleted meta: should raise 404
|
273 |
response = self.get('/api/v1.1/servers/%d/meta/foo2' % vm.id, |
274 |
vm.userid) |
275 |
self.assertEqual(response.status_code, 404) |
276 |
|
277 |
def test_invalid_metadata(self): |
278 |
vm = self.vm
|
279 |
response = self.post('/api/v1.1/servers/%d/meta' % vm.id, vm.userid) |
280 |
self.assertBadRequest(response)
|
281 |
self.assertEqual(len(vm.metadata.all()), 1) |
282 |
|
283 |
def test_invalid_metadata_server(self): |
284 |
response = self.post('/api/v1.1/servers/42/meta', 'user') |
285 |
self.assertItemNotFound(response)
|
286 |
|
287 |
def test_get_meta_invalid_key(self): |
288 |
vm = self.vm
|
289 |
response = self.get('/api/v1.1/servers/%d/meta/foo2' % vm.id, |
290 |
vm.userid) |
291 |
self.assertItemNotFound(response)
|
292 |
|
293 |
|
294 |
@patch('synnefo.api.util.get_image') |
295 |
@patch('synnefo.logic.rapi_pool.GanetiRapiClient') |
296 |
class ServerActionAPITest(BaseAPITest): |
297 |
def test_actions(self, mrapi, mimage): |
298 |
actions = ['start', 'shutdown', 'reboot'] |
299 |
vm = mfactory.VirtualMachineFactory() |
300 |
vm.operstate = "STOPPED"
|
301 |
vm.save() |
302 |
for action in actions: |
303 |
val = {'type': 'HARD'} if action == 'reboot' else {} |
304 |
request = {action: val} |
305 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
306 |
vm.userid, json.dumps(request), 'json')
|
307 |
self.assertEqual(response.status_code, 202) |
308 |
if action == 'shutdown': |
309 |
self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
|
310 |
"STOP")
|
311 |
else:
|
312 |
self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
|
313 |
action.upper()) |
314 |
|
315 |
def test_action_in_building_vm(self, mrapi, mimage): |
316 |
"""Test building in progress"""
|
317 |
vm = mfactory.VirtualMachineFactory() |
318 |
request = {'start': '{}'} |
319 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
320 |
vm.userid, json.dumps(request), 'json')
|
321 |
self.assertEqual(response.status_code, 409) |
322 |
self.assertFalse(mrapi.mock_calls)
|
323 |
|
324 |
def test_destroy_build_vm(self, mrapi, mimage): |
325 |
"""Test building in progress"""
|
326 |
vm = mfactory.VirtualMachineFactory() |
327 |
response = self.delete('/api/v1.1/servers/%d' % vm.id, |
328 |
vm.userid) |
329 |
self.assertSuccess(response)
|
330 |
mrapi().RemoveInstance.assert_called_once() |
331 |
|
332 |
def test_firewall(self, mrapi, mimage): |
333 |
vm = mfactory.VirtualMachineFactory() |
334 |
vm.operstate = "STOPPED"
|
335 |
vm.save() |
336 |
request = {'firewallProfile': {'profile': 'PROTECTED'}} |
337 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
338 |
vm.userid, json.dumps(request), 'json')
|
339 |
self.assertEqual(response.status_code, 202) |
340 |
mrapi().ModifyInstance.assert_called_once() |
341 |
|
342 |
def test_unsupported_firewall(self, mrapi, mimage): |
343 |
vm = mfactory.VirtualMachineFactory() |
344 |
vm.operstate = "STOPPED"
|
345 |
vm.save() |
346 |
request = {'firewallProfile': {'profile': 'FOO'}} |
347 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
348 |
vm.userid, json.dumps(request), 'json')
|
349 |
self.assertBadRequest(response)
|
350 |
self.assertFalse(mrapi.mock_calls)
|
351 |
|
352 |
|
353 |
class ServerVNCConsole(BaseAPITest): |
354 |
def test_not_active_server(self): |
355 |
"""Test console req for not ACTIVE server returns badRequest"""
|
356 |
vm = mfactory.VirtualMachineFactory() |
357 |
data = json.dumps({'console': {'type': 'vnc'}}) |
358 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
359 |
vm.userid, data, 'json')
|
360 |
self.assertBadRequest(response)
|
361 |
|
362 |
def test_active_server(self): |
363 |
"""Test console req for ACTIVE server"""
|
364 |
vm = mfactory.VirtualMachineFactory() |
365 |
vm.operstate = 'STARTED'
|
366 |
vm.save() |
367 |
|
368 |
data = json.dumps({'console': {'type': 'vnc'}}) |
369 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
370 |
vm.userid, data, 'json')
|
371 |
self.assertEqual(response.status_code, 200) |
372 |
reply = json.loads(response.content) |
373 |
self.assertEqual(reply.keys(), ['console']) |
374 |
console = reply['console']
|
375 |
self.assertEqual(console['type'], 'vnc') |
376 |
self.assertEqual(set(console.keys()), |
377 |
set(['type', 'host', 'port', 'password'])) |
378 |
|
379 |
def test_wrong_console_type(self): |
380 |
"""Test console req for ACTIVE server"""
|
381 |
vm = mfactory.VirtualMachineFactory() |
382 |
vm.operstate = 'STARTED'
|
383 |
vm.save() |
384 |
|
385 |
data = json.dumps({'console': {'type': 'foo'}}) |
386 |
response = self.post('/api/v1.1/servers/%d/action' % vm.id, |
387 |
vm.userid, data, 'json')
|
388 |
self.assertBadRequest(response)
|