Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / test / networks.py @ 2cd55da6

History | View | Annotate | Download (19.9 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 mock import patch
36

    
37
from snf_django.utils.testing import BaseAPITest, mocked_quotaholder
38
from synnefo.db.models import Network, NetworkInterface
39
from synnefo.db import models_factory as mfactory
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

    
45
class ComputeAPITest(BaseAPITest):
46
    def __init__(self, *args, **kwargs):
47
        super(ComputeAPITest, self).__init__(*args, **kwargs)
48
        self.compute_path = get_service_path(cyclades_services, 'compute',
49
                                             version='v2.0')
50

    
51
    def myget(self, path, *args, **kwargs):
52
        path = join_urls(self.compute_path, path)
53
        return self.get(path, *args, **kwargs)
54

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

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

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

    
67

    
68
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
69
class NetworkAPITest(ComputeAPITest):
70
    def setUp(self):
71
        self.mac_prefixes = mfactory.MacPrefixPoolTableFactory()
72
        self.bridges = mfactory.BridgePoolTableFactory(base="link")
73
        self.user = 'dummy-user'
74
        self.net1 = mfactory.NetworkFactory(userid=self.user)
75
        self.vm1 = mfactory.VirtualMachineFactory(userid=self.user)
76
        self.nic1 = mfactory.NetworkInterfaceFactory(network=self.net1,
77
                                                     machine=self.vm1)
78
        self.nic2 = mfactory.NetworkInterfaceFactory(network=self.net1,
79
                                                     machine=self.vm1)
80
        self.net2 = mfactory.NetworkFactory(userid=self.user)
81
        self.nic3 = mfactory.NetworkInterfaceFactory(network=self.net2)
82
        super(NetworkAPITest, self).setUp()
83

    
84
    def assertNetworksEqual(self, db_net, api_net, detail=False):
85
        self.assertEqual(str(db_net.id), api_net["id"])
86
        self.assertEqual(db_net.name, api_net['name'])
87
        if detail:
88
            self.assertEqual(db_net.state, api_net['status'])
89
            self.assertEqual(db_net.flavor, api_net['type'])
90
            self.assertEqual(db_net.subnet, api_net['cidr'])
91
            self.assertEqual(db_net.subnet6, api_net['cidr6'])
92
            self.assertEqual(db_net.gateway, api_net['gateway'])
93
            self.assertEqual(db_net.gateway6, api_net['gateway6'])
94
            self.assertEqual(db_net.dhcp, api_net['dhcp'])
95
            self.assertEqual(db_net.public, api_net['public'])
96
            db_nics = ["nic-%d-%d" % (nic.machine.id, nic.index) for nic in
97
                       db_net.nics.filter(machine__userid=db_net.userid)]
98
            self.assertEqual(db_nics, api_net['attachments'])
99

    
100
    def test_create_network_1(self, mrapi):
101
        request = {
102
            'network': {'name': 'foo', "type": "MAC_FILTERED"}
103
        }
104
        with mocked_quotaholder():
105
            response = self.mypost('networks/', 'user1',
106
                                   json.dumps(request), 'json')
107
        self.assertEqual(response.status_code, 202)
108
        db_networks = Network.objects.filter(userid='user1')
109
        self.assertEqual(len(db_networks), 1)
110
        db_net = db_networks[0]
111
        api_net = json.loads(response.content)['network']
112
        self.assertNetworksEqual(db_net, api_net)
113
        mrapi.CreateNetwork.assert_called()
114
        mrapi.ConnectNetwork.assert_called()
115

    
116
    def test_invalid_data_1(self, mrapi):
117
        """Test invalid flavor"""
118
        request = {
119
            'network': {'name': 'foo', 'type': 'LoLo'}
120
            }
121
        response = self.mypost('networks/', 'user1',
122
                               json.dumps(request), 'json')
123
        self.assertBadRequest(response)
124
        self.assertEqual(len(Network.objects.filter(userid='user1')), 0)
125

    
126
    def test_invalid_data_2(self, mrapi):
127
        """Test invalid data/subnet"""
128
        request = {
129
            'network': {'name': 'foo',
130
                        'cidr': '10.0.0.0/8', "type":
131
                        "MAC_FILTERED"}
132
        }
133
        response = self.mypost('networks/', 'user1',
134
                               json.dumps(request), 'json')
135
        self.assertFault(response, 413, "overLimit")
136

    
137
    def test_invalid_data_3(self, mrapi):
138
        """Test unauthorized to create public network"""
139
        request = {
140
                'network': {'name': 'foo',
141
                            "public": "True",
142
                            "type": "MAC_FILTERED"}
143
            }
144
        response = self.mypost('networks/', 'user1',
145
                               json.dumps(request), 'json')
146
        self.assertFault(response, 403, "forbidden")
147

    
148
    def test_invalid_data_4(self, mrapi):
149
        """Test unauthorized to create network not in settings"""
150
        request = {
151
                'network': {'name': 'foo', 'type': 'CUSTOM'}
152
            }
153
        response = self.mypost('networks/', 'user1',
154
                               json.dumps(request), 'json')
155
        self.assertFault(response, 403, "forbidden")
156

    
157
    def test_invalid_subnet(self, mrapi):
158
        """Test invalid subnet"""
159
        request = {
160
            'network': {'name': 'foo',
161
                        'cidr': '10.0.0.10/27',
162
                        "type": "MAC_FILTERED"}
163
        }
164
        response = self.mypost('networks/', 'user1',
165
                               json.dumps(request), 'json')
166
        self.assertBadRequest(response)
167

    
168
    def test_invalid_gateway_1(self, mrapi):
169
        request = {
170
            'network': {'name': 'foo',
171
                        'cidr': '10.0.0.0/28',
172
                        'gateway': '10.0.0.0.300'}
173
        }
174
        response = self.mypost('networks/', 'user1',
175
                               json.dumps(request), 'json')
176
        self.assertBadRequest(response)
177

    
178
    def test_invalid_gateway_2(self, mrapi):
179
        request = {
180
            'network': {'name': 'foo',
181
                        'cidr': '10.0.0.0/28',
182
                        'gateway': '10.2.0.1'}
183
        }
184
        response = self.mypost('networks/', 'user1',
185
                               json.dumps(request), 'json')
186
        self.assertBadRequest(response)
187

    
188
    def test_invalid_network6(self, mrapi):
189
        request = {
190
            'network': {'name': 'foo',
191
                        'cidr': '10.0.0.0/28',
192
                        'subnet6': '10.0.0.0/28',
193
                        'gateway': '10.2.0.1'}
194
        }
195
        response = self.mypost('networks/', 'user1',
196
                               json.dumps(request), 'json')
197
        self.assertBadRequest(response)
198

    
199
    def test_invalid_gateway6(self, mrapi):
200
        request = {
201
            'network': {'name': 'foo',
202
                        'cidr': '10.0.0.0/28',
203
                        'subnet6': '2001:0db8:0123:4567:89ab:cdef:1234:5678',
204
                        'gateway': '10.2.0.1'}
205
        }
206
        response = self.mypost('networks/', 'user1',
207
                               json.dumps(request), 'json')
208
        self.assertBadRequest(response)
209

    
210
    def test_list_networks(self, mrapi):
211
        """Test that expected list of networks is returned."""
212
        # Create a deleted network
213
        mfactory.NetworkFactory(userid=self.user, deleted=True)
214

    
215
        response = self.myget('networks/', self.user)
216
        self.assertSuccess(response)
217

    
218
        db_nets = Network.objects.filter(userid=self.user, deleted=False)
219
        api_nets = json.loads(response.content)["networks"]
220

    
221
        self.assertEqual(len(db_nets), len(api_nets))
222
        for api_net in api_nets:
223
            net_id = api_net['id']
224
            self.assertNetworksEqual(Network.objects.get(id=net_id), api_net)
225

    
226
    def test_list_networks_detail(self, mrapi):
227
        """Test that expected networks details are returned."""
228
        # Create a deleted network
229
        mfactory.NetworkFactory(userid=self.user, deleted=True)
230

    
231
        response = self.myget('networks/detail', self.user)
232
        self.assertSuccess(response)
233

    
234
        db_nets = Network.objects.filter(userid=self.user, deleted=False)
235
        api_nets = json.loads(response.content)["networks"]
236

    
237
        self.assertEqual(len(db_nets), len(api_nets))
238
        for api_net in api_nets:
239
            net_id = api_net['id']
240
            self.assertNetworksEqual(Network.objects.get(id=net_id), api_net,
241
                                     detail=True)
242

    
243
    def test_get_network_building_nics(self, mrapi):
244
        net = mfactory.NetworkFactory()
245
        machine = mfactory.VirtualMachineFactory(userid=net.userid)
246
        mfactory.NetworkInterfaceFactory(network=net, machine=machine,
247
                                         state="BUILDING")
248
        response = self.myget('networks/%d' % net.id, net.userid)
249
        self.assertSuccess(response)
250
        api_net = json.loads(response.content)["network"]
251
        self.assertEqual(len(api_net["attachments"]), 0)
252

    
253
    def test_network_details_1(self, mrapi):
254
        """Test that expected details for a network are returned"""
255
        response = self.myget('networks/%d' % self.net1.id, self.net1.userid)
256
        self.assertSuccess(response)
257
        api_net = json.loads(response.content)["network"]
258
        self.assertNetworksEqual(self.net1, api_net, detail=True)
259

    
260
    def test_invalid_network(self, mrapi):
261
        """Test details for non-existing network."""
262
        response = self.myget('networks/%d' % 42, self.net1.userid)
263
        self.assertItemNotFound(response)
264

    
265
    def test_rename_network(self, mrapi):
266
        request = {'network': {'name': "new_name"}}
267
        response = self.myput('networks/%d' % self.net2.id,
268
                              self.net2.userid, json.dumps(request), 'json')
269
        self.assertEqual(response.status_code, 204)
270
        self.assertEqual(Network.objects.get(id=self.net2.id).name, "new_name")
271
        # Check invalid
272
        request = {'name': "new_name"}
273
        response = self.myput('networks/%d' % self.net2.id,
274
                              self.net2.userid, json.dumps(request), 'json')
275
        self.assertBadRequest(response)
276

    
277
    def test_rename_deleted_network(self, mrapi):
278
        net = mfactory.NetworkFactory(deleted=True)
279
        request = {'network': {'name': "new_name"}}
280
        response = self.myput('networks/%d' % net.id,
281
                              net.userid, json.dumps(request), 'json')
282
        self.assertBadRequest(response)
283

    
284
    def test_rename_public_network(self, mrapi):
285
        net = mfactory.NetworkFactory(public=True)
286
        request = {'network': {'name': "new_name"}}
287
        response = self.myput('networks/%d' % net.id,
288
                              self.net2.userid, json.dumps(request), 'json')
289
        self.assertFault(response, 403, 'forbidden')
290

    
291
    def test_delete_network(self, mrapi):
292
        net = mfactory.NetworkFactory(deleted=False, state='ACTIVE',
293
                                      link="link-10")
294
        with mocked_quotaholder():
295
            response = self.mydelete('networks/%d' % net.id, net.userid)
296
        self.assertEqual(response.status_code, 204)
297
        net = Network.objects.get(id=net.id, userid=net.userid)
298
        self.assertEqual(net.action, 'DESTROY')
299
        mrapi.DeleteNetwork.assert_called()
300

    
301
    def test_delete_public_network(self, mrapi):
302
        net = mfactory.NetworkFactory(public=True)
303
        response = self.mydelete('networks/%d' % net.id, self.net2.userid)
304
        self.assertFault(response, 403, 'forbidden')
305
        self.assertFalse(mrapi.called)
306

    
307
    def test_delete_deleted_network(self, mrapi):
308
        net = mfactory.NetworkFactory(deleted=True)
309
        response = self.mydelete('networks/%d' % net.id, net.userid)
310
        self.assertBadRequest(response)
311

    
312
    def test_delete_network_in_use(self, mrapi):
313
        net = self.net1
314
        response = self.mydelete('networks/%d' % net.id, net.userid)
315
        self.assertFault(response, 421, 'networkInUse')
316
        self.assertFalse(mrapi.called)
317

    
318
    def test_add_nic(self, mrapi):
319
        user = 'userr'
320
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
321
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
322
        mrapi().ModifyInstance.return_value = 1
323
        request = {'add': {'serverRef': vm.id}}
324
        response = self.mypost('networks/%d/action' % net.id,
325
                               net.userid, json.dumps(request), 'json')
326
        self.assertEqual(response.status_code, 202)
327

    
328
    def test_add_nic_to_deleted_network(self, mrapi):
329
        user = 'userr'
330
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
331
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user,
332
                                      deleted=True)
333
        request = {'add': {'serverRef': vm.id}}
334
        response = self.mypost('networks/%d/action' % net.id,
335
                               net.userid, json.dumps(request), 'json')
336
        self.assertBadRequest(response)
337
        self.assertFalse(mrapi.called)
338

    
339
    def test_add_nic_to_public_network(self, mrapi):
340
        user = 'userr'
341
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
342
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user, public=True)
343
        request = {'add': {'serverRef': vm.id}}
344
        response = self.mypost('networks/%d/action' % net.id,
345
                               net.userid, json.dumps(request), 'json')
346
        self.assertFault(response, 403, 'forbidden')
347
        self.assertFalse(mrapi.called)
348

    
349
    def test_add_nic_malformed_1(self, mrapi):
350
        user = 'userr'
351
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
352
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
353
        request = {'add': {'serveRef': vm.id}}
354
        response = self.mypost('networks/%d/action' % net.id,
355
                               net.userid, json.dumps(request), 'json')
356
        self.assertBadRequest(response)
357
        self.assertFalse(mrapi.called)
358

    
359
    def test_add_nic_malformed_2(self, mrapi):
360
        user = 'userr'
361
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
362
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
363
        request = {'add': {'serveRef': [vm.id, 22]}}
364
        response = self.mypost('networks/%d/action' % net.id,
365
                               net.userid, json.dumps(request), 'json')
366
        self.assertBadRequest(response)
367
        self.assertFalse(mrapi.called)
368

    
369
    def test_add_nic_not_active(self, mrapi):
370
        """Test connecting VM to non-active network"""
371
        user = 'dummy'
372
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
373
        net = mfactory.NetworkFactory(state='PENDING', subnet='10.0.0.0/31',
374
                                      userid=user)
375
        request = {'add': {'serverRef': vm.id}}
376
        response = self.mypost('networks/%d/action' % net.id,
377
                               net.userid, json.dumps(request), 'json')
378
        # Test that returns BuildInProgress
379
        self.assertEqual(response.status_code, 409)
380
        self.assertFalse(mrapi.called)
381

    
382
    def test_add_nic_full_network(self, mrapi):
383
        """Test connecting VM to a full network"""
384
        user = 'userr'
385
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user,
386
                                            operstate="STARTED")
387
        net = mfactory.NetworkFactory(state='ACTIVE', subnet='10.0.0.0/30',
388
                                      userid=user, dhcp=True)
389
        pool = net.get_pool()
390
        while not pool.empty():
391
            pool.get()
392
        pool.save()
393
        pool = net.get_pool()
394
        self.assertTrue(pool.empty())
395
        request = {'add': {'serverRef': vm.id}}
396
        response = self.mypost('networks/%d/action' % net.id,
397
                               net.userid, json.dumps(request), 'json')
398
        # Test that returns OverLimit
399
        self.assertEqual(response.status_code, 413)
400
        self.assertFalse(mrapi.called)
401

    
402
    def test_remove_nic(self, mrapi):
403
        user = 'userr'
404
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user,
405
                                            operstate="ACTIVE")
406
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
407
        nic = mfactory.NetworkInterfaceFactory(machine=vm, network=net)
408
        mrapi().ModifyInstance.return_value = 1
409
        request = {'remove': {'attachment': 'nic-%s-%s' % (vm.id, nic.index)}}
410
        response = self.mypost('networks/%d/action' % net.id,
411
                               net.userid, json.dumps(request), 'json')
412
        self.assertEqual(response.status_code, 202)
413
        self.assertTrue(NetworkInterface.objects.get(id=nic.id).dirty)
414
        vm.task = None
415
        vm.task_job_id = None
416
        vm.save()
417
        # Remove dirty nic
418
        response = self.mypost('networks/%d/action' % net.id,
419
                               net.userid, json.dumps(request), 'json')
420
        self.assertFault(response, 409, 'buildInProgress')
421

    
422
    def test_remove_nic_malformed(self, mrapi):
423
        user = 'userr'
424
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
425
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
426
        nic = mfactory.NetworkInterfaceFactory(machine=vm, network=net)
427
        request = {'remove':
428
                    {'att234achment': 'nic-%s-%s' % (vm.id, nic.index)}
429
                  }
430
        response = self.mypost('networks/%d/action' % net.id,
431
                               net.userid, json.dumps(request), 'json')
432
        self.assertBadRequest(response)
433

    
434
    def test_remove_nic_malformed_2(self, mrapi):
435
        user = 'userr'
436
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
437
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
438
        request = {'remove':
439
                    {'attachment': 'nic-%s' % vm.id}
440
                  }
441
        response = self.mypost('networks/%d/action' % net.id,
442
                               net.userid, json.dumps(request), 'json')
443
        self.assertBadRequest(response)
444

    
445
    def test_catch_wrong_api_paths(self, *args):
446
        response = self.myget('nonexistent')
447
        self.assertEqual(response.status_code, 400)
448
        try:
449
            error = json.loads(response.content)
450
        except ValueError:
451
            self.assertTrue(False)
452

    
453
    def test_method_not_allowed(self, *args):
454
        # /networks/ allows only POST, GET
455
        response = self.myput('networks', '', '')
456
        self.assertMethodNotAllowed(response)
457
        response = self.mydelete('networks')
458
        self.assertMethodNotAllowed(response)
459

    
460
        # /networks/<srvid>/ allows only GET, PUT, DELETE
461
        response = self.mypost("networks/42")
462
        self.assertMethodNotAllowed(response)