Statistics
| Branch: | Tag: | Revision:

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

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

    
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(base="link")
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
        with mocked_quotaholder():
78
            response = self.post('/api/v1.1/networks/', 'user1',
79
                                 json.dumps(request), 'json')
80
        self.assertEqual(response.status_code, 202)
81
        db_networks = Network.objects.filter(userid='user1')
82
        self.assertEqual(len(db_networks), 1)
83
        db_net = db_networks[0]
84
        api_net = json.loads(response.content)['network']
85
        self.assertNetworksEqual(db_net, api_net)
86
        mrapi.CreateNetwork.assert_called()
87
        mrapi.ConnectNetwork.assert_called()
88

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    
267
    def test_delete_network(self, mrapi):
268
        net = mfactory.NetworkFactory(deleted=False, state='ACTIVE',
269
                                      link="link-10")
270
        with mocked_quotaholder():
271
            response = self.delete('/api/v1.1/networks/%d' % net.id,
272
                                    net.userid)
273
        self.assertEqual(response.status_code, 204)
274
        net = Network.objects.get(id=net.id, userid=net.userid)
275
        self.assertEqual(net.action, 'DESTROY')
276
        mrapi.DeleteNetwork.assert_called()
277

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

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

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

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

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

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

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

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

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

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

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

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

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