Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (18.3 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

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

    
211
        request = {
212
                    "server": {
213
                        "name": "new-server-test",
214
                        "userid": "test_user",
215
                        "imageRef": 1,
216
                        "flavorRef": flavor.id,
217
                        "metadata": {
218
                            "My Server Name": "Apache1"
219
                        },
220
                        "personality": []
221
                    }
222
        }
223
        with mocked_quotaholder():
224
            response = self.mypost('servers', 'test_user',
225
                                   json.dumps(request), 'json')
226
        self.assertEqual(response.status_code, 202)
227
        mrapi().CreateInstance.assert_called_once()
228

    
229
        api_server = json.loads(response.content)['server']
230
        self.assertEqual(api_server['status'], "BUILD")
231
        self.assertEqual(api_server['progress'], 0)
232
        self.assertEqual(api_server['metadata'],
233
                        {"My Server Name":  "Apache1"})
234
        self.assertTrue('adminPass' in api_server)
235

    
236
        db_vm = VirtualMachine.objects.get(userid='test_user')
237
        self.assertEqual(api_server['name'], db_vm.name)
238
        self.assertEqual(api_server['status'], db_vm.operstate)
239

    
240
        # Test drained flag in Network:
241
        network.drained = True
242
        network.save()
243
        with mocked_quotaholder():
244
            response = self.mypost('servers', 'test_user',
245
                                    json.dumps(request), 'json')
246
        self.assertEqual(response.status_code, 503, "serviceUnavailable")
247

    
248
    def test_create_server_no_flavor(self, mrapi, mimage):
249
        request = {
250
                    "server": {
251
                        "name": "new-server-test",
252
                        "userid": "test_user",
253
                        "imageRef": 1,
254
                        "flavorRef": 42,
255
                        "metadata": {
256
                            "My Server Name": "Apache1"
257
                        },
258
                        "personality": []
259
                    }
260
        }
261
        response = self.mypost('servers', 'test_user',
262
                               json.dumps(request), 'json')
263
        self.assertItemNotFound(response)
264

    
265

    
266
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
267
class ServerDestroyAPITest(ComputeAPITest):
268
    def test_delete_server(self, mrapi):
269
        vm = mfactory.VirtualMachineFactory()
270
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
271
        self.assertEqual(response.status_code, 204)
272
        mrapi().DeleteInstance.assert_called_once()
273

    
274
    def test_non_existing_delete_server(self, mrapi):
275
        vm = mfactory.VirtualMachineFactory()
276
        response = self.mydelete('servers/%d' % 42, vm.userid)
277
        self.assertItemNotFound(response)
278
        self.assertFalse(mrapi.mock_calls)
279

    
280

    
281
class ServerMetadataAPITest(ComputeAPITest):
282
    def setUp(self):
283
        self.vm = mfactory.VirtualMachineFactory()
284
        self.metadata = mfactory.VirtualMachineMetadataFactory(vm=self.vm)
285
        super(ServerMetadataAPITest, self).setUp()
286

    
287
    def test_get_metadata(self):
288
        vm = self.vm
289
        create_meta = lambda: mfactory.VirtualMachineMetadataFactory(vm=vm)
290
        metadata = [create_meta(), create_meta(), create_meta()]
291
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
292
        self.assertTrue(response.status_code in [200, 203])
293
        api_metadata = json.loads(response.content)['metadata']
294
        self.assertEqual(len(api_metadata), len(metadata) + 1)
295
        for db_m in metadata:
296
            self.assertEqual(api_metadata[db_m.meta_key], db_m.meta_value)
297

    
298
        request = {'metadata':
299
                        {'foo': 'bar'},
300
                        metadata[0].meta_key: 'bar2'
301
                  }
302
        response = self.mypost('servers/%d/metadata' % vm.id,
303
                             vm.userid, json.dumps(request), 'json')
304
        metadata2 = VirtualMachineMetadata.objects.filter(vm=vm)
305
        response = self.myget('servers/%d/metadata' % vm.id, vm.userid)
306
        self.assertTrue(response.status_code in [200, 203])
307
        api_metadata2 = json.loads(response.content)['metadata']
308
        self.assertTrue('foo' in api_metadata2.keys())
309
        self.assertTrue(api_metadata2[metadata[0].meta_key], 'bar2')
310
        self.assertEqual(len(api_metadata2), len(metadata2))
311
        for db_m in metadata2:
312
            self.assertEqual(api_metadata2[db_m.meta_key], db_m.meta_value)
313

    
314
        # Create new meta
315
        request = {'meta': {'foo2': 'bar2'}}
316
        response = self.myput('servers/%d/metadata/foo2' % vm.id,
317
                              vm.userid, json.dumps(request), 'json')
318

    
319
        # Get the new meta
320
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
321
        meta = json.loads(response.content)['meta']
322
        self.assertEqual(meta['foo2'], 'bar2')
323

    
324
        # Delete the new meta
325
        response = self.mydelete('servers/%d/metadata/foo2' % vm.id, vm.userid)
326
        self.assertEqual(response.status_code, 204)
327

    
328
        # Try to get the deleted meta: should raise 404
329
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
330
        self.assertEqual(response.status_code, 404)
331

    
332
    def test_invalid_metadata(self):
333
        vm = self.vm
334
        response = self.mypost('servers/%d/metadata' % vm.id, vm.userid)
335
        self.assertBadRequest(response)
336
        self.assertEqual(len(vm.metadata.all()), 1)
337

    
338
    def test_invalid_metadata_server(self):
339
        response = self.mypost('servers/42/metadata', 'user')
340
        self.assertItemNotFound(response)
341

    
342
    def test_get_meta_invalid_key(self):
343
        vm = self.vm
344
        response = self.myget('servers/%d/metadata/foo2' % vm.id, vm.userid)
345
        self.assertItemNotFound(response)
346

    
347

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

    
369
    def test_action_in_building_vm(self, mrapi, mimage):
370
        """Test building in progress"""
371
        vm = mfactory.VirtualMachineFactory()
372
        request = {'start': '{}'}
373
        response = self.mypost('servers/%d/action' % vm.id,
374
                               vm.userid, json.dumps(request), 'json')
375
        self.assertEqual(response.status_code, 409)
376
        self.assertFalse(mrapi.mock_calls)
377

    
378
    def test_destroy_build_vm(self, mrapi, mimage):
379
        """Test building in progress"""
380
        vm = mfactory.VirtualMachineFactory()
381
        response = self.mydelete('servers/%d' % vm.id, vm.userid)
382
        self.assertSuccess(response)
383
        mrapi().RemoveInstance.assert_called_once()
384

    
385
    def test_firewall(self, mrapi, mimage):
386
        vm = mfactory.VirtualMachineFactory()
387
        vm.operstate = "STOPPED"
388
        vm.save()
389
        request = {'firewallProfile': {'profile': 'PROTECTED'}}
390
        response = self.mypost('servers/%d/action' % vm.id,
391
                               vm.userid, json.dumps(request), 'json')
392
        self.assertEqual(response.status_code, 202)
393
        mrapi().ModifyInstance.assert_called_once()
394

    
395
    def test_unsupported_firewall(self, mrapi, mimage):
396
        vm = mfactory.VirtualMachineFactory()
397
        vm.operstate = "STOPPED"
398
        vm.save()
399
        request = {'firewallProfile': {'profile': 'FOO'}}
400
        response = self.mypost('servers/%d/action' % vm.id,
401
                               vm.userid, json.dumps(request), 'json')
402
        self.assertBadRequest(response)
403
        self.assertFalse(mrapi.mock_calls)
404

    
405

    
406
class ServerVNCConsole(ComputeAPITest):
407
    def test_not_active_server(self):
408
        """Test console req for not ACTIVE server returns badRequest"""
409
        vm = mfactory.VirtualMachineFactory()
410
        data = json.dumps({'console': {'type': 'vnc'}})
411
        response = self.mypost('servers/%d/action' % vm.id,
412
                               vm.userid, data, 'json')
413
        self.assertBadRequest(response)
414

    
415
    def test_active_server(self):
416
        """Test console req for ACTIVE server"""
417
        vm = mfactory.VirtualMachineFactory()
418
        vm.operstate = 'STARTED'
419
        vm.save()
420

    
421
        data = json.dumps({'console': {'type': 'vnc'}})
422
        response = self.mypost('servers/%d/action' % vm.id,
423
                               vm.userid, data, 'json')
424
        self.assertEqual(response.status_code, 200)
425
        reply = json.loads(response.content)
426
        self.assertEqual(reply.keys(), ['console'])
427
        console = reply['console']
428
        self.assertEqual(console['type'], 'vnc')
429
        self.assertEqual(set(console.keys()),
430
                         set(['type', 'host', 'port', 'password']))
431

    
432
    def test_wrong_console_type(self):
433
        """Test console req for ACTIVE server"""
434
        vm = mfactory.VirtualMachineFactory()
435
        vm.operstate = 'STARTED'
436
        vm.save()
437

    
438
        data = json.dumps({'console': {'type': 'foo'}})
439
        response = self.mypost('servers/%d/action' % vm.id,
440
                               vm.userid, data, 'json')
441
        self.assertBadRequest(response)