Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (22.7 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
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
from synnefo.cyclades_settings import cyclades_services
41
from synnefo.lib.services import get_service_path
42
from synnefo.lib import join_urls
43

    
44
from mock import patch
45

    
46

    
47
class ComputeAPITest(BaseAPITest):
48
    def setUp(self, *args, **kwargs):
49
        super(ComputeAPITest, self).setUp(*args, **kwargs)
50
        self.compute_path = get_service_path(cyclades_services, 'compute',
51
                                             version='v2.0')
52
    def myget(self, path, *args, **kwargs):
53
        path = join_urls(self.compute_path, path)
54
        return self.get(path, *args, **kwargs)
55

    
56
    def myput(self, path, *args, **kwargs):
57
        path = join_urls(self.compute_path, path)
58
        return self.put(path, *args, **kwargs)
59

    
60
    def mypost(self, path, *args, **kwargs):
61
        path = join_urls(self.compute_path, path)
62
        return self.post(path, *args, **kwargs)
63

    
64
    def mydelete(self, path, *args, **kwargs):
65
        path = join_urls(self.compute_path, path)
66
        return self.delete(path, *args, **kwargs)
67

    
68

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

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

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

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

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

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

    
124
        db_vm_meta = mfactory.VirtualMachineMetadataFactory(vm=db_vm)
125

    
126
        response = self.myget('servers/%d' % db_vm.id, user)
127
        server = json.loads(response.content)['server']
128

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

    
147
        metadata = server['metadata']
148
        self.assertEqual(len(metadata), 1)
149
        self.assertEqual(metadata[db_vm_meta.meta_key], db_vm_meta.meta_value)
150
        self.assertSuccess(response)
151

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

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

    
171
    def test_noauthorized(self):
172
        """Test 404 for detail of other user vm"""
173
        db_vm = self.vm2
174

    
175
        response = self.myget('servers/%d' % db_vm.id, 'wrong_user')
176
        self.assertItemNotFound(response)
177

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

    
183
    def test_create_server_empty(self):
184
        """Test if the create server call returns a 400 badRequest if
185
           no attributes are specified."""
186

    
187
        response = self.mypost('servers', params={})
188
        self.assertBadRequest(response)
189

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

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

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

    
213
        # /servers/<srvid>/ allows only GET, PUT, DELETE
214
        response = self.mypost("servers/42")
215
        self.assertMethodNotAllowed(response)
216

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

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

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

    
233

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

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

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

    
277
        db_vm = VirtualMachine.objects.get(userid='test_user')
278
        self.assertEqual(api_server['name'], db_vm.name)
279
        self.assertEqual(api_server['status'], db_vm.operstate)
280

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

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

    
306

    
307
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
308
class ServerDestroyAPITest(ComputeAPITest):
309
    def test_delete_server(self, mrapi):
310
        vm = mfactory.VirtualMachineFactory()
311
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
312
        self.assertEqual(response.status_code, 204)
313
        mrapi().DeleteInstance.assert_called_once()
314

    
315
    def test_non_existing_delete_server(self, mrapi):
316
        vm = mfactory.VirtualMachineFactory()
317
        response = self.mydelete('servers/%d' % 42, vm.userid)
318
        self.assertItemNotFound(response)
319
        self.assertFalse(mrapi.mock_calls)
320

    
321

    
322
class ServerMetadataAPITest(ComputeAPITest):
323
    def setUp(self):
324
        self.vm = mfactory.VirtualMachineFactory()
325
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
326
        super(ServerMetadataAPITest, self).setUp()
327

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

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

    
355
        # Create new meta
356
        request = {'meta': {'foo2': 'bar2'}}
357
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
358
                              vm.userid, json.dumps(request), 'json')
359

    
360
        # Get the new meta
361
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
362
        meta = json.loads(response.content)['meta']
363
        self.assertEqual(meta['foo2'], 'bar2')
364

    
365
        # Delete the new meta
366
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
367
        self.assertEqual(response.status_code, 204)
368

    
369
        # Try to get the deleted meta: should raise 404
370
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
371
        self.assertEqual(response.status_code, 404)
372

    
373
    def test_invalid_metadata(self):
374
        vm = self.vm
375
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
376
        self.assertBadRequest(response)
377
        self.assertEqual(len(vm.metadata.all()), 1)
378

    
379
    def test_invalid_metadata_server(self):
380
        response = self.mypost('servers/42/metadata', 'user')
381
        self.assertItemNotFound(response)
382

    
383
    def test_get_meta_invalid_key(self):
384
        vm = self.vm
385
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
386
        self.assertItemNotFound(response)
387

    
388

    
389
@patch('synnefo.api.util.get_image')
390
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
391
class ServerActionAPITest(ComputeAPITest):
392
    def test_actions(self, mrapi, mimage):
393
        actions = ['start', 'shutdown', 'reboot']
394
        vm = mfactory.VirtualMachineFactory()
395
        vm.operstate = "STOPPED"
396
        vm.save()
397
        for action in actions:
398
            val = {'type': 'HARD'} if action == 'reboot' else {}
399
            request = {action: val}
400
            response = self.mypost('servers/%d/action' % vm.id,
401
                                   vm.userid, json.dumps(request), 'json')
402
            self.assertEqual(response.status_code, 202)
403
            if action == 'shutdown':
404
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
405
                                 "STOP")
406
            else:
407
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
408
                                 action.upper())
409

    
410
    def test_action_in_building_vm(self, mrapi, mimage):
411
        """Test building in progress"""
412
        vm = mfactory.VirtualMachineFactory()
413
        request = {'start': '{}'}
414
        response = self.mypost('servers/%d/action' % vm.id,
415
                               vm.userid, json.dumps(request), 'json')
416
        self.assertEqual(response.status_code, 409)
417
        self.assertFalse(mrapi.mock_calls)
418

    
419
    def test_destroy_build_vm(self, mrapi, mimage):
420
        """Test building in progress"""
421
        vm = mfactory.VirtualMachineFactory()
422
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
423
        self.assertSuccess(response)
424
        mrapi().RemoveInstance.assert_called_once()
425

    
426
    def test_firewall(self, mrapi, mimage):
427
        vm = mfactory.VirtualMachineFactory()
428
        vm.operstate = "STOPPED"
429
        vm.save()
430
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
431
        response = self.mypost('servers/%d/action' % vm.id,
432
                               vm.userid, json.dumps(request), 'json')
433
        self.assertEqual(response.status_code, 202)
434
        mrapi().ModifyInstance.assert_called_once()
435

    
436
    def test_unsupported_firewall(self, mrapi, mimage):
437
        vm = mfactory.VirtualMachineFactory()
438
        vm.operstate = "STOPPED"
439
        vm.save()
440
        request = {'firewallProfile': {'profile': 'FOO'}}
441
        response = self.mypost('servers/%d/action' % vm.id,
442
                               vm.userid, json.dumps(request), 'json')
443
        self.assertBadRequest(response)
444
        self.assertFalse(mrapi.mock_calls)
445

    
446
    def test_resize_vm(self, mrapi, mimage):
447
        flavor = mfactory.FlavorFactory(cpu=1, ram=1024)
448
        # Check building VM
449
        vm = self.get_vm(flavor=flavor, operstate="BUILD")
450
        request = {'resize': {'flavorRef': flavor.id}}
451
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
452
                             vm.userid, json.dumps(request), 'json')
453
        self.assertFault(response, 409, "buildInProgress")
454
        # Check same Flavor
455
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
456
        request = {'resize': {'flavorRef': flavor.id}}
457
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
458
                             vm.userid, json.dumps(request), 'json')
459
        self.assertSuccess(response)
460
        # Check flavor with different disk
461
        flavor2 = mfactory.FlavorFactory(disk=1024)
462
        flavor3 = mfactory.FlavorFactory(disk=2048)
463
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
464
        request = {'resize': {'flavorRef': flavor3.id}}
465
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
466
                             vm.userid, json.dumps(request), 'json')
467
        self.assertBadRequest(response)
468
        flavor2 = mfactory.FlavorFactory(disk_template="foo")
469
        flavor3 = mfactory.FlavorFactory(disk_template="baz")
470
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
471
        request = {'resize': {'flavorRef': flavor3.id}}
472
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
473
                             vm.userid, json.dumps(request), 'json')
474
        self.assertBadRequest(response)
475
        # Check success
476
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
477
        flavor4 = mfactory.FlavorFactory(disk_template=flavor.disk_template,
478
                                         disk=flavor.disk,
479
                                         cpu=4, ram=2048)
480
        request = {'resize': {'flavorRef': flavor4.id}}
481
        mrapi().ModifyInstance.return_value = 42
482
        response = self.post('/api/v1.1/servers/%d/action' % vm.id,
483
                             vm.userid, json.dumps(request), 'json')
484
        self.assertEqual(response.status_code, 202)
485
        vm = VirtualMachine.objects.get(id=vm.id)
486
        self.assertEqual(vm.backendjobid, 42)
487
        name, args, kwargs = mrapi().ModifyInstance.mock_calls[0]
488
        self.assertEqual(kwargs["beparams"]["vcpus"], 4)
489
        self.assertEqual(kwargs["beparams"]["minmem"], 2048)
490
        self.assertEqual(kwargs["beparams"]["maxmem"], 2048)
491

    
492
    def get_vm(self, flavor, operstate):
493
        vm = mfactory.VirtualMachineFactory(flavor=flavor)
494
        vm.operstate = operstate
495
        vm.backendjobstatus = "success"
496
        vm.save()
497
        return vm
498

    
499

    
500
class ServerVNCConsole(ComputeAPITest):
501
    def test_not_active_server(self):
502
        """Test console req for not ACTIVE server returns badRequest"""
503
        vm = mfactory.VirtualMachineFactory()
504
        data = json.dumps({'console': {'type': 'vnc'}})
505
        response = self.mypost('servers/%d/action' % vm.id,
506
                               vm.userid, data, 'json')
507
        self.assertBadRequest(response)
508

    
509
    def test_active_server(self):
510
        """Test console req for ACTIVE server"""
511
        vm = mfactory.VirtualMachineFactory()
512
        vm.operstate = 'STARTED'
513
        vm.save()
514

    
515
        data = json.dumps({'console': {'type': 'vnc'}})
516
        response = self.mypost('servers/%d/action' % vm.id,
517
                               vm.userid, data, 'json')
518
        self.assertEqual(response.status_code, 200)
519
        reply = json.loads(response.content)
520
        self.assertEqual(reply.keys(), ['console'])
521
        console = reply['console']
522
        self.assertEqual(console['type'], 'vnc')
523
        self.assertEqual(set(console.keys()),
524
                         set(['type', 'host', 'port', 'password']))
525

    
526
    def test_wrong_console_type(self):
527
        """Test console req for ACTIVE server"""
528
        vm = mfactory.VirtualMachineFactory()
529
        vm.operstate = 'STARTED'
530
        vm.save()
531

    
532
        data = json.dumps({'console': {'type': 'foo'}})
533
        response = self.mypost('servers/%d/action' % vm.id,
534
                               vm.userid, data, 'json')
535
        self.assertBadRequest(response)