Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / test / networks.py @ 4e1133ab

History | View | Annotate | Download (18.2 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 synnefo.api.tests import BaseAPITest
38
from synnefo.db.models import Network, NetworkInterface
39
from synnefo.db import models_factory as mfactory
40

    
41

    
42
@patch('synnefo.logic.rapi_pool.GanetiRapiClient')
43
class NetworkAPITest(BaseAPITest):
44
    def setUp(self):
45
        self.mac_prefixes = mfactory.MacPrefixPoolTableFactory()
46
        self.bridges = mfactory.BridgePoolTableFactory()
47
        self.user = 'dummy-user'
48
        self.net1 = mfactory.NetworkFactory(userid=self.user)
49
        self.vm1 = mfactory.VirtualMachineFactory(userid=self.user)
50
        self.nic1 = mfactory.NetworkInterfaceFactory(network=self.net1,
51
                                                     machine=self.vm1)
52
        self.nic2 = mfactory.NetworkInterfaceFactory(network=self.net1,
53
                                                     machine=self.vm1)
54
        self.net2 = mfactory.NetworkFactory(userid=self.user)
55
        self.nic3 = mfactory.NetworkInterfaceFactory(network=self.net2)
56

    
57
    def assertNetworksEqual(self, db_net, api_net, detail=False):
58
        self.assertEqual(str(db_net.id), api_net["id"])
59
        self.assertEqual(db_net.name, api_net['name'])
60
        if detail:
61
            self.assertEqual(db_net.state, api_net['status'])
62
            self.assertEqual(db_net.flavor, api_net['type'])
63
            self.assertEqual(db_net.subnet, api_net['cidr'])
64
            self.assertEqual(db_net.subnet6, api_net['cidr6'])
65
            self.assertEqual(db_net.gateway, api_net['gateway'])
66
            self.assertEqual(db_net.gateway6, api_net['gateway6'])
67
            self.assertEqual(db_net.dhcp, api_net['dhcp'])
68
            self.assertEqual(db_net.public, api_net['public'])
69
            db_nics = ["nic-%d-%d" % (nic.machine.id, nic.index) for nic in
70
                       db_net.nics.filter(machine__userid=db_net.userid)]
71
            self.assertEqual(db_nics, api_net['attachments']['values'])
72

    
73
    def test_create_network_1(self, mrapi):
74
        request = {
75
            'network': {'name': 'foo', "type": "MAC_FILTERED"}
76
        }
77
        response = self.post('/api/v1.1/networks/', 'user1',
78
                             json.dumps(request), 'json')
79
        self.assertEqual(response.status_code, 202)
80
        db_networks = Network.objects.filter(userid='user1')
81
        self.assertEqual(len(db_networks), 1)
82
        db_net = db_networks[0]
83
        api_net = json.loads(response.content)['network']
84
        self.assertNetworksEqual(db_net, api_net)
85
        mrapi.CreateNetwork.assert_called()
86
        mrapi.ConnectNetwork.assert_called()
87

    
88
    def test_invalid_data_1(self, mrapi):
89
        """Test invalid flavor"""
90
        request = {
91
            'network': {'name': 'foo', 'type': 'LoLo'}
92
            }
93
        response = self.post('/api/v1.1/networks/', 'user1',
94
                             json.dumps(request), 'json')
95
        self.assertBadRequest(response)
96
        self.assertEqual(len(Network.objects.filter(userid='user1')), 0)
97

    
98
    def test_invalid_data_2(self, mrapi):
99
        """Test invalid data/subnet"""
100
        request = {
101
            'network': {'name': 'foo',
102
                        'cidr': '10.0.0.0/8', "type":
103
                        "MAC_FILTERED"}
104
        }
105
        response = self.post('/api/v1.1/networks/', 'user1',
106
                             json.dumps(request), 'json')
107
        self.assertFault(response, 413, "overLimit")
108

    
109
    def test_invalid_data_3(self, mrapi):
110
        """Test unauthorized to create public network"""
111
        request = {
112
                'network': {'name': 'foo',
113
                            "public": "True",
114
                            "type": "MAC_FILTERED"}
115
            }
116
        response = self.post('/api/v1.1/networks/', 'user1',
117
                             json.dumps(request), 'json')
118
        self.assertFault(response, 403, "forbidden")
119

    
120
    def test_invalid_data_4(self, mrapi):
121
        """Test unauthorized to create network not in settings"""
122
        request = {
123
                'network': {'name': 'foo', 'type': 'CUSTOM'}
124
            }
125
        response = self.post('/api/v1.1/networks/', 'user1',
126
                             json.dumps(request), 'json')
127
        self.assertFault(response, 403, "forbidden")
128

    
129
    def test_invalid_subnet(self, mrapi):
130
        """Test invalid subnet"""
131
        request = {
132
            'network': {'name': 'foo',
133
                        'cidr': '10.0.0.10/27',
134
                        "type": "MAC_FILTERED"}
135
        }
136
        response = self.post('/api/v1.1/networks/', 'user1',
137
                             json.dumps(request), 'json')
138
        self.assertBadRequest(response)
139

    
140
    def test_invalid_gateway_1(self, mrapi):
141
        request = {
142
            'network': {'name': 'foo',
143
                        'cidr': '10.0.0.0/28',
144
                        'gateway': '10.0.0.0.300'}
145
        }
146
        response = self.post('/api/v1.1/networks/', 'user1',
147
                             json.dumps(request), 'json')
148
        self.assertBadRequest(response)
149

    
150
    def test_invalid_gateway_2(self, mrapi):
151
        request = {
152
            'network': {'name': 'foo',
153
                        'cidr': '10.0.0.0/28',
154
                        'gateway': '10.2.0.1'}
155
        }
156
        response = self.post('/api/v1.1/networks/', 'user1',
157
                             json.dumps(request), 'json')
158
        self.assertBadRequest(response)
159

    
160
    def test_invalid_network6(self, mrapi):
161
        request = {
162
            'network': {'name': 'foo',
163
                        'cidr': '10.0.0.0/28',
164
                        'subnet6': '10.0.0.0/28',
165
                        'gateway': '10.2.0.1'}
166
        }
167
        response = self.post('/api/v1.1/networks/', 'user1',
168
                             json.dumps(request), 'json')
169
        self.assertBadRequest(response)
170

    
171
    def test_invalid_gateway6(self, mrapi):
172
        request = {
173
            'network': {'name': 'foo',
174
                        'cidr': '10.0.0.0/28',
175
                        'subnet6': '2001:0db8:0123:4567:89ab:cdef:1234:5678',
176
                        'gateway': '10.2.0.1'}
177
        }
178
        response = self.post('/api/v1.1/networks/', 'user1',
179
                             json.dumps(request), 'json')
180
        self.assertBadRequest(response)
181

    
182
    def test_list_networks(self, mrapi):
183
        """Test that expected list of networks is returned."""
184
        # Create a deleted network
185
        mfactory.NetworkFactory(userid=self.user, deleted=True)
186

    
187
        response = self.get('/api/v1.1/networks/', self.user)
188
        self.assertSuccess(response)
189

    
190
        db_nets = Network.objects.filter(userid=self.user, deleted=False)
191
        api_nets = json.loads(response.content)["networks"]["values"]
192

    
193
        self.assertEqual(len(db_nets), len(api_nets))
194
        for api_net in api_nets:
195
            net_id = api_net['id']
196
            self.assertNetworksEqual(Network.objects.get(id=net_id), api_net)
197

    
198
    def test_list_networks_detail(self, mrapi):
199
        """Test that expected networks details are returned."""
200
        # Create a deleted network
201
        mfactory.NetworkFactory(userid=self.user, deleted=True)
202

    
203
        response = self.get('/api/v1.1/networks/detail', self.user)
204
        self.assertSuccess(response)
205

    
206
        db_nets = Network.objects.filter(userid=self.user, deleted=False)
207
        api_nets = json.loads(response.content)["networks"]["values"]
208

    
209
        self.assertEqual(len(db_nets), len(api_nets))
210
        for api_net in api_nets:
211
            net_id = api_net['id']
212
            self.assertNetworksEqual(Network.objects.get(id=net_id), api_net,
213
                                     detail=True)
214

    
215
    def test_get_network_building_nics(self, mrapi):
216
        net = mfactory.NetworkFactory()
217
        machine = mfactory.VirtualMachineFactory(userid=net.userid)
218
        mfactory.NetworkInterfaceFactory(network=net, machine=machine,
219
                                         state="BUILDING")
220
        response = self.get('/api/v1.1/networks/%d' % net.id,
221
                            net.userid)
222
        self.assertSuccess(response)
223
        api_net = json.loads(response.content)["network"]
224
        self.assertEqual(len(api_net["attachments"]["values"]), 0)
225

    
226
    def test_network_details_1(self, mrapi):
227
        """Test that expected details for a network are returned"""
228
        response = self.get('/api/v1.1/networks/%d' % self.net1.id,
229
                            self.net1.userid)
230
        self.assertSuccess(response)
231
        api_net = json.loads(response.content)["network"]
232
        self.assertNetworksEqual(self.net1, api_net, detail=True)
233

    
234
    def test_invalid_network(self, mrapi):
235
        """Test details for non-existing network."""
236
        response = self.get('/api/v1.1/networks/%d' % 42,
237
                            self.net1.userid)
238
        self.assertItemNotFound(response)
239

    
240
    def test_rename_network(self, mrapi):
241
        request = {'network': {'name': "new_name"}}
242
        response = self.put('/api/v1.1/networks/%d' % self.net2.id,
243
                            self.net2.userid, json.dumps(request), 'json')
244
        self.assertEqual(response.status_code, 204)
245
        self.assertEqual(Network.objects.get(id=self.net2.id).name, "new_name")
246
        # Check invalid
247
        request = {'name': "new_name"}
248
        response = self.put('/api/v1.1/networks/%d' % self.net2.id,
249
                            self.net2.userid, json.dumps(request), 'json')
250
        self.assertBadRequest(response)
251

    
252
    def test_rename_deleted_network(self, mrapi):
253
        net = mfactory.NetworkFactory(deleted=True)
254
        request = {'network': {'name': "new_name"}}
255
        response = self.put('/api/v1.1/networks/%d' % net.id,
256
                            net.userid, json.dumps(request), 'json')
257
        self.assertBadRequest(response)
258

    
259
    def test_rename_public_network(self, mrapi):
260
        net = mfactory.NetworkFactory(public=True)
261
        request = {'network': {'name': "new_name"}}
262
        response = self.put('/api/v1.1/networks/%d' % net.id,
263
                            self.net2.userid, json.dumps(request), 'json')
264
        self.assertFault(response, 403, 'forbidden')
265

    
266
    def test_delete_network(self, mrapi):
267
        net = mfactory.NetworkFactory()
268
        response = self.delete('/api/v1.1/networks/%d' % net.id,
269
                                net.userid)
270
        self.assertEqual(response.status_code, 204)
271
        net = Network.objects.get(id=net.id, userid=net.userid)
272
        self.assertEqual(net.action, 'DESTROY')
273
        mrapi.DeleteNetwork.assert_called()
274

    
275
    def test_delete_public_network(self, mrapi):
276
        net = mfactory.NetworkFactory(public=True)
277
        response = self.delete('/api/v1.1/networks/%d' % net.id,
278
                                self.net2.userid)
279
        self.assertFault(response, 403, 'forbidden')
280
        self.assertFalse(mrapi.called)
281

    
282
    def test_delete_deleted_network(self, mrapi):
283
        net = mfactory.NetworkFactory(deleted=True)
284
        response = self.delete('/api/v1.1/networks/%d' % net.id,
285
                                net.userid)
286
        self.assertBadRequest(response)
287

    
288
    def test_delete_network_in_use(self, mrapi):
289
        net = self.net1
290
        response = self.delete('/api/v1.1/networks/%d' % net.id,
291
                                net.userid)
292
        self.assertFault(response, 421, 'networkInUse')
293
        self.assertFalse(mrapi.called)
294

    
295
    def test_add_nic(self, mrapi):
296
        user = 'userr'
297
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
298
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
299
        request = {'add': {'serverRef': vm.id}}
300
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
301
                             net.userid, json.dumps(request), 'json')
302
        self.assertEqual(response.status_code, 202)
303

    
304
    def test_add_nic_to_deleted_network(self, mrapi):
305
        user = 'userr'
306
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
307
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user,
308
                                      deleted=True)
309
        request = {'add': {'serverRef': vm.id}}
310
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
311
                             net.userid, json.dumps(request), 'json')
312
        self.assertBadRequest(response)
313
        self.assertFalse(mrapi.called)
314

    
315
    def test_add_nic_to_public_network(self, mrapi):
316
        user = 'userr'
317
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
318
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user, public=True)
319
        request = {'add': {'serverRef': vm.id}}
320
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
321
                             net.userid, json.dumps(request), 'json')
322
        self.assertFault(response, 403, 'forbidden')
323
        self.assertFalse(mrapi.called)
324

    
325
    def test_add_nic_malformed_1(self, mrapi):
326
        user = 'userr'
327
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
328
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
329
        request = {'add': {'serveRef': vm.id}}
330
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
331
                             net.userid, json.dumps(request), 'json')
332
        self.assertBadRequest(response)
333
        self.assertFalse(mrapi.called)
334

    
335
    def test_add_nic_malformed_2(self, mrapi):
336
        user = 'userr'
337
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
338
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
339
        request = {'add': {'serveRef': [vm.id, 22]}}
340
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
341
                             net.userid, json.dumps(request), 'json')
342
        self.assertBadRequest(response)
343
        self.assertFalse(mrapi.called)
344

    
345
    def test_add_nic_not_active(self, mrapi):
346
        """Test connecting VM to non-active network"""
347
        user = 'dummy'
348
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
349
        net = mfactory.NetworkFactory(state='PENDING', subnet='10.0.0.0/31',
350
                                      userid=user)
351
        request = {'add': {'serveRef': vm.id}}
352
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
353
                             net.userid, json.dumps(request), 'json')
354
        # Test that returns BuildInProgress
355
        self.assertEqual(response.status_code, 409)
356
        self.assertFalse(mrapi.called)
357

    
358
    def test_add_nic_full_network(self, mrapi):
359
        """Test connecting VM to a full network"""
360
        user = 'userr'
361
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
362
        net = mfactory.NetworkFactory(state='ACTIVE', subnet='10.0.0.0/30',
363
                                      userid=user, dhcp=True)
364
        pool = net.get_pool()
365
        while not pool.empty():
366
            pool.get()
367
        pool.save()
368
        pool = net.get_pool()
369
        self.assertTrue(pool.empty())
370
        request = {'add': {'serverRef': vm.id}}
371
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
372
                             net.userid, json.dumps(request), 'json')
373
        # Test that returns OverLimit
374
        self.assertEqual(response.status_code, 413)
375
        self.assertFalse(mrapi.called)
376

    
377
    def test_remove_nic(self, mrapi):
378
        user = 'userr'
379
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
380
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
381
        nic = mfactory.NetworkInterfaceFactory(machine=vm, network=net)
382
        request = {'remove': {'attachment': 'nic-%s-%s' % (vm.id, nic.index)}}
383
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
384
                             net.userid, json.dumps(request), 'json')
385
        self.assertEqual(response.status_code, 202)
386
        self.assertTrue(NetworkInterface.objects.get(id=nic.id).dirty)
387
        # Remove dirty nic
388
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
389
                             net.userid, json.dumps(request), 'json')
390
        self.assertFault(response, 409, 'buildInProgress')
391

    
392
    def test_remove_nic_malformed(self, mrapi):
393
        user = 'userr'
394
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
395
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
396
        nic = mfactory.NetworkInterfaceFactory(machine=vm, network=net)
397
        request = {'remove':
398
                    {'att234achment': 'nic-%s-%s' % (vm.id, nic.index)}
399
                  }
400
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
401
                             net.userid, json.dumps(request), 'json')
402
        self.assertBadRequest(response)
403

    
404
    def test_remove_nic_malformed_2(self, mrapi):
405
        user = 'userr'
406
        vm = mfactory.VirtualMachineFactory(name='yo', userid=user)
407
        net = mfactory.NetworkFactory(state='ACTIVE', userid=user)
408
        request = {'remove':
409
                    {'attachment': 'nic-%s' % vm.id}
410
                  }
411
        response = self.post('/api/v1.1/networks/%d/action' % net.id,
412
                             net.userid, json.dumps(request), 'json')
413
        self.assertBadRequest(response)