Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / tests / servers.py @ 92d2d1ce

History | View | Annotate | Download (35.8 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
                               IPAddress)
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
from synnefo.logic.rapi import GanetiApiError
48

    
49
from mock import patch, Mock
50

    
51

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

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

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

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

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

    
74

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

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

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

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

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

    
121
    def test_server_detail(self):
122
        """Test if a server details are returned."""
123
        db_vm = self.vm2
124
        user = self.vm2.userid
125
        ip4 = mfactory.IPv4AddressFactory(nic__machine=self.vm2)
126
        nic = ip4.nic
127
        net = ip4.network
128
        ip6 = mfactory.IPv6AddressFactory(nic=nic, network=net)
129
        nic.mac = "aa:00:11:22:33:44"
130
        nic.save()
131

    
132
        db_vm_meta = mfactory.VirtualMachineMetadataFactory(vm=db_vm)
133

    
134
        response = self.myget('servers/%d' % db_vm.id, user)
135
        server = json.loads(response.content)['server']
136

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

    
157
        metadata = server['metadata']
158
        self.assertEqual(len(metadata), 1)
159
        self.assertEqual(metadata[db_vm_meta.meta_key], db_vm_meta.meta_value)
160
        self.assertSuccess(response)
161

    
162
    def test_server_fqdn(self):
163
        vm = mfactory.VirtualMachineFactory()
164
        with override_settings(settings,
165
                               CYCLADES_SERVERS_FQDN="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"], "vm.example.org")
169
        with override_settings(settings, CYCLADES_SERVERS_FQDN=
170
                               "snf-%(id)s.vm.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"],
174
                             "snf-%d.vm.example.org" % vm.id)
175
        with override_settings(settings,
176
                               CYCLADES_SERVERS_FQDN=
177
                               "snf-%(id)s.vm-%(id)s.example.org"):
178
            response = self.myget("servers/%d" % vm.id, vm.userid)
179
            server = json.loads(response.content)['server']
180
            self.assertEqual(server["SNF:fqdn"], "snf-%d.vm-%d.example.org" %
181
                             (vm.id, vm.id))
182
        # No setting, no NICs
183
        with override_settings(settings,
184
                               CYCLADES_SERVERS_FQDN=None):
185
            response = self.myget("servers/%d" % vm.id, vm.userid)
186
            server = json.loads(response.content)['server']
187
            self.assertEqual(server["SNF:fqdn"], "")
188

    
189
        # IPv6 NIC
190
        ipv6_address = mfactory.IPv6AddressFactory(nic__machine=vm,
191
                                                   network__public=True)
192
        with override_settings(settings,
193
                               CYCLADES_SERVERS_FQDN=None):
194
            response = self.myget("servers/%d" % vm.id, vm.userid)
195
            server = json.loads(response.content)['server']
196
            self.assertEqual(server["SNF:fqdn"], ipv6_address.address)
197

    
198
        # IPv4 NIC
199
        ipv4_address = mfactory.IPv4AddressFactory(nic__machine=vm,
200
                                                   network__public=True)
201
        with override_settings(settings,
202
                               CYCLADES_SERVERS_FQDN=None):
203
            response = self.myget("servers/%d" % vm.id, vm.userid)
204
            server = json.loads(response.content)['server']
205
            self.assertEqual(server["SNF:fqdn"], ipv4_address.address)
206

    
207
    def test_server_port_forwarding(self):
208
        vm = mfactory.VirtualMachineFactory()
209
        ports = {
210
            22: ("foo", 61000),
211
            80: lambda ip, id, fqdn, user: ("bar", 61001)}
212
        with override_settings(settings,
213
                               CYCLADES_PORT_FORWARDING=ports):
214
            response = self.myget("servers/%d" % vm.id, vm.userid)
215
            server = json.loads(response.content)['server']
216
            self.assertEqual(server["SNF:port_forwarding"],
217
                             {"22": {"host": "foo", "port": "61000"},
218
                              "80": {"host": "bar", "port": "61001"}})
219

    
220
        def _port_from_ip(ip, base):
221
            fields = ip.split('.', 4)
222
            return (base + 256*int(fields[2]) + int(fields[3]))
223

    
224
        ports = {
225
            22: lambda ip, id, fqdn, user:
226
            ip and ("gate", _port_from_ip(ip, 10000)) or None}
227
        with override_settings(settings,
228
                               CYCLADES_PORT_FORWARDING=ports):
229
            response = self.myget("servers/%d" % vm.id, vm.userid)
230
            server = json.loads(response.content)['server']
231
            self.assertEqual(server["SNF:port_forwarding"], {})
232

    
233
        mfactory.IPv4AddressFactory(nic__machine=vm,
234
                                    network__public=True,
235
                                    address="192.168.2.2")
236
        with override_settings(settings,
237
                               CYCLADES_PORT_FORWARDING=ports):
238
            response = self.myget("servers/%d" % vm.id, vm.userid)
239
            server = json.loads(response.content)['server']
240
            self.assertEqual(server["SNF:port_forwarding"],
241
                             {"22": {"host": "gate", "port": "10514"}})
242

    
243
    def test_server_building_nics(self):
244
        db_vm = self.vm2
245
        user = self.vm2.userid
246
        net1 = mfactory.NetworkFactory()
247
        net2 = mfactory.NetworkFactory()
248
        net3 = mfactory.NetworkFactory()
249
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net1,
250
                                         state="BUILDING")
251
        nic2 = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net2,
252
                                                state="ACTIVE")
253
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net3,
254
                                         state="BUILDING")
255

    
256
        response = self.myget('servers/%d' % db_vm.id, user)
257
        server = json.loads(response.content)['server']
258
        nics = server["attachments"]
259
        self.assertEqual(len(nics), 1)
260
        self.assertEqual(nics[0]["network_id"], str(nic2.network_id))
261

    
262
    def test_noauthorized(self):
263
        """Test 404 for detail of other user vm"""
264
        db_vm = self.vm2
265

    
266
        response = self.myget('servers/%d' % db_vm.id, 'wrong_user')
267
        self.assertItemNotFound(response)
268

    
269
    def test_wrong_server(self):
270
        """Test 404 response if server does not exist."""
271
        response = self.myget('servers/%d' % 5000)
272
        self.assertItemNotFound(response)
273

    
274
    def test_create_server_empty(self):
275
        """Test if the create server call returns a 400 badRequest if
276
           no attributes are specified."""
277

    
278
        response = self.mypost('servers', params={})
279
        self.assertBadRequest(response)
280

    
281
    def test_rename_server(self):
282
        vm = self.vm2
283
        request = {'server': {'name': 'new_name'}}
284
        response = self.myput('servers/%d' % vm.id, vm.userid,
285
                              json.dumps(request), 'json')
286
        self.assertSuccess(response)
287
        self.assertEqual(VirtualMachine.objects.get(id=vm.id).name, "new_name")
288

    
289
    def test_catch_wrong_api_paths(self):
290
        response = self.myget('nonexistent')
291
        self.assertEqual(response.status_code, 400)
292
        try:
293
            json.loads(response.content)
294
        except ValueError:
295
            self.assertTrue(False)
296

    
297
    def test_method_not_allowed(self, *args):
298
        # /servers/ allows only POST, GET
299
        response = self.myput('servers', '', '')
300
        self.assertMethodNotAllowed(response)
301
        response = self.mydelete('servers')
302
        self.assertMethodNotAllowed(response)
303

    
304
        # /servers/<srvid>/ allows only GET, PUT, DELETE
305
        response = self.mypost("servers/42")
306
        self.assertMethodNotAllowed(response)
307

    
308
        # /imags/<srvid>/metadata/ allows only POST, GET
309
        response = self.myput('servers/42/metadata', '', '')
310
        self.assertMethodNotAllowed(response)
311
        response = self.mydelete('servers/42/metadata')
312
        self.assertMethodNotAllowed(response)
313

    
314
        # /imags/<srvid>/metadata/ allows only POST, GET
315
        response = self.myput('servers/42/metadata', '', '')
316
        self.assertMethodNotAllowed(response)
317
        response = self.mydelete('servers/42/metadata')
318
        self.assertMethodNotAllowed(response)
319

    
320
        # /imags/<srvid>/metadata/<key> allows only PUT, GET, DELETE
321
        response = self.mypost('servers/42/metadata/foo')
322
        self.assertMethodNotAllowed(response)
323

    
324

    
325
fixed_image = Mock()
326
fixed_image.return_value = {'location': 'pithos://foo',
327
                            'checksum': '1234',
328
                            "id": 1,
329
                            "name": "test_image",
330
                            "size": "41242",
331
                            'disk_format': 'diskdump'}
332

    
333

    
334
@patch('synnefo.api.util.get_image', fixed_image)
335
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
336
class ServerCreateAPITest(ComputeAPITest):
337
    def setUp(self):
338
        self.flavor = mfactory.FlavorFactory()
339
        # Create public network and backend
340
        subnet = mfactory.IPv4SubnetFactory(network__public=True)
341
        self.network = subnet.network
342
        self.backend = mfactory.BackendFactory()
343
        mfactory.BackendNetworkFactory(network=self.network,
344
                                       backend=self.backend,
345
                                       operstate="ACTIVE")
346
        self.request = {
347
            "server": {
348
                "name": "new-server-test",
349
                "userid": "test_user",
350
                "imageRef": 1,
351
                "flavorRef": self.flavor.id,
352
                "metadata": {
353
                    "My Server Name": "Apache1"
354
                },
355
                "personality": []
356
            }
357
        }
358

    
359
    def test_create_server(self, mrapi):
360
        """Test if the create server call returns the expected response
361
           if a valid request has been speficied."""
362

    
363
        mrapi().CreateInstance.return_value = 12
364
        with override_settings(settings, DEFAULT_INSTANCE_NETWORKS=[]):
365
            with mocked_quotaholder():
366
                response = self.mypost('servers', 'test_user',
367
                                       json.dumps(self.request), 'json')
368
        self.assertEqual(response.status_code, 202)
369
        mrapi().CreateInstance.assert_called_once()
370

    
371
        api_server = json.loads(response.content)['server']
372
        self.assertEqual(api_server['status'], "BUILD")
373
        self.assertEqual(api_server['progress'], 0)
374
        self.assertEqual(api_server['metadata'],
375
                         {"My Server Name":  "Apache1"})
376
        self.assertTrue('adminPass' in api_server)
377

    
378
        db_vm = VirtualMachine.objects.get(userid='test_user')
379
        self.assertEqual(api_server['name'], db_vm.name)
380
        self.assertEqual(api_server['status'], db_vm.operstate)
381

    
382
        # Test drained flag in Network:
383
        self.network.drained = True
384
        self.network.save()
385
        with mocked_quotaholder():
386
            response = self.mypost('servers', 'test_user',
387
                                   json.dumps(self.request), 'json')
388
        self.assertEqual(response.status_code, 503, "serviceUnavailable")
389

    
390
    def test_create_network_settings(self, mrapi):
391
        mrapi().CreateInstance.return_value = 12
392
        # Create public network and backend
393
        subnet1 = mfactory.IPv4SubnetFactory()
394
        bnet1 = mfactory.BackendNetworkFactory(network=subnet1.network,
395
                                               backend=self.backend,
396
                                               operstate="ACTIVE")
397
        subnet2 = mfactory.IPv4SubnetFactory()
398
        bnet2 = mfactory.BackendNetworkFactory(network=subnet2.network,
399
                                               backend=self.backend,
400
                                               operstate="ACTIVE")
401
        subnet3 = mfactory.IPv4SubnetFactory(network__userid="test_user")
402
        bnet3 = mfactory.BackendNetworkFactory(network=subnet3.network,
403
                                               backend=self.backend,
404
                                               operstate="ACTIVE")
405
        subnet4 = mfactory.IPv4SubnetFactory(network__userid="test_user")
406
        bnet4 = mfactory.BackendNetworkFactory(network=subnet4.network,
407
                                               backend=self.backend,
408
                                               operstate="ACTIVE")
409
        # User requested private networks
410
        request = deepcopy(self.request)
411
        request["server"]["networks"] = [bnet3.network.id, bnet4.network.id]
412
        with override_settings(settings,
413
                               DEFAULT_INSTANCE_NETWORKS=[
414
                                   "SNF:ANY_PUBLIC",
415
                                   bnet1.network.id,
416
                                   bnet2.network.id]):
417
            with mocked_quotaholder():
418
                response = self.mypost('servers', 'test_user',
419
                                       json.dumps(request), 'json')
420
        self.assertEqual(response.status_code, 202)
421
        name, args, kwargs = mrapi().CreateInstance.mock_calls[0]
422
        self.assertEqual(len(kwargs["nics"]), 5)
423
        self.assertEqual(kwargs["nics"][0]["network"],
424
                         self.network.backend_id)
425
        self.assertEqual(kwargs["nics"][1]["network"],
426
                         bnet1.network.backend_id)
427
        self.assertEqual(kwargs["nics"][2]["network"],
428
                         bnet2.network.backend_id)
429
        self.assertEqual(kwargs["nics"][3]["network"],
430
                         bnet3.network.backend_id)
431
        self.assertEqual(kwargs["nics"][4]["network"],
432
                         bnet4.network.backend_id)
433

    
434
        request["server"]["floating_ips"] = []
435
        with override_settings(settings,
436
                               DEFAULT_INSTANCE_NETWORKS=[bnet2.network.id]):
437
            with mocked_quotaholder():
438
                response = self.mypost('servers', 'test_user',
439
                                       json.dumps(request), 'json')
440
        self.assertEqual(response.status_code, 202)
441
        name, args, kwargs = mrapi().CreateInstance.mock_calls[1]
442
        self.assertEqual(len(kwargs["nics"]), 3)
443
        self.assertEqual(kwargs["nics"][0]["network"],
444
                         bnet2.network.backend_id)
445
        self.assertEqual(kwargs["nics"][1]["network"],
446
                         bnet3.network.backend_id)
447
        self.assertEqual(kwargs["nics"][2]["network"],
448
                         bnet4.network.backend_id)
449

    
450
        # test invalid network in DEFAULT_INSTANCE_NETWORKS
451
        with override_settings(settings, DEFAULT_INSTANCE_NETWORKS=[42]):
452
            response = self.mypost('servers', 'test_user',
453
                                   json.dumps(request), 'json')
454
        self.assertFault(response, 500, "internalServerError")
455

    
456
        # test connect to public netwok
457
        request = deepcopy(self.request)
458
        request["server"]["networks"] = [self.network.id]
459
        with override_settings(settings,
460
                               DEFAULT_INSTANCE_NETWORKS=["SNF:ANY_PUBLIC"]):
461
            response = self.mypost('servers', 'test_user',
462
                                   json.dumps(request), 'json')
463
        self.assertFault(response, 403, "forbidden")
464
        # test wrong user
465
        request = deepcopy(self.request)
466
        request["server"]["networks"] = [bnet3.network.id]
467
        with override_settings(settings,
468
                               DEFAULT_INSTANCE_NETWORKS=["SNF:ANY_PUBLIC"]):
469
            with mocked_quotaholder():
470
                response = self.mypost('servers', 'dummy_user',
471
                                       json.dumps(request), 'json')
472
        self.assertItemNotFound(response)
473

    
474
        # Test floating IPs
475
        request = deepcopy(self.request)
476
        request["server"]["networks"] = [bnet4.network.id]
477
        fp1 = mfactory.FloatingIPFactory(address="10.0.0.2",
478
                                         userid="test_user",
479
                                         nic=None)
480
        fp2 = mfactory.FloatingIPFactory(address="10.0.0.3",
481
                                         userid="test_user",
482
                                         nic=None)
483
        request["server"]["floating_ips"] = [fp1.address, fp2.address]
484
        with override_settings(settings,
485
                               DEFAULT_INSTANCE_NETWORKS=[bnet3.network.id]):
486
            with mocked_quotaholder():
487
                response = self.mypost('servers', 'test_user',
488
                                       json.dumps(request), 'json')
489
        self.assertEqual(response.status_code, 202)
490
        api_server = json.loads(response.content)['server']
491
        vm = VirtualMachine.objects.get(id=api_server["id"])
492
        fp1 = IPAddress.objects.get(floating_ip=True, id=fp1.id)
493
        fp2 = IPAddress.objects.get(floating_ip=True, id=fp2.id)
494
        self.assertEqual(fp1.nic.machine, vm)
495
        self.assertEqual(fp2.nic.machine, vm)
496
        name, args, kwargs = mrapi().CreateInstance.mock_calls[2]
497
        self.assertEqual(len(kwargs["nics"]), 4)
498
        self.assertEqual(kwargs["nics"][0]["network"],
499
                         bnet3.network.backend_id)
500
        self.assertEqual(kwargs["nics"][1]["network"], fp1.network.backend_id)
501
        self.assertEqual(kwargs["nics"][1]["ip"], fp1.address)
502
        self.assertEqual(kwargs["nics"][2]["network"], fp2.network.backend_id)
503
        self.assertEqual(kwargs["nics"][2]["ip"], fp2.address)
504
        self.assertEqual(kwargs["nics"][3]["network"],
505
                         bnet4.network.backend_id)
506

    
507
    def test_create_server_no_flavor(self, mrapi):
508
        request = deepcopy(self.request)
509
        request["server"]["flavorRef"] = 42
510
        with mocked_quotaholder():
511
            response = self.mypost('servers', 'test_user',
512
                                   json.dumps(request), 'json')
513
        self.assertItemNotFound(response)
514

    
515
    def test_create_server_error(self, mrapi):
516
        """Test if the create server call returns the expected response
517
           if a valid request has been speficied."""
518
        mrapi().CreateInstance.side_effect = GanetiApiError("..ganeti is down")
519

    
520
        request = self.request
521
        with mocked_quotaholder():
522
            response = self.mypost('servers', 'test_user',
523
                                   json.dumps(request), 'json')
524
        self.assertEqual(response.status_code, 202)
525
        mrapi().CreateInstance.assert_called_once()
526
        vm = VirtualMachine.objects.get()
527
        # The VM has not been deleted
528
        self.assertFalse(vm.deleted)
529
        # but is in "ERROR" operstate
530
        self.assertEqual(vm.operstate, "ERROR")
531

    
532

    
533
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
534
class ServerDestroyAPITest(ComputeAPITest):
535
    def test_delete_server(self, mrapi):
536
        vm = mfactory.VirtualMachineFactory()
537
        mrapi().DeleteInstance.return_value = 12
538
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
539
        self.assertEqual(response.status_code, 204)
540
        mrapi().DeleteInstance.assert_called_once()
541

    
542
    def test_non_existing_delete_server(self, mrapi):
543
        vm = mfactory.VirtualMachineFactory()
544
        response = self.mydelete('servers/%d' % 42, vm.userid)
545
        self.assertItemNotFound(response)
546
        self.assertFalse(mrapi.mock_calls)
547

    
548

    
549
class ServerMetadataAPITest(ComputeAPITest):
550
    def setUp(self):
551
        self.vm = mfactory.VirtualMachineFactory()
552
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
553
        super(ServerMetadataAPITest, self).setUp()
554

    
555
    def test_get_metadata(self):
556
        vm = self.vm
557
        create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
558
        metadata = [create_meta(), create_meta(), create_meta()]
559
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
560
        self.assertTrue(response.status_code in [200, 203])
561
        api_metadata = json.loads(response.content)['metadata']
562
        self.assertEqual(len(api_metadata), len(metadata) + 1)
563
        for db_m in metadata:
564
            self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
565

    
566
        request = {
567
            'metadata': {
568
                'foo': 'bar'
569
            },
570
            metadata[0].meta_key: 'bar2'
571
        }
572
        response = self.mypost('servers/%d/metadata' % vm.id,
573
                               vm.userid, json.dumps(request), 'json')
574
        metadata2 = VirtualMachineMetadata.objects.filter(vm=vm)
575
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
576
        self.assertTrue(response.status_code in [200, 203])
577
        api_metadata2 = json.loads(response.content)['metadata']
578
        self.assertTrue('foo' in api_metadata2.keys())
579
        self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2')
580
        self.assertEqual(len(api_metadata2), len(metadata2))
581
        for db_m in metadata2:
582
            self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
583

    
584
        # Create new meta
585
        request = {'meta': {'foo2': 'bar2'}}
586
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
587
                              vm.userid, json.dumps(request), 'json')
588

    
589
        # Get the new meta
590
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
591
        meta = json.loads(response.content)['meta']
592
        self.assertEqual(meta['foo2'], 'bar2')
593

    
594
        # Delete the new meta
595
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
596
        self.assertEqual(response.status_code, 204)
597

    
598
        # Try to get the deleted meta: should raise 404
599
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
600
        self.assertEqual(response.status_code, 404)
601

    
602
    def test_invalid_metadata(self):
603
        vm = self.vm
604
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
605
        self.assertBadRequest(response)
606
        self.assertEqual(len(vm.metadata.all()), 1)
607

    
608
    def test_invalid_metadata_server(self):
609
        response = self.mypost('servers/42/metadata', 'user')
610
        self.assertItemNotFound(response)
611

    
612
    def test_get_meta_invalid_key(self):
613
        vm = self.vm
614
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
615
        self.assertItemNotFound(response)
616

    
617

    
618
@patch('synnefo.api.util.get_image')
619
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
620
class ServerActionAPITest(ComputeAPITest):
621
    def test_actions(self, mrapi, mimage):
622
        actions = ['start', 'shutdown', 'reboot']
623
        vm = mfactory.VirtualMachineFactory()
624
        vm.operstate = "STOPPED"
625
        vm.save()
626
        mrapi().StartupInstance.return_value = 0
627
        mrapi().ShutdownInstance.return_value = 1
628
        mrapi().RebootInstance.return_value = 2
629
        for jobId, action in enumerate(actions):
630
            if action in ["shutdown", "reboot"]:
631
                vm.operstate = "STARTED"
632
            else:
633
                vm.operstate = "STOPPED"
634
            vm.task = None
635
            vm.task_job_id = None
636
            vm.save()
637
            val = {'type': 'HARD'} if action == 'reboot' else {}
638
            request = {action: val}
639
            response = self.mypost('servers/%d/action' % vm.id,
640
                                   vm.userid, json.dumps(request), 'json')
641
            self.assertEqual(response.status_code, 202)
642
            if action == 'shutdown':
643
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
644
                                 "STOP")
645
            else:
646
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).task,
647
                                 action.upper())
648
            self.assertEqual(VirtualMachine.objects.get(id=vm.id).task_job_id,
649
                             jobId)
650

    
651
    def test_action_in_building_vm(self, mrapi, mimage):
652
        """Test building in progress"""
653
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
654
        request = {'start': {}}
655
        with mocked_quotaholder():
656
            response = self.mypost('servers/%d/action' % vm.id,
657
                                   vm.userid, json.dumps(request), 'json')
658
        self.assertEqual(response.status_code, 409)
659
        self.assertFalse(mrapi.mock_calls)
660

    
661
    def test_destroy_build_vm(self, mrapi, mimage):
662
        """Test building in progress"""
663
        vm = mfactory.VirtualMachineFactory()
664
        mrapi().DeleteInstance.return_value = 2
665
        response = self.mydelete('servers/%d' % vm.id,
666
                                 vm.userid)
667
        self.assertSuccess(response)
668
        mrapi().RemoveInstance.assert_called_once()
669

    
670
    def test_firewall(self, mrapi, mimage):
671
        vm = mfactory.VirtualMachineFactory()
672
        vm.operstate = "STOPPED"
673
        vm.save()
674
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
675
        response = self.mypost('servers/%d/action' % vm.id,
676
                               vm.userid, json.dumps(request), 'json')
677
        self.assertBadRequest(response)
678
        request = {'firewallProfile': {'profile': 'PROTECTED', "nic": "10"}}
679
        response = self.mypost('servers/%d/action' % vm.id,
680
                               vm.userid, json.dumps(request), 'json')
681
        self.assertItemNotFound(response)
682
        nic = mfactory.NetworkInterfaceFactory(machine=vm)
683
        request = {'firewallProfile': {'profile': 'PROTECTED', "nic": nic.id}}
684
        response = self.mypost('servers/%d/action' % vm.id,
685
                               vm.userid, json.dumps(request), 'json')
686
        self.assertSuccess(response)
687
        mrapi().ModifyInstance.assert_called_once()
688

    
689
    def test_unsupported_firewall(self, mrapi, mimage):
690
        vm = mfactory.VirtualMachineFactory()
691
        vm.operstate = "STOPPED"
692
        vm.save()
693
        request = {'firewallProfile': {'profile': 'FOO'}}
694
        response = self.mypost('servers/%d/action' % vm.id,
695
                               vm.userid, json.dumps(request), 'json')
696
        self.assertBadRequest(response)
697
        self.assertFalse(mrapi.mock_calls)
698

    
699
    def test_resize_vm(self, mrapi, mimage):
700
        flavor = mfactory.FlavorFactory(cpu=1, ram=1024)
701
        # Check building VM
702
        vm = self.get_vm(flavor=flavor, operstate="BUILD")
703
        request = {'resize': {'flavorRef': flavor.id}}
704
        response = self.mypost('servers/%d/action' % vm.id,
705
                               vm.userid, json.dumps(request), 'json')
706
        self.assertFault(response, 409, "buildInProgress")
707
        # Check same Flavor
708
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
709
        request = {'resize': {'flavorRef': flavor.id}}
710
        response = self.mypost('servers/%d/action' % vm.id,
711
                               vm.userid, json.dumps(request), 'json')
712
        self.assertBadRequest(response)
713
        # Check flavor with different disk
714
        flavor2 = mfactory.FlavorFactory(disk=1024)
715
        flavor3 = mfactory.FlavorFactory(disk=2048)
716
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
717
        request = {'resize': {'flavorRef': flavor3.id}}
718
        response = self.mypost('servers/%d/action' % vm.id,
719
                               vm.userid, json.dumps(request), 'json')
720
        self.assertBadRequest(response)
721
        flavor2 = mfactory.FlavorFactory(disk_template="foo")
722
        flavor3 = mfactory.FlavorFactory(disk_template="baz")
723
        vm = self.get_vm(flavor=flavor2, operstate="STOPPED")
724
        request = {'resize': {'flavorRef': flavor3.id}}
725
        response = self.mypost('servers/%d/action' % vm.id,
726
                               vm.userid, json.dumps(request), 'json')
727
        self.assertBadRequest(response)
728
        # Check success
729
        vm = self.get_vm(flavor=flavor, operstate="STOPPED")
730
        flavor4 = mfactory.FlavorFactory(disk_template=flavor.disk_template,
731
                                         disk=flavor.disk,
732
                                         cpu=4, ram=2048)
733
        request = {'resize': {'flavorRef': flavor4.id}}
734
        mrapi().ModifyInstance.return_value = 42
735
        response = self.mypost('servers/%d/action' % vm.id,
736
                               vm.userid, json.dumps(request), 'json')
737
        self.assertEqual(response.status_code, 202)
738
        vm = VirtualMachine.objects.get(id=vm.id)
739
        self.assertEqual(vm.task_job_id, 42)
740
        name, args, kwargs = mrapi().ModifyInstance.mock_calls[0]
741
        self.assertEqual(kwargs["beparams"]["vcpus"], 4)
742
        self.assertEqual(kwargs["beparams"]["minmem"], 2048)
743
        self.assertEqual(kwargs["beparams"]["maxmem"], 2048)
744

    
745
    def test_action_on_resizing_vm(self, mrapi, mimage):
746
        vm = mfactory.VirtualMachineFactory()
747
        vm.operstate = "RESIZE"
748
        vm.save()
749
        for action in VirtualMachine.ACTIONS:
750
            request = {action[0]: ""}
751
            response = self.mypost('servers/%d/action' % vm.id,
752
                                   vm.userid, json.dumps(request), 'json')
753
            self.assertBadRequest(response)
754
        # however you can destroy
755
        mrapi().DeleteInstance.return_value = 42
756
        response = self.mydelete('servers/%d' % vm.id,
757
                                 vm.userid)
758
        self.assertSuccess(response)
759

    
760
    def get_vm(self, flavor, operstate):
761
        vm = mfactory.VirtualMachineFactory(flavor=flavor)
762
        vm.operstate = operstate
763
        vm.backendjobstatus = "success"
764
        vm.save()
765
        return vm
766

    
767

    
768
class ServerVNCConsole(ComputeAPITest):
769
    def test_not_active_server(self):
770
        """Test console req for not ACTIVE server returns badRequest"""
771
        vm = mfactory.VirtualMachineFactory(operstate="BUILD")
772
        data = json.dumps({'console': {'type': 'vnc'}})
773
        response = self.mypost('servers/%d/action' % vm.id,
774
                               vm.userid, data, 'json')
775
        self.assertBadRequest(response)
776

    
777
    def test_active_server(self):
778
        """Test console req for ACTIVE server"""
779
        vm = mfactory.VirtualMachineFactory()
780
        vm.operstate = 'STARTED'
781
        vm.save()
782

    
783
        data = json.dumps({'console': {'type': 'vnc'}})
784
        with override_settings(settings, TEST=True):
785
            response = self.mypost('servers/%d/action' % vm.id,
786
                                   vm.userid, data, 'json')
787
        self.assertEqual(response.status_code, 200)
788
        reply = json.loads(response.content)
789
        self.assertEqual(reply.keys(), ['console'])
790
        console = reply['console']
791
        self.assertEqual(console['type'], 'vnc')
792
        self.assertEqual(set(console.keys()),
793
                         set(['type', 'host', 'port', 'password']))
794

    
795
    def test_wrong_console_type(self):
796
        """Test console req for ACTIVE server"""
797
        vm = mfactory.VirtualMachineFactory()
798
        vm.operstate = 'STARTED'
799
        vm.save()
800

    
801
        data = json.dumps({'console': {'type': 'foo'}})
802
        response = self.mypost('servers/%d/action' % vm.id,
803
                               vm.userid, data, 'json')
804
        self.assertBadRequest(response)