Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / test / servers.py @ a95c82a9

History | View | Annotate | Download (24.2 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 snf_django.utils.testing import (BaseAPITest, mocked_quotaholder,
37
                                      override_settings)
38
from synnefo.db.models import VirtualMachine, VirtualMachineMetadata
39
from synnefo.db import models_factory as mfactory
40
from synnefo.logic.utils import get_rsapi_state
41
from synnefo.cyclades_settings import cyclades_services
42
from synnefo.lib.services import get_service_path
43
from synnefo.lib import join_urls
44
from synnefo import settings
45

    
46
from mock import patch
47

    
48

    
49
class ComputeAPITest(BaseAPITest):
50
    def setUp(self, *args, **kwargs):
51
        super(ComputeAPITest, self).setUp(*args, **kwargs)
52
        self.compute_path = get_service_path(cyclades_services, 'compute',
53
                                             version='v2.0')
54

    
55
    def myget(self, path, *args, **kwargs):
56
        path = join_urls(self.compute_path, path)
57
        return self.get(path, *args, **kwargs)
58

    
59
    def myput(self, path, *args, **kwargs):
60
        path = join_urls(self.compute_path, path)
61
        return self.put(path, *args, **kwargs)
62

    
63
    def mypost(self, path, *args, **kwargs):
64
        path = join_urls(self.compute_path, path)
65
        return self.post(path, *args, **kwargs)
66

    
67
    def mydelete(self, path, *args, **kwargs):
68
        path = join_urls(self.compute_path, path)
69
        return self.delete(path, *args, **kwargs)
70

    
71

    
72
class ServerAPITest(ComputeAPITest):
73
    def setUp(self):
74
        self.user1 = 'user1'
75
        self.user2 = 'user2'
76
        self.vm1 = mfactory.VirtualMachineFactory(userid=self.user1)
77
        self.vm2 = mfactory.VirtualMachineFactory(userid=self.user2)
78
        self.vm3 = mfactory.VirtualMachineFactory(deleted=True,
79
                                                  userid=self.user1)
80
        self.vm4 = mfactory.VirtualMachineFactory(userid=self.user2)
81
        super(ServerAPITest, self).setUp()
82

    
83
    def test_server_list_1(self):
84
        """Test if the expected list of servers is returned."""
85
        response = self.myget('servers')
86
        self.assertSuccess(response)
87
        servers = json.loads(response.content)['servers']
88
        self.assertEqual(servers, [])
89

    
90
    def test_server_list_2(self):
91
        """Test if the expected list of servers is returned."""
92
        response = self.myget('servers', self.user1)
93
        self.assertSuccess(response)
94
        servers = json.loads(response.content)['servers']
95
        db_server = self.vm1
96
        server = servers[0]
97
        self.assertEqual(server["name"], db_server.name)
98
        self.assertEqual(server["id"], db_server.id)
99

    
100
    def test_server_list_detail(self):
101
        """Test if the servers list details are returned."""
102
        user = self.user2
103
        user_vms = {self.vm2.id: self.vm2,
104
                    self.vm4.id: self.vm4}
105

    
106
        response = self.myget('servers/detail', user)
107
        servers = json.loads(response.content)['servers']
108
        self.assertEqual(len(servers), len(user_vms))
109
        for api_vm in servers:
110
            db_vm = user_vms[api_vm['id']]
111
            self.assertEqual(api_vm['flavor']["id"], db_vm.flavor.id)
112
            self.assertEqual(api_vm['hostId'], db_vm.hostid)
113
            self.assertEqual(api_vm['id'], db_vm.id)
114
            self.assertEqual(api_vm['image']["id"], db_vm.imageid)
115
            self.assertEqual(api_vm['name'], db_vm.name)
116
            self.assertEqual(api_vm['status'], get_rsapi_state(db_vm))
117
            self.assertSuccess(response)
118

    
119
    def test_server_detail(self):
120
        """Test if a server details are returned."""
121
        db_vm = self.vm2
122
        user = self.vm2.userid
123
        net = mfactory.NetworkFactory()
124
        nic = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net,
125
                                              ipv6="::babe")
126

    
127
        db_vm_meta = mfactory.VirtualMachineMetadataFactory(vm=db_vm)
128

    
129
        response = self.myget('servers/%d' % db_vm.id, user)
130
        server = json.loads(response.content)['server']
131

    
132
        self.assertEqual(server['flavor']["id"], db_vm.flavor.id)
133
        self.assertEqual(server['hostId'], db_vm.hostid)
134
        self.assertEqual(server['id'], db_vm.id)
135
        self.assertEqual(server['image']["id"], db_vm.imageid)
136
        self.assertEqual(server['name'], db_vm.name)
137
        self.assertEqual(server['status'], get_rsapi_state(db_vm))
138
        api_nic = server['attachments'][0]
139
        self.assertEqual(api_nic['network_id'], str(net.id))
140
        self.assertEqual(api_nic['mac_address'], nic.mac)
141
        self.assertEqual(api_nic['firewallProfile'], nic.firewall_profile)
142
        self.assertEqual(api_nic['ipv4'], nic.ipv4)
143
        self.assertEqual(api_nic['ipv6'], nic.ipv6)
144
        self.assertEqual(api_nic['id'], 'nic-%s-%s' % (db_vm.id, nic.index))
145
        api_address = server["addresses"]
146
        self.assertEqual(api_address[str(net.id)],
147
               [{"version": 4, "addr": nic.ipv4, "OS-EXT-IPS:type": "fixed"},
148
                {"version": 6, "addr": nic.ipv6, "OS-EXT-IPS:type": "fixed"}])
149

    
150
        metadata = server['metadata']
151
        self.assertEqual(len(metadata), 1)
152
        self.assertEqual(metadata[db_vm_meta.meta_key], db_vm_meta.meta_value)
153
        self.assertSuccess(response)
154

    
155
    def test_server_building_nics(self):
156
        db_vm = self.vm2
157
        user = self.vm2.userid
158
        net1 = mfactory.NetworkFactory()
159
        net2 = mfactory.NetworkFactory()
160
        net3 = mfactory.NetworkFactory()
161
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net1,
162
                                         state="BUILDING")
163
        nic2 = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net2,
164
                                                state="ACTIVE")
165
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net3,
166
                                         state="BUILDING")
167

    
168
        response = self.myget('servers/%d' % db_vm.id, user)
169
        server = json.loads(response.content)['server']
170
        nics = server["attachments"]
171
        self.assertEqual(len(nics), 1)
172
        self.assertEqual(nics[0]["network_id"], str(nic2.network_id))
173

    
174
    def test_noauthorized(self):
175
        """Test 404 for detail of other user vm"""
176
        db_vm = self.vm2
177

    
178
        response = self.myget('servers/%d' % db_vm.id, 'wrong_user')
179
        self.assertItemNotFound(response)
180

    
181
    def test_wrong_server(self):
182
        """Test 404 response if server does not exist."""
183
        response = self.myget('servers/%d' % 5000)
184
        self.assertItemNotFound(response)
185

    
186
    def test_create_server_empty(self):
187
        """Test if the create server call returns a 400 badRequest if
188
           no attributes are specified."""
189

    
190
        response = self.mypost('servers', params={})
191
        self.assertBadRequest(response)
192

    
193
    def test_rename_server(self):
194
        vm = self.vm2
195
        request = {'server': {'name': 'new_name'}}
196
        response = self.myput('servers/%d' % vm.id, vm.userid,
197
                              json.dumps(request), 'json')
198
        self.assertSuccess(response)
199
        self.assertEqual(VirtualMachine.objects.get(id=vm.id).name, "new_name")
200

    
201
    def test_catch_wrong_api_paths(self):
202
        response = self.myget('nonexistent')
203
        self.assertEqual(response.status_code, 400)
204
        try:
205
            error = json.loads(response.content)
206
        except ValueError:
207
            self.assertTrue(False)
208

    
209
    def test_method_not_allowed(self, *args):
210
        # /servers/ allows only POST, GET
211
        response = self.myput('servers', '', '')
212
        self.assertMethodNotAllowed(response)
213
        response = self.mydelete('servers')
214
        self.assertMethodNotAllowed(response)
215

    
216
        # /servers/<srvid>/ allows only GET, PUT, DELETE
217
        response = self.mypost("servers/42")
218
        self.assertMethodNotAllowed(response)
219

    
220
        # /imags/<srvid>/metadata/ allows only POST, GET
221
        response = self.myput('servers/42/metadata', '', '')
222
        self.assertMethodNotAllowed(response)
223
        response = self.mydelete('servers/42/metadata')
224
        self.assertMethodNotAllowed(response)
225

    
226
        # /imags/<srvid>/metadata/ allows only POST, GET
227
        response = self.myput('servers/42/metadata', '', '')
228
        self.assertMethodNotAllowed(response)
229
        response = self.mydelete('servers/42/metadata')
230
        self.assertMethodNotAllowed(response)
231

    
232
        # /imags/<srvid>/metadata/<key> allows only PUT, GET, DELETE
233
        response = self.mypost('servers/42/metadata/foo')
234
        self.assertMethodNotAllowed(response)
235

    
236

    
237
@patch('synnefo.api.util.get_image')
238
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
239
class ServerCreateAPITest(ComputeAPITest):
240
    def test_create_server(self, mrapi, mimage):
241
        """Test if the create server call returns the expected response
242
           if a valid request has been speficied."""
243
        mimage.return_value = {'location': 'pithos://foo',
244
                               'checksum': '1234',
245
                               "id": 1,
246
                               "name": "test_image",
247
                               'disk_format': 'diskdump'}
248
        mrapi().CreateInstance.return_value = 12
249
        flavor = mfactory.FlavorFactory()
250
        # Create public network and backend
251
        network = mfactory.NetworkFactory(public=True)
252
        backend = mfactory.BackendFactory()
253
        mfactory.BackendNetworkFactory(network=network, backend=backend)
254

    
255
        request = {
256
                    "server": {
257
                        "name": "new-server-test",
258
                        "userid": "test_user",
259
                        "imageRef": 1,
260
                        "flavorRef": flavor.id,
261
                        "metadata": {
262
                            "My Server Name": "Apache1"
263
                        },
264
                        "personality": []
265
                    }
266
        }
267
        with mocked_quotaholder():
268
            response = self.mypost('servers', 'test_user',
269
                                   json.dumps(request), 'json')
270
        self.assertEqual(response.status_code, 202)
271
        mrapi().CreateInstance.assert_called_once()
272

    
273
        api_server = json.loads(response.content)['server']
274
        self.assertEqual(api_server['status'], "BUILD")
275
        self.assertEqual(api_server['progress'], 0)
276
        self.assertEqual(api_server['metadata'],
277
                        {"My Server Name":  "Apache1"})
278
        self.assertTrue('adminPass' in api_server)
279

    
280
        db_vm = VirtualMachine.objects.get(userid='test_user')
281
        self.assertEqual(api_server['name'], db_vm.name)
282
        self.assertEqual(api_server['status'], db_vm.operstate)
283

    
284
        # Test drained flag in Network:
285
        network.drained = True
286
        network.save()
287
        with mocked_quotaholder():
288
            response = self.mypost('servers', 'test_user',
289
                                    json.dumps(request), 'json')
290
        self.assertEqual(response.status_code, 503, "serviceUnavailable")
291

    
292
    def test_create_server_no_flavor(self, mrapi, mimage):
293
        request = {
294
                    "server": {
295
                        "name": "new-server-test",
296
                        "userid": "test_user",
297
                        "imageRef": 1,
298
                        "flavorRef": 42,
299
                        "metadata": {
300
                            "My Server Name": "Apache1"
301
                        },
302
                        "personality": []
303
                    }
304
        }
305
        response = self.mypost('servers', 'test_user',
306
                               json.dumps(request), 'json')
307
        self.assertItemNotFound(response)
308

    
309

    
310
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
311
class ServerDestroyAPITest(ComputeAPITest):
312
    def test_delete_server(self, mrapi):
313
        vm = mfactory.VirtualMachineFactory()
314
        mrapi().DeleteInstance.return_value = 12
315
        response = self.mydelete('/api/v1.1/servers/%d' % vm.id, vm.userid)
316
        self.assertEqual(response.status_code, 204)
317
        mrapi().DeleteInstance.assert_called_once()
318

    
319
    def test_non_existing_delete_server(self, mrapi):
320
        vm = mfactory.VirtualMachineFactory()
321
        response = self.mydelete('servers/%d' % 42, vm.userid)
322
        self.assertItemNotFound(response)
323
        self.assertFalse(mrapi.mock_calls)
324

    
325

    
326
class ServerMetadataAPITest(ComputeAPITest):
327
    def setUp(self):
328
        self.vm = mfactory.VirtualMachineFactory()
329
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
330
        super(ServerMetadataAPITest, self).setUp()
331

    
332
    def test_get_metadata(self):
333
        vm = self.vm
334
        create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
335
        metadata = [create_meta(), create_meta(), create_meta()]
336
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
337
        self.assertTrue(response.status_code in [200, 203])
338
        api_metadata = json.loads(response.content)['metadata']
339
        self.assertEqual(len(api_metadata), len(metadata) + 1)
340
        for db_m in metadata:
341
            self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
342

    
343
        request = {'metadata':
344
                        {'foo': 'bar'},
345
                        metadata[0].meta_key: 'bar2'
346
                  }
347
        response = self.mypost('servers/%d/metadata' % vm.id,
348
                             vm.userid, json.dumps(request), 'json')
349
        metadata2 = VirtualMachineMetadata.objects.filter(vm=vm)
350
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
351
        self.assertTrue(response.status_code in [200, 203])
352
        api_metadata2 = json.loads(response.content)['metadata']
353
        self.assertTrue('foo' in api_metadata2.keys())
354
        self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2')
355
        self.assertEqual(len(api_metadata2), len(metadata2))
356
        for db_m in metadata2:
357
            self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
358

    
359
        # Create new meta
360
        request = {'meta': {'foo2': 'bar2'}}
361
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
362
                              vm.userid, json.dumps(request), 'json')
363

    
364
        # Get the new meta
365
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
366
        meta = json.loads(response.content)['meta']
367
        self.assertEqual(meta['foo2'], 'bar2')
368

    
369
        # Delete the new meta
370
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
371
        self.assertEqual(response.status_code, 204)
372

    
373
        # Try to get the deleted meta: should raise 404
374
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
375
        self.assertEqual(response.status_code, 404)
376

    
377
    def test_invalid_metadata(self):
378
        vm = self.vm
379
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
380
        self.assertBadRequest(response)
381
        self.assertEqual(len(vm.metadata.all()), 1)
382

    
383
    def test_invalid_metadata_server(self):
384
        response = self.mypost('servers/42/metadata', 'user')
385
        self.assertItemNotFound(response)
386

    
387
    def test_get_meta_invalid_key(self):
388
        vm = self.vm
389
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
390
        self.assertItemNotFound(response)
391

    
392

    
393
@patch('synnefo.api.util.get_image')
394
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
395
class ServerActionAPITest(ComputeAPITest):
396
    def test_actions(self, mrapi, mimage):
397
        actions = ['start', 'shutdown', 'reboot']
398
        vm = mfactory.VirtualMachineFactory()
399
        vm.operstate = "STOPPED"
400
        vm.save()
401
        mrapi().StartupInstance.return_value = 0
402
        mrapi().ShutdownInstance.return_value = 1
403
        mrapi().RebootInstance.return_value = 2
404
        for jobId, action in enumerate(actions):
405
            if action in ["shutdown", "reboot"]:
406
                vm.operstate = "STARTED"
407
            else:
408
                vm.operstate = "STOPPED"
409
            vm.task = None
410
            vm.task_job_id = None
411
            vm.save()
412
            val = {'type': 'HARD'} if action == 'reboot' else {}
413
            request = {action: val}
414
            response = self.mypost('servers/%d/action' % vm.id,
415
                                   vm.userid, json.dumps(request), 'json')
416
            self.assertEqual(response.status_code, 202)
417
            if action == 'shutdown':
418
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
419
                                 "STOP")
420
            else:
421
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
422
                                 action.upper())
423
            self.assertEqual(VirtualMachine.objects.get(id=vm.id).task_job_id,
424
                             jobId)
425

    
426
    def test_action_in_building_vm(self, mrapi, mimage):
427
        """Test building in progress"""
428
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
429
        request = {'start': {}}
430
        with mocked_quotaholder():
431
            response = self.mypost('/api/v1.1/servers/%d/action' % vm.id,
432
                                   vm.userid, json.dumps(request), 'json')
433
        self.assertEqual(response.status_code, 409)
434
        self.assertFalse(mrapi.mock_calls)
435

    
436
    def test_destroy_build_vm(self, mrapi, mimage):
437
        """Test building in progress"""
438
        vm = mfactory.VirtualMachineFactory()
439
        mrapi().DeleteInstance.return_value = 2
440
        response = self.mydelete('/api/v1.1/servers/%d' % vm.id,
441
                                 vm.userid)
442
        self.assertSuccess(response)
443
        mrapi().RemoveInstance.assert_called_once()
444

    
445
    def test_firewall(self, mrapi, mimage):
446
        vm = mfactory.VirtualMachineFactory()
447
        vm.operstate = "STOPPED"
448
        vm.save()
449
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
450
        response = self.mypost('servers/%d/action' % vm.id,
451
                               vm.userid, json.dumps(request), 'json')
452
        self.assertEqual(response.status_code, 202)
453
        mrapi().ModifyInstance.assert_called_once()
454

    
455
    def test_unsupported_firewall(self, mrapi, mimage):
456
        vm = mfactory.VirtualMachineFactory()
457
        vm.operstate = "STOPPED"
458
        vm.save()
459
        request = {'firewallProfile': {'profile': 'FOO'}}
460
        response = self.mypost('servers/%d/action' % vm.id,
461
                               vm.userid, json.dumps(request), 'json')
462
        self.assertBadRequest(response)
463
        self.assertFalse(mrapi.mock_calls)
464

    
465
    def test_resize_vm(self, mrapi, mimage):
466
        flavor = mfactory.FlavorFactory(cpu=1, ram=1024)
467
        # Check building VM
468
        vm = self.get_vm(flavor=flavor, operstate="BUILD")
469
        request = {'resize': {'flavorRef': flavor.id}}
470
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
471
                             vm.userid, json.dumps(request), 'json')
472
        self.assertFault(response, 409, "buildInProgress")
473
        # Check same Flavor
474
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
475
        request = {'resize': {'flavorRef': flavor.id}}
476
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
477
                             vm.userid, json.dumps(request), 'json')
478
        self.assertBadRequest(response)
479
        # Check flavor with different disk
480
        flavor2 = mfactory.FlavorFactory(disk=1024)
481
        flavor3 = mfactory.FlavorFactory(disk=2048)
482
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
483
        request = {'resize': {'flavorRef': flavor3.id}}
484
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
485
                             vm.userid, json.dumps(request), 'json')
486
        self.assertBadRequest(response)
487
        flavor2 = mfactory.FlavorFactory(disk_template="foo")
488
        flavor3 = mfactory.FlavorFactory(disk_template="baz")
489
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
490
        request = {'resize': {'flavorRef': flavor3.id}}
491
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
492
                             vm.userid, json.dumps(request), 'json')
493
        self.assertBadRequest(response)
494
        # Check success
495
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
496
        flavor4 = mfactory.FlavorFactory(disk_template=flavor.disk_template,
497
                                         disk=flavor.disk,
498
                                         cpu=4, ram=2048)
499
        request = {'resize': {'flavorRef': flavor4.id}}
500
        mrapi().ModifyInstance.return_value = 42
501
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
502
                             vm.userid, json.dumps(request), 'json')
503
        self.assertEqual(response.status_code, 202)
504
        vm = VirtualMachine.objects.get(id=vm.id)
505
        self.assertEqual(vm.task_job_id, 42)
506
        name, args, kwargs = mrapi().ModifyInstance.mock_calls[0]
507
        self.assertEqual(kwargs["beparams"]["vcpus"], 4)
508
        self.assertEqual(kwargs["beparams"]["minmem"], 2048)
509
        self.assertEqual(kwargs["beparams"]["maxmem"], 2048)
510

    
511
    def test_action_on_resizing_vm(self, mrapi, mimage):
512
        vm = mfactory.VirtualMachineFactory()
513
        vm.operstate = "RESIZE"
514
        vm.save()
515
        for action in VirtualMachine.ACTIONS:
516
            request = {action[0]: ""}
517
            response = self.post('/api/v1.1/servers/%d/action' % vm.id,
518
                                 vm.userid, json.dumps(request), 'json')
519
            self.assertBadRequest(response)
520
        # however you can destroy
521
        mrapi().DeleteInstance.return_value = 42
522
        response = self.delete('/api/v1.1/servers/%d' % vm.id,
523
                               vm.userid)
524
        self.assertSuccess(response)
525

    
526
    def get_vm(self, flavor, operstate):
527
        vm = mfactory.VirtualMachineFactory(flavor=flavor)
528
        vm.operstate = operstate
529
        vm.backendjobstatus = "success"
530
        vm.save()
531
        return vm
532

    
533

    
534
class ServerVNCConsole(ComputeAPITest):
535
    def test_not_active_server(self):
536
        """Test console req for not ACTIVE server returns badRequest"""
537
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
538
        data = json.dumps({'console': {'type': 'vnc'}})
539
        response = self.mypost('servers/%d/action' % vm.id,
540
                               vm.userid, data, 'json')
541
        self.assertBadRequest(response)
542

    
543
    def test_active_server(self):
544
        """Test console req for ACTIVE server"""
545
        vm = mfactory.VirtualMachineFactory()
546
        vm.operstate = 'STARTED'
547
        vm.save()
548

    
549
        data = json.dumps({'console': {'type': 'vnc'}})
550
        with override_settings(settings, TEST=True):
551
            response = self.mypost('/api/v1.1/servers/%d/action' % vm.id,
552
                                   vm.userid, data, 'json')
553
        self.assertEqual(response.status_code, 200)
554
        reply = json.loads(response.content)
555
        self.assertEqual(reply.keys(), ['console'])
556
        console = reply['console']
557
        self.assertEqual(console['type'], 'vnc')
558
        self.assertEqual(set(console.keys()),
559
                         set(['type', 'host', 'port', 'password']))
560

    
561
    def test_wrong_console_type(self):
562
        """Test console req for ACTIVE server"""
563
        vm = mfactory.VirtualMachineFactory()
564
        vm.operstate = 'STARTED'
565
        vm.save()
566

    
567
        data = json.dumps({'console': {'type': 'foo'}})
568
        response = self.mypost('servers/%d/action' % vm.id,
569
                               vm.userid, data, 'json')
570
        self.assertBadRequest(response)