Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / tests / servers.py @ 7e82ac17

History | View | Annotate | Download (32.5 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
from copy import deepcopy
36

    
37
from snf_django.utils.testing import (BaseAPITest, mocked_quotaholder,
38
                                      override_settings)
39
from synnefo.db.models import (VirtualMachine, VirtualMachineMetadata,
40
                               FloatingIP)
41
from synnefo.db import models_factory as mfactory
42
from synnefo.logic.utils import get_rsapi_state
43
from synnefo.cyclades_settings import cyclades_services
44
from synnefo.lib.services import get_service_path
45
from synnefo.lib import join_urls
46
from django.conf import settings
47

    
48
from mock import patch, Mock
49

    
50

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

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

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

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

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

    
73

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

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

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

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

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

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

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

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

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

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

    
157
    def test_server_fqdn(self):
158
        vm = mfactory.VirtualMachineFactory()
159
        with override_settings(settings,
160
                               CYCLADES_SERVERS_FQDN="vm.example.org"):
161
            response = self.myget("servers/%d" % vm.id, vm.userid)
162
            server = json.loads(response.content)['server']
163
            self.assertEqual(server["SNF:fqdn"], "vm.example.org")
164
        with override_settings(settings,
165
                               CYCLADES_SERVERS_FQDN="snf-%(id)s.vm.example.org"):
166
            response = self.myget("servers/%d" % vm.id, vm.userid)
167
            server = json.loads(response.content)['server']
168
            self.assertEqual(server["SNF:fqdn"], "snf-%d.vm.example.org" % vm.id)
169
        with override_settings(settings,
170
                               CYCLADES_SERVERS_FQDN="snf-%(id)s.vm-%(id)s.example.org"):
171
            response = self.myget("servers/%d" % vm.id, vm.userid)
172
            server = json.loads(response.content)['server']
173
            self.assertEqual(server["SNF:fqdn"], "snf-%d.vm-%d.example.org" %
174
                             (vm.id, vm.id))
175
        # No setting, no NICs
176
        with override_settings(settings,
177
                               CYCLADES_SERVERS_FQDN=None):
178
            response = self.myget("servers/%d" % vm.id, vm.userid)
179
            server = json.loads(response.content)['server']
180
            self.assertEqual(server["SNF:fqdn"], "")
181

    
182
        # IPv6 NIC
183
        nic = mfactory.NetworkInterfaceFactory(machine=vm, ipv4=None,
184
                                               ipv6="babe::", state="ACTIVE",
185
                                               network__public=True)
186
        with override_settings(settings,
187
                               CYCLADES_SERVERS_FQDN=None):
188
            response = self.myget("servers/%d" % vm.id, vm.userid)
189
            server = json.loads(response.content)['server']
190
            self.assertEqual(server["SNF:fqdn"], nic.ipv6)
191

    
192
        # IPv4 NIC
193
        nic = mfactory.NetworkInterfaceFactory(machine=vm, network__public=True,
194
                                      state="ACTIVE")
195
        with override_settings(settings,
196
                               CYCLADES_SERVERS_FQDN=None):
197
            response = self.myget("servers/%d" % vm.id, vm.userid)
198
            server = json.loads(response.content)['server']
199
            self.assertEqual(server["SNF:fqdn"], nic.ipv4)
200

    
201

    
202
    def test_server_building_nics(self):
203
        db_vm = self.vm2
204
        user = self.vm2.userid
205
        net1 = mfactory.NetworkFactory()
206
        net2 = mfactory.NetworkFactory()
207
        net3 = mfactory.NetworkFactory()
208
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net1,
209
                                         state="BUILDING")
210
        nic2 = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net2,
211
                                                state="ACTIVE")
212
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net3,
213
                                         state="BUILDING")
214

    
215
        response = self.myget('servers/%d' % db_vm.id, user)
216
        server = json.loads(response.content)['server']
217
        nics = server["attachments"]
218
        self.assertEqual(len(nics), 1)
219
        self.assertEqual(nics[0]["network_id"], str(nic2.network_id))
220

    
221
    def test_noauthorized(self):
222
        """Test 404 for detail of other user vm"""
223
        db_vm = self.vm2
224

    
225
        response = self.myget('servers/%d' % db_vm.id, 'wrong_user')
226
        self.assertItemNotFound(response)
227

    
228
    def test_wrong_server(self):
229
        """Test 404 response if server does not exist."""
230
        response = self.myget('servers/%d' % 5000)
231
        self.assertItemNotFound(response)
232

    
233
    def test_create_server_empty(self):
234
        """Test if the create server call returns a 400 badRequest if
235
           no attributes are specified."""
236

    
237
        response = self.mypost('servers', params={})
238
        self.assertBadRequest(response)
239

    
240
    def test_rename_server(self):
241
        vm = self.vm2
242
        request = {'server': {'name': 'new_name'}}
243
        response = self.myput('servers/%d' % vm.id, vm.userid,
244
                              json.dumps(request), 'json')
245
        self.assertSuccess(response)
246
        self.assertEqual(VirtualMachine.objects.get(id=vm.id).name, "new_name")
247

    
248
    def test_catch_wrong_api_paths(self):
249
        response = self.myget('nonexistent')
250
        self.assertEqual(response.status_code, 400)
251
        try:
252
            error = json.loads(response.content)
253
        except ValueError:
254
            self.assertTrue(False)
255

    
256
    def test_method_not_allowed(self, *args):
257
        # /servers/ allows only POST, GET
258
        response = self.myput('servers', '', '')
259
        self.assertMethodNotAllowed(response)
260
        response = self.mydelete('servers')
261
        self.assertMethodNotAllowed(response)
262

    
263
        # /servers/<srvid>/ allows only GET, PUT, DELETE
264
        response = self.mypost("servers/42")
265
        self.assertMethodNotAllowed(response)
266

    
267
        # /imags/<srvid>/metadata/ allows only POST, GET
268
        response = self.myput('servers/42/metadata', '', '')
269
        self.assertMethodNotAllowed(response)
270
        response = self.mydelete('servers/42/metadata')
271
        self.assertMethodNotAllowed(response)
272

    
273
        # /imags/<srvid>/metadata/ allows only POST, GET
274
        response = self.myput('servers/42/metadata', '', '')
275
        self.assertMethodNotAllowed(response)
276
        response = self.mydelete('servers/42/metadata')
277
        self.assertMethodNotAllowed(response)
278

    
279
        # /imags/<srvid>/metadata/<key> allows only PUT, GET, DELETE
280
        response = self.mypost('servers/42/metadata/foo')
281
        self.assertMethodNotAllowed(response)
282

    
283

    
284
fixed_image = Mock()
285
fixed_image.return_value = {'location': 'pithos://foo',
286
                            'checksum': '1234',
287
                            "id": 1,
288
                            "name": "test_image",
289
                            'disk_format': 'diskdump'}
290

    
291

    
292
@patch('synnefo.api.util.get_image', fixed_image)
293
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
294
class ServerCreateAPITest(ComputeAPITest):
295
    def setUp(self):
296
        self.flavor = mfactory.FlavorFactory()
297
        # Create public network and backend
298
        self.network = mfactory.NetworkFactory(public=True)
299
        self.backend = mfactory.BackendFactory()
300
        mfactory.BackendNetworkFactory(network=self.network,
301
                                       backend=self.backend,
302
                                       operstate="ACTIVE")
303
        self.request = {
304
                    "server": {
305
                        "name": "new-server-test",
306
                        "userid": "test_user",
307
                        "imageRef": 1,
308
                        "flavorRef": self.flavor.id,
309
                        "metadata": {
310
                            "My Server Name": "Apache1"
311
                        },
312
                        "personality": []
313
                    }
314
        }
315

    
316
    def test_create_server(self, mrapi):
317
        """Test if the create server call returns the expected response
318
           if a valid request has been speficied."""
319

    
320
        mrapi().CreateInstance.return_value = 12
321
        with override_settings(settings, DEFAULT_INSTANCE_NETWORKS=[]):
322
            with mocked_quotaholder():
323
                response = self.mypost('servers', 'test_user',
324
                                       json.dumps(self.request), 'json')
325
        self.assertEqual(response.status_code, 202)
326
        mrapi().CreateInstance.assert_called_once()
327

    
328
        api_server = json.loads(response.content)['server']
329
        self.assertEqual(api_server['status'], "BUILD")
330
        self.assertEqual(api_server['progress'], 0)
331
        self.assertEqual(api_server['metadata'],
332
                        {"My Server Name":  "Apache1"})
333
        self.assertTrue('adminPass' in api_server)
334

    
335
        db_vm = VirtualMachine.objects.get(userid='test_user')
336
        self.assertEqual(api_server['name'], db_vm.name)
337
        self.assertEqual(api_server['status'], db_vm.operstate)
338

    
339
        # Test drained flag in Network:
340
        self.network.drained = True
341
        self.network.save()
342
        with mocked_quotaholder():
343
            response = self.mypost('servers', 'test_user',
344
                                   json.dumps(self.request), 'json')
345
        self.assertEqual(response.status_code, 503, "serviceUnavailable")
346

    
347
    def test_create_network_settings(self, mrapi):
348
        mrapi().CreateInstance.return_value = 12
349
        bnet1 = mfactory.BackendNetworkFactory(operstate="ACTIVE",
350
                                               backend=self.backend)
351
        bnet2 = mfactory.BackendNetworkFactory(operstate="ACTIVE",
352
                                               backend=self.backend)
353
        bnet3 = mfactory.BackendNetworkFactory(network__userid="test_user",
354
                                               operstate="ACTIVE",
355
                                               backend=self.backend)
356
        bnet4 = mfactory.BackendNetworkFactory(network__userid="test_user",
357
                                               operstate="ACTIVE",
358
                                               backend=self.backend)
359
        # User requested private networks
360
        request = deepcopy(self.request)
361
        request["server"]["networks"] = [bnet3.network.id, bnet4.network.id]
362
        with override_settings(settings,
363
                DEFAULT_INSTANCE_NETWORKS=["SNF:ANY_PUBLIC", bnet1.network.id,
364
                                           bnet2.network.id]):
365
            with mocked_quotaholder():
366
                response = self.mypost('servers', 'test_user',
367
                                       json.dumps(request), 'json')
368
        self.assertEqual(response.status_code, 202)
369
        name, args, kwargs = mrapi().CreateInstance.mock_calls[0]
370
        self.assertEqual(len(kwargs["nics"]), 5)
371
        self.assertEqual(kwargs["nics"][0]["network"],
372
                         self.network.backend_id)
373
        self.assertEqual(kwargs["nics"][1]["network"],
374
                         bnet1.network.backend_id)
375
        self.assertEqual(kwargs["nics"][2]["network"],
376
                         bnet2.network.backend_id)
377
        self.assertEqual(kwargs["nics"][3]["network"],
378
                         bnet3.network.backend_id)
379
        self.assertEqual(kwargs["nics"][4]["network"],
380
                         bnet4.network.backend_id)
381

    
382
        request["server"]["floating_ips"] = []
383
        with override_settings(settings,
384
                DEFAULT_INSTANCE_NETWORKS=[bnet2.network.id]):
385
            with mocked_quotaholder():
386
                response = self.mypost('servers', 'test_user',
387
                                       json.dumps(request), 'json')
388
        self.assertEqual(response.status_code, 202)
389
        name, args, kwargs = mrapi().CreateInstance.mock_calls[1]
390
        self.assertEqual(len(kwargs["nics"]), 3)
391
        self.assertEqual(kwargs["nics"][0]["network"],
392
                         bnet2.network.backend_id)
393
        self.assertEqual(kwargs["nics"][1]["network"],
394
                         bnet3.network.backend_id)
395
        self.assertEqual(kwargs["nics"][2]["network"],
396
                         bnet4.network.backend_id)
397

    
398
        # test invalid network in DEFAULT_INSTANCE_NETWORKS
399
        with override_settings(settings, DEFAULT_INSTANCE_NETWORKS=[42]):
400
            response = self.mypost('servers', 'test_user',
401
                                   json.dumps(request), 'json')
402
        self.assertFault(response, 500, "internalServerError")
403

    
404
        # test connect to public netwok
405
        request = deepcopy(self.request)
406
        request["server"]["networks"] = [self.network.id]
407
        with override_settings(settings,
408
                               DEFAULT_INSTANCE_NETWORKS=["SNF:ANY_PUBLIC"]):
409
            response = self.mypost('servers', 'test_user',
410
                                    json.dumps(request), 'json')
411
        self.assertFault(response, 403, "forbidden")
412
        # test wrong user
413
        request = deepcopy(self.request)
414
        request["server"]["networks"] = [bnet3.network.id]
415
        with override_settings(settings,
416
                               DEFAULT_INSTANCE_NETWORKS=["SNF:ANY_PUBLIC"]):
417
            with mocked_quotaholder():
418
                response = self.mypost('servers', 'dummy_user',
419
                                       json.dumps(request), 'json')
420
        self.assertItemNotFound(response)
421

    
422
        # Test floating IPs
423
        request = deepcopy(self.request)
424
        request["server"]["networks"] = [bnet4.network.id]
425
        network = mfactory.NetworkFactory(subnet="10.0.0.0/24")
426
        mfactory.BackendNetworkFactory(network=network,
427
                                       backend=self.backend,
428
                                       operstate="ACTIVE")
429
        fp1 = mfactory.FloatingIPFactory(ipv4="10.0.0.2",
430
                                         userid="test_user",
431
                                         network=network, machine=None)
432
        fp2 = mfactory.FloatingIPFactory(ipv4="10.0.0.3", network=network,
433
                                         userid="test_user",
434
                                         machine=None)
435
        request["server"]["floating_ips"] = [fp1.ipv4, fp2.ipv4]
436
        with override_settings(settings,
437
                DEFAULT_INSTANCE_NETWORKS=[bnet3.network.id]):
438
            with mocked_quotaholder():
439
                response = self.mypost('servers', 'test_user',
440
                                     json.dumps(request), 'json')
441
        self.assertEqual(response.status_code, 202)
442
        api_server = json.loads(response.content)['server']
443
        vm = VirtualMachine.objects.get(id=api_server["id"])
444
        fp1 = FloatingIP.objects.get(id=fp1.id)
445
        fp2 = FloatingIP.objects.get(id=fp2.id)
446
        self.assertEqual(fp1.machine, vm)
447
        self.assertEqual(fp2.machine, vm)
448
        name, args, kwargs = mrapi().CreateInstance.mock_calls[2]
449
        self.assertEqual(len(kwargs["nics"]), 4)
450
        self.assertEqual(kwargs["nics"][0]["network"],
451
                         bnet3.network.backend_id)
452
        self.assertEqual(kwargs["nics"][1]["network"], network.backend_id)
453
        self.assertEqual(kwargs["nics"][1]["ip"], fp1.ipv4)
454
        self.assertEqual(kwargs["nics"][2]["network"], network.backend_id)
455
        self.assertEqual(kwargs["nics"][2]["ip"], fp2.ipv4)
456
        self.assertEqual(kwargs["nics"][3]["network"],
457
                         bnet4.network.backend_id)
458

    
459
    def test_create_server_no_flavor(self, mrapi):
460
        request = deepcopy(self.request)
461
        request["server"]["flavorRef"] = 42
462
        with mocked_quotaholder():
463
            response = self.mypost('servers', 'test_user',
464
                                   json.dumps(request), 'json')
465
        self.assertItemNotFound(response)
466

    
467

    
468
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
469
class ServerDestroyAPITest(ComputeAPITest):
470
    def test_delete_server(self, mrapi):
471
        vm = mfactory.VirtualMachineFactory()
472
        mrapi().DeleteInstance.return_value = 12
473
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
474
        self.assertEqual(response.status_code, 204)
475
        mrapi().DeleteInstance.assert_called_once()
476

    
477
    def test_non_existing_delete_server(self, mrapi):
478
        vm = mfactory.VirtualMachineFactory()
479
        response = self.mydelete('servers/%d' % 42, vm.userid)
480
        self.assertItemNotFound(response)
481
        self.assertFalse(mrapi.mock_calls)
482

    
483

    
484
class ServerMetadataAPITest(ComputeAPITest):
485
    def setUp(self):
486
        self.vm = mfactory.VirtualMachineFactory()
487
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
488
        super(ServerMetadataAPITest, self).setUp()
489

    
490
    def test_get_metadata(self):
491
        vm = self.vm
492
        create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
493
        metadata = [create_meta(), create_meta(), create_meta()]
494
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
495
        self.assertTrue(response.status_code in [200, 203])
496
        api_metadata = json.loads(response.content)['metadata']
497
        self.assertEqual(len(api_metadata), len(metadata) + 1)
498
        for db_m in metadata:
499
            self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
500

    
501
        request = {'metadata':
502
                        {'foo': 'bar'},
503
                        metadata[0].meta_key: 'bar2'
504
                  }
505
        response = self.mypost('servers/%d/metadata' % vm.id,
506
                             vm.userid, json.dumps(request), 'json')
507
        metadata2 = VirtualMachineMetadata.objects.filter(vm=vm)
508
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
509
        self.assertTrue(response.status_code in [200, 203])
510
        api_metadata2 = json.loads(response.content)['metadata']
511
        self.assertTrue('foo' in api_metadata2.keys())
512
        self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2')
513
        self.assertEqual(len(api_metadata2), len(metadata2))
514
        for db_m in metadata2:
515
            self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
516

    
517
        # Create new meta
518
        request = {'meta': {'foo2': 'bar2'}}
519
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
520
                              vm.userid, json.dumps(request), 'json')
521

    
522
        # Get the new meta
523
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
524
        meta = json.loads(response.content)['meta']
525
        self.assertEqual(meta['foo2'], 'bar2')
526

    
527
        # Delete the new meta
528
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
529
        self.assertEqual(response.status_code, 204)
530

    
531
        # Try to get the deleted meta: should raise 404
532
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
533
        self.assertEqual(response.status_code, 404)
534

    
535
    def test_invalid_metadata(self):
536
        vm = self.vm
537
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
538
        self.assertBadRequest(response)
539
        self.assertEqual(len(vm.metadata.all()), 1)
540

    
541
    def test_invalid_metadata_server(self):
542
        response = self.mypost('servers/42/metadata', 'user')
543
        self.assertItemNotFound(response)
544

    
545
    def test_get_meta_invalid_key(self):
546
        vm = self.vm
547
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
548
        self.assertItemNotFound(response)
549

    
550

    
551
@patch('synnefo.api.util.get_image')
552
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
553
class ServerActionAPITest(ComputeAPITest):
554
    def test_actions(self, mrapi, mimage):
555
        actions = ['start', 'shutdown', 'reboot']
556
        vm = mfactory.VirtualMachineFactory()
557
        vm.operstate = "STOPPED"
558
        vm.save()
559
        mrapi().StartupInstance.return_value = 0
560
        mrapi().ShutdownInstance.return_value = 1
561
        mrapi().RebootInstance.return_value = 2
562
        for jobId, action in enumerate(actions):
563
            if action in ["shutdown", "reboot"]:
564
                vm.operstate = "STARTED"
565
            else:
566
                vm.operstate = "STOPPED"
567
            vm.task = None
568
            vm.task_job_id = None
569
            vm.save()
570
            val = {'type': 'HARD'} if action == 'reboot' else {}
571
            request = {action: val}
572
            response = self.mypost('servers/%d/action' % vm.id,
573
                                   vm.userid, json.dumps(request), 'json')
574
            self.assertEqual(response.status_code, 202)
575
            if action == 'shutdown':
576
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
577
                                 "STOP")
578
            else:
579
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
580
                                 action.upper())
581
            self.assertEqual(VirtualMachine.objects.get(id=vm.id).task_job_id,
582
                             jobId)
583

    
584
    def test_action_in_building_vm(self, mrapi, mimage):
585
        """Test building in progress"""
586
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
587
        request = {'start': {}}
588
        with mocked_quotaholder():
589
            response = self.mypost('servers/%d/action' % vm.id,
590
                                   vm.userid, json.dumps(request), 'json')
591
        self.assertEqual(response.status_code, 409)
592
        self.assertFalse(mrapi.mock_calls)
593

    
594
    def test_destroy_build_vm(self, mrapi, mimage):
595
        """Test building in progress"""
596
        vm = mfactory.VirtualMachineFactory()
597
        mrapi().DeleteInstance.return_value = 2
598
        response = self.mydelete('servers/%d' % vm.id,
599
                                 vm.userid)
600
        self.assertSuccess(response)
601
        mrapi().RemoveInstance.assert_called_once()
602

    
603
    def test_firewall(self, mrapi, mimage):
604
        vm = mfactory.VirtualMachineFactory()
605
        vm.operstate = "STOPPED"
606
        vm.save()
607
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
608
        response = self.mypost('servers/%d/action' % vm.id,
609
                               vm.userid, json.dumps(request), 'json')
610
        self.assertEqual(response.status_code, 202)
611
        mrapi().ModifyInstance.assert_called_once()
612

    
613
    def test_unsupported_firewall(self, mrapi, mimage):
614
        vm = mfactory.VirtualMachineFactory()
615
        vm.operstate = "STOPPED"
616
        vm.save()
617
        request = {'firewallProfile': {'profile': 'FOO'}}
618
        response = self.mypost('servers/%d/action' % vm.id,
619
                               vm.userid, json.dumps(request), 'json')
620
        self.assertBadRequest(response)
621
        self.assertFalse(mrapi.mock_calls)
622

    
623
    def test_resize_vm(self, mrapi, mimage):
624
        flavor = mfactory.FlavorFactory(cpu=1, ram=1024)
625
        # Check building VM
626
        vm = self.get_vm(flavor=flavor, operstate="BUILD")
627
        request = {'resize': {'flavorRef': flavor.id}}
628
        response = self.mypost('servers/%d/action' % vm.id,
629
                             vm.userid, json.dumps(request), 'json')
630
        self.assertFault(response, 409, "buildInProgress")
631
        # Check same Flavor
632
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
633
        request = {'resize': {'flavorRef': flavor.id}}
634
        response = self.mypost('servers/%d/action' % vm.id,
635
                             vm.userid, json.dumps(request), 'json')
636
        self.assertBadRequest(response)
637
        # Check flavor with different disk
638
        flavor2 = mfactory.FlavorFactory(disk=1024)
639
        flavor3 = mfactory.FlavorFactory(disk=2048)
640
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
641
        request = {'resize': {'flavorRef': flavor3.id}}
642
        response = self.mypost('servers/%d/action' % vm.id,
643
                             vm.userid, json.dumps(request), 'json')
644
        self.assertBadRequest(response)
645
        flavor2 = mfactory.FlavorFactory(disk_template="foo")
646
        flavor3 = mfactory.FlavorFactory(disk_template="baz")
647
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
648
        request = {'resize': {'flavorRef': flavor3.id}}
649
        response = self.mypost('servers/%d/action' % vm.id,
650
                             vm.userid, json.dumps(request), 'json')
651
        self.assertBadRequest(response)
652
        # Check success
653
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
654
        flavor4 = mfactory.FlavorFactory(disk_template=flavor.disk_template,
655
                                         disk=flavor.disk,
656
                                         cpu=4, ram=2048)
657
        request = {'resize': {'flavorRef': flavor4.id}}
658
        mrapi().ModifyInstance.return_value = 42
659
        response = self.mypost('servers/%d/action' % vm.id,
660
                             vm.userid, json.dumps(request), 'json')
661
        self.assertEqual(response.status_code, 202)
662
        vm = VirtualMachine.objects.get(id=vm.id)
663
        self.assertEqual(vm.task_job_id, 42)
664
        name, args, kwargs = mrapi().ModifyInstance.mock_calls[0]
665
        self.assertEqual(kwargs["beparams"]["vcpus"], 4)
666
        self.assertEqual(kwargs["beparams"]["minmem"], 2048)
667
        self.assertEqual(kwargs["beparams"]["maxmem"], 2048)
668

    
669
    def test_action_on_resizing_vm(self, mrapi, mimage):
670
        vm = mfactory.VirtualMachineFactory()
671
        vm.operstate = "RESIZE"
672
        vm.save()
673
        for action in VirtualMachine.ACTIONS:
674
            request = {action[0]: ""}
675
            response = self.mypost('servers/%d/action' % vm.id,
676
                                 vm.userid, json.dumps(request), 'json')
677
            self.assertBadRequest(response)
678
        # however you can destroy
679
        mrapi().DeleteInstance.return_value = 42
680
        response = self.mydelete('servers/%d' % vm.id,
681
                                 vm.userid)
682
        self.assertSuccess(response)
683

    
684
    def get_vm(self, flavor, operstate):
685
        vm = mfactory.VirtualMachineFactory(flavor=flavor)
686
        vm.operstate = operstate
687
        vm.backendjobstatus = "success"
688
        vm.save()
689
        return vm
690

    
691

    
692
class ServerVNCConsole(ComputeAPITest):
693
    def test_not_active_server(self):
694
        """Test console req for not ACTIVE server returns badRequest"""
695
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
696
        data = json.dumps({'console': {'type': 'vnc'}})
697
        response = self.mypost('servers/%d/action' % vm.id,
698
                               vm.userid, data, 'json')
699
        self.assertBadRequest(response)
700

    
701
    def test_active_server(self):
702
        """Test console req for ACTIVE server"""
703
        vm = mfactory.VirtualMachineFactory()
704
        vm.operstate = 'STARTED'
705
        vm.save()
706

    
707
        data = json.dumps({'console': {'type': 'vnc'}})
708
        with override_settings(settings, TEST=True):
709
            response = self.mypost('servers/%d/action' % vm.id,
710
                                   vm.userid, data, 'json')
711
        self.assertEqual(response.status_code, 200)
712
        reply = json.loads(response.content)
713
        self.assertEqual(reply.keys(), ['console'])
714
        console = reply['console']
715
        self.assertEqual(console['type'], 'vnc')
716
        self.assertEqual(set(console.keys()),
717
                         set(['type', 'host', 'port', 'password']))
718

    
719
    def test_wrong_console_type(self):
720
        """Test console req for ACTIVE server"""
721
        vm = mfactory.VirtualMachineFactory()
722
        vm.operstate = 'STARTED'
723
        vm.save()
724

    
725
        data = json.dumps({'console': {'type': 'foo'}})
726
        response = self.mypost('servers/%d/action' % vm.id,
727
                               vm.userid, data, 'json')
728
        self.assertBadRequest(response)