Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (18.6 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
        self.assertEqual(servers, [{'name': db_server.name,
94
                                    'id': db_server.id}])
95

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

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

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

    
122
        db_vm_meta = mfactory.VirtualMachineMetadataFactory(vm=db_vm)
123

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

    
127
        self.assertEqual(server['flavor'], db_vm.flavor.id)
128
        self.assertEqual(server['hostId'], db_vm.hostid)
129
        self.assertEqual(server['id'], db_vm.id)
130
        self.assertEqual(server['image'], db_vm.imageid)
131
        self.assertEqual(server['name'], db_vm.name)
132
        self.assertEqual(server['status'], get_rsapi_state(db_vm))
133
        api_nic = server['attachments'][0]
134
        self.assertEqual(api_nic['network_id'], str(net.id))
135
        self.assertEqual(api_nic['mac_address'], nic.mac)
136
        self.assertEqual(api_nic['firewallProfile'], nic.firewall_profile)
137
        self.assertEqual(api_nic['ipv4'], nic.ipv4)
138
        self.assertEqual(api_nic['ipv6'], nic.ipv6)
139
        self.assertEqual(api_nic['id'], 'nic-%s-%s' % (db_vm.id, nic.index))
140

    
141
        metadata = server['metadata']
142
        self.assertEqual(len(metadata), 1)
143
        self.assertEqual(metadata[db_vm_meta.meta_key], db_vm_meta.meta_value)
144
        self.assertSuccess(response)
145

    
146
    def test_server_building_nics(self):
147
        db_vm = self.vm2
148
        user = self.vm2.userid
149
        net1 = mfactory.NetworkFactory()
150
        net2 = mfactory.NetworkFactory()
151
        net3 = mfactory.NetworkFactory()
152
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net1,
153
                                         state="BUILDING")
154
        nic2 = mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net2,
155
                                                state="ACTIVE")
156
        mfactory.NetworkInterfaceFactory(machine=self.vm2, network=net3,
157
                                         state="BUILDING")
158

    
159
        response = self.myget('servers/%d' % db_vm.id, user)
160
        server = json.loads(response.content)['server']
161
        nics = server["attachments"]
162
        self.assertEqual(len(nics), 1)
163
        self.assertEqual(nics[0]["network_id"], str(nic2.network_id))
164

    
165
    def test_noauthorized(self):
166
        """Test 404 for detail of other user vm"""
167
        db_vm = self.vm2
168

    
169
        response = self.myget('servers/%d' % db_vm.id, 'wrong_user')
170
        self.assertItemNotFound(response)
171

    
172
    def test_wrong_server(self):
173
        """Test 404 response if server does not exist."""
174
        response = self.myget('servers/%d' % 5000)
175
        self.assertItemNotFound(response)
176

    
177
    def test_create_server_empty(self):
178
        """Test if the create server call returns a 400 badRequest if
179
           no attributes are specified."""
180

    
181
        response = self.mypost('servers', params={})
182
        self.assertBadRequest(response)
183

    
184
    def test_rename_server(self):
185
        vm = self.vm2
186
        request = {'server': {'name': 'new_name'}}
187
        response = self.myput('servers/%d' % vm.id, vm.userid,
188
                              json.dumps(request), 'json')
189
        self.assertSuccess(response)
190
        self.assertEqual(VirtualMachine.objects.get(id=vm.id).name, "new_name")
191

    
192
    def test_catch_wrong_api_paths(self):
193
        response = self.myget('nonexistent')
194
        self.assertEqual(response.status_code, 400)
195
        try:
196
            error = json.loads(response.content)
197
        except ValueError:
198
            self.assertTrue(False)
199

    
200

    
201
@patch('synnefo.api.util.get_image')
202
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
203
class ServerCreateAPITest(ComputeAPITest):
204
    def test_create_server(self, mrapi, mimage):
205
        """Test if the create server call returns the expected response
206
           if a valid request has been speficied."""
207
        mimage.return_value = {'location': 'pithos://foo',
208
                               'checksum': '1234',
209
                               "id": 1,
210
                               "name": "test_image",
211
                               'disk_format': 'diskdump'}
212
        mrapi().CreateInstance.return_value = 12
213
        flavor = mfactory.FlavorFactory()
214
        # Create public network and backend
215
        network = mfactory.NetworkFactory(public=True)
216
        backend = mfactory.BackendFactory()
217
        mfactory.BackendNetworkFactory(network=network, backend=backend)
218

    
219
        request = {
220
                    "server": {
221
                        "name": "new-server-test",
222
                        "userid": "test_user",
223
                        "imageRef": 1,
224
                        "flavorRef": flavor.id,
225
                        "metadata": {
226
                            "My Server Name": "Apache1"
227
                        },
228
                        "personality": []
229
                    }
230
        }
231
        with mocked_quotaholder():
232
            response = self.mypost('servers', 'test_user',
233
                                   json.dumps(request), 'json')
234
        self.assertEqual(response.status_code, 202)
235
        mrapi().CreateInstance.assert_called_once()
236

    
237
        api_server = json.loads(response.content)['server']
238
        self.assertEqual(api_server['status'], "BUILD")
239
        self.assertEqual(api_server['progress'], 0)
240
        self.assertEqual(api_server['metadata'],
241
                        {"My Server Name":  "Apache1"})
242
        self.assertTrue('adminPass' in api_server)
243

    
244
        db_vm = VirtualMachine.objects.get(userid='test_user')
245
        self.assertEqual(api_server['name'], db_vm.name)
246
        self.assertEqual(api_server['status'], db_vm.operstate)
247

    
248
        # Test drained flag in Network:
249
        network.drained = True
250
        network.save()
251
        with mocked_quotaholder():
252
            response = self.mypost('servers', 'test_user',
253
                                    json.dumps(request), 'json')
254
        self.assertEqual(response.status_code, 503, "serviceUnavailable")
255

    
256
    def test_create_server_no_flavor(self, mrapi, mimage):
257
        request = {
258
                    "server": {
259
                        "name": "new-server-test",
260
                        "userid": "test_user",
261
                        "imageRef": 1,
262
                        "flavorRef": 42,
263
                        "metadata": {
264
                            "My Server Name": "Apache1"
265
                        },
266
                        "personality": []
267
                    }
268
        }
269
        response = self.mypost('servers', 'test_user',
270
                               json.dumps(request), 'json')
271
        self.assertItemNotFound(response)
272

    
273

    
274
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
275
class ServerDestroyAPITest(ComputeAPITest):
276
    def test_delete_server(self, mrapi):
277
        vm = mfactory.VirtualMachineFactory()
278
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
279
        self.assertEqual(response.status_code, 204)
280
        mrapi().DeleteInstance.assert_called_once()
281

    
282
    def test_non_existing_delete_server(self, mrapi):
283
        vm = mfactory.VirtualMachineFactory()
284
        response = self.mydelete('servers/%d' % 42, vm.userid)
285
        self.assertItemNotFound(response)
286
        self.assertFalse(mrapi.mock_calls)
287

    
288

    
289
class ServerMetadataAPITest(ComputeAPITest):
290
    def setUp(self):
291
        self.vm = mfactory.VirtualMachineFactory()
292
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
293
        super(ServerMetadataAPITest, self).setUp()
294

    
295
    def test_get_metadata(self):
296
        vm = self.vm
297
        create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
298
        metadata = [create_meta(), create_meta(), create_meta()]
299
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
300
        self.assertTrue(response.status_code in [200, 203])
301
        api_metadata = json.loads(response.content)['metadata']
302
        self.assertEqual(len(api_metadata), len(metadata) + 1)
303
        for db_m in metadata:
304
            self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
305

    
306
        request = {'metadata':
307
                        {'foo': 'bar'},
308
                        metadata[0].meta_key: 'bar2'
309
                  }
310
        response = self.mypost('servers/%d/metadata' % vm.id,
311
                             vm.userid, json.dumps(request), 'json')
312
        metadata2 = VirtualMachineMetadata.objects.filter(vm=vm)
313
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
314
        self.assertTrue(response.status_code in [200, 203])
315
        api_metadata2 = json.loads(response.content)['metadata']
316
        self.assertTrue('foo' in api_metadata2.keys())
317
        self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2')
318
        self.assertEqual(len(api_metadata2), len(metadata2))
319
        for db_m in metadata2:
320
            self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
321

    
322
        # Create new meta
323
        request = {'meta': {'foo2': 'bar2'}}
324
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
325
                              vm.userid, json.dumps(request), 'json')
326

    
327
        # Get the new meta
328
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
329
        meta = json.loads(response.content)['meta']
330
        self.assertEqual(meta['foo2'], 'bar2')
331

    
332
        # Delete the new meta
333
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
334
        self.assertEqual(response.status_code, 204)
335

    
336
        # Try to get the deleted meta: should raise 404
337
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
338
        self.assertEqual(response.status_code, 404)
339

    
340
    def test_invalid_metadata(self):
341
        vm = self.vm
342
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
343
        self.assertBadRequest(response)
344
        self.assertEqual(len(vm.metadata.all()), 1)
345

    
346
    def test_invalid_metadata_server(self):
347
        response = self.mypost('servers/42/metadata', 'user')
348
        self.assertItemNotFound(response)
349

    
350
    def test_get_meta_invalid_key(self):
351
        vm = self.vm
352
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
353
        self.assertItemNotFound(response)
354

    
355

    
356
@patch('synnefo.api.util.get_image')
357
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
358
class ServerActionAPITest(ComputeAPITest):
359
    def test_actions(self, mrapi, mimage):
360
        actions = ['start', 'shutdown', 'reboot']
361
        vm = mfactory.VirtualMachineFactory()
362
        vm.operstate = "STOPPED"
363
        vm.save()
364
        for action in actions:
365
            val = {'type': 'HARD'} if action == 'reboot' else {}
366
            request = {action: val}
367
            response = self.mypost('servers/%d/action' % vm.id,
368
                                   vm.userid, json.dumps(request), 'json')
369
            self.assertEqual(response.status_code, 202)
370
            if action == 'shutdown':
371
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
372
                                 "STOP")
373
            else:
374
                self.assertEqual(VirtualMachine.objects.get(id=vm.id).action,
375
                                 action.upper())
376

    
377
    def test_action_in_building_vm(self, mrapi, mimage):
378
        """Test building in progress"""
379
        vm = mfactory.VirtualMachineFactory()
380
        request = {'start': '{}'}
381
        response = self.mypost('servers/%d/action' % vm.id,
382
                               vm.userid, json.dumps(request), 'json')
383
        self.assertEqual(response.status_code, 409)
384
        self.assertFalse(mrapi.mock_calls)
385

    
386
    def test_destroy_build_vm(self, mrapi, mimage):
387
        """Test building in progress"""
388
        vm = mfactory.VirtualMachineFactory()
389
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
390
        self.assertSuccess(response)
391
        mrapi().RemoveInstance.assert_called_once()
392

    
393
    def test_firewall(self, mrapi, mimage):
394
        vm = mfactory.VirtualMachineFactory()
395
        vm.operstate = "STOPPED"
396
        vm.save()
397
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
398
        response = self.mypost('servers/%d/action' % vm.id,
399
                               vm.userid, json.dumps(request), 'json')
400
        self.assertEqual(response.status_code, 202)
401
        mrapi().ModifyInstance.assert_called_once()
402

    
403
    def test_unsupported_firewall(self, mrapi, mimage):
404
        vm = mfactory.VirtualMachineFactory()
405
        vm.operstate = "STOPPED"
406
        vm.save()
407
        request = {'firewallProfile': {'profile': 'FOO'}}
408
        response = self.mypost('servers/%d/action' % vm.id,
409
                               vm.userid, json.dumps(request), 'json')
410
        self.assertBadRequest(response)
411
        self.assertFalse(mrapi.mock_calls)
412

    
413

    
414
class ServerVNCConsole(ComputeAPITest):
415
    def test_not_active_server(self):
416
        """Test console req for not ACTIVE server returns badRequest"""
417
        vm = mfactory.VirtualMachineFactory()
418
        data = json.dumps({'console': {'type': 'vnc'}})
419
        response = self.mypost('servers/%d/action' % vm.id,
420
                               vm.userid, data, 'json')
421
        self.assertBadRequest(response)
422

    
423
    def test_active_server(self):
424
        """Test console req for ACTIVE server"""
425
        vm = mfactory.VirtualMachineFactory()
426
        vm.operstate = 'STARTED'
427
        vm.save()
428

    
429
        data = json.dumps({'console': {'type': 'vnc'}})
430
        response = self.mypost('servers/%d/action' % vm.id,
431
                               vm.userid, data, 'json')
432
        self.assertEqual(response.status_code, 200)
433
        reply = json.loads(response.content)
434
        self.assertEqual(reply.keys(), ['console'])
435
        console = reply['console']
436
        self.assertEqual(console['type'], 'vnc')
437
        self.assertEqual(set(console.keys()),
438
                         set(['type', 'host', 'port', 'password']))
439

    
440
    def test_wrong_console_type(self):
441
        """Test console req for ACTIVE server"""
442
        vm = mfactory.VirtualMachineFactory()
443
        vm.operstate = 'STARTED'
444
        vm.save()
445

    
446
        data = json.dumps({'console': {'type': 'foo'}})
447
        response = self.mypost('servers/%d/action' % vm.id,
448
                               vm.userid, data, 'json')
449
        self.assertBadRequest(response)