Statistics
| Branch: | Tag: | Revision:

root / kamaki / clients / test / cyclades.py @ 5ad6a119

History | View | Annotate | Download (19.9 kB)

1
# Copyright 2012-2013 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
from mock import patch, Mock
34
from unittest import TestCase
35
from json import loads
36

    
37
from kamaki.clients import Client, ClientError
38
from kamaki.clients.cyclades_rest_api import CycladesClientApi
39

    
40
img_ref = "1m4g3-r3f3r3nc3"
41
vm_name = "my new VM"
42
fid = 42
43
vm_send = dict(server=dict(
44
    flavorRef=fid,
45
    name=vm_name,
46
    imageRef=img_ref,
47
    metadata=dict(os="debian", users="root")))
48
vm_recv = dict(server=dict(
49
    status="BUILD",
50
    updated="2013-03-01T10:04:00.637152+00:00",
51
    hostId="",
52
    name=vm_name,
53
    imageRef=img_ref,
54
    created="2013-03-01T10:04:00.087324+00:00",
55
    flavorRef=fid,
56
    adminPass="n0n3sh@11p@55",
57
    suspended=False,
58
    progress=0,
59
    id=31173,
60
    metadata=dict(values=dict(os="debian", users="root"))))
61
img_recv = dict(image=dict(
62
    status="ACTIVE",
63
    updated="2013-02-26T11:10:14+00:00",
64
    name="Debian Base",
65
    created="2013-02-26T11:03:29+00:00",
66
    progress=100,
67
    id=img_ref,
68
    metadata=dict(values=dict(
69
        partition_table="msdos",
70
        kernel="2.6.32",
71
        osfamily="linux",
72
        users="root",
73
        gui="No GUI",
74
        sortorder="1",
75
        os="debian",
76
        root_partition="1",
77
        description="Debian 6.0.7 (Squeeze) Base System"))))
78
vm_list = dict(servers=dict(values=[
79
    dict(name='n1', id=1),
80
    dict(name='n2', id=2)]))
81

    
82

    
83
class Cyclades(TestCase):
84

    
85
    def assert_dicts_are_equal(self, d1, d2):
86
        for k, v in d1.items():
87
            self.assertTrue(k in d2)
88
            if isinstance(v, dict):
89
                self.assert_dicts_are_equal(v, d2[k])
90
            else:
91
                self.assertEqual(unicode(v), unicode(d2[k]))
92

    
93
    class FR(object):
94
        """FR stands for Fake Response"""
95
        json = vm_recv
96
        headers = {}
97
        content = json
98
        status = None
99
        status_code = 200
100

    
101
        def release(self):
102
            pass
103

    
104
    """Set up a Cyclades thorough test"""
105
    def setUp(self):
106
        self.url = 'http://cyclades.example.com'
107
        self.token = 'cyc14d3s70k3n'
108
        from kamaki.clients.cyclades import CycladesClient
109
        self.client = CycladesClient(self.url, self.token)
110
        from kamaki.clients.connection.kamakicon import KamakiHTTPConnection
111
        self.C = KamakiHTTPConnection
112

    
113
    def tearDown(self):
114
        self.FR.status_code = 200
115
        self.FR.json = vm_recv
116

    
117
    def test_create_server(self):
118
        self.client.get_image_details = Mock(return_value=img_recv['image'])
119
        with patch.object(Client, 'request', side_effect=ClientError(
120
                'REQUEST ENTITY TOO LARGE',
121
                status=403)):
122
            self.assertRaises(
123
                ClientError,
124
                self.client.create_server,
125
                vm_name, fid, img_ref)
126

    
127
        with patch.object(
128
                self.C,
129
                'perform_request',
130
                return_value=self.FR()) as perform_req:
131
            self.assertRaises(
132
                ClientError,
133
                self.client.create_server,
134
                vm_name, fid, img_ref)
135
            self.FR.status_code = 202
136
            r = self.client.create_server(vm_name, fid, img_ref)
137
            self.assertEqual(self.client.http_client.url, self.url)
138
            self.assertEqual(self.client.http_client.path, '/servers')
139
            (method, data, a_headers, a_params) = perform_req.call_args[0]
140
            self.assert_dicts_are_equal(loads(data), vm_send)
141
            self.assert_dicts_are_equal(r, vm_recv['server'])
142
            prsn = 'Personality string (does not work with real servers)'
143
            self.client.create_server(vm_name, fid, img_ref, prsn)
144
            (method, data, a_headers, a_params) = perform_req.call_args[0]
145
            data = loads(data)
146
            self.assertTrue('personality' in data['server'])
147
            self.assertEqual(prsn, data['server']['personality'])
148

    
149
    def test_list_servers(self):
150
        self.FR.json = vm_list
151
        with patch.object(
152
                self.C,
153
                'perform_request',
154
                return_value=self.FR()) as perform_req:
155
            r = self.client.list_servers()
156
            self.assertEqual(self.client.http_client.url, self.url)
157
            self.assertEqual(self.client.http_client.path, '/servers')
158
            (method, data, a_headers, a_params) = perform_req.call_args[0]
159
            self.assert_dicts_are_equal(dict(values=r), vm_list['servers'])
160
            r = self.client.list_servers(detail=True)
161
            self.assertEqual(self.client.http_client.url, self.url)
162
            self.assertEqual(self.client.http_client.path, '/servers/detail')
163
        with patch.object(
164
                CycladesClientApi,
165
                'servers_get',
166
                return_value=self.FR()) as servers_get:
167
            self.client.list_servers(changes_since=True)
168
            self.assertTrue(servers_get.call_args[1]['changes_since'])
169

    
170
    def test_get_server_details(self):
171
        vm_id = vm_recv['server']['id']
172
        with patch.object(self.C, 'perform_request', return_value=self.FR()):
173
            r = self.client.get_server_details(vm_id)
174
            self.assertEqual(self.client.http_client.url, self.url)
175
            self.assertEqual(
176
                self.client.http_client.path,
177
                '/servers/%s' % vm_id)
178
            self.assert_dicts_are_equal(r, vm_recv['server'])
179

    
180
    def test_update_server_name(self):
181
        vm_id = vm_recv['server']['id']
182
        new_name = vm_name + '_new'
183
        self.FR.status_code = 204
184
        with patch.object(
185
                self.C,
186
                'perform_request',
187
                return_value=self.FR()) as perform_req:
188
            self.client.update_server_name(vm_id, new_name)
189
            self.assertEqual(self.client.http_client.url, self.url)
190
            self.assertEqual(
191
                self.client.http_client.path,
192
                '/servers/%s' % vm_id)
193
            (method, data, a_headers, a_params) = perform_req.call_args[0]
194
            self.assert_dicts_are_equal(
195
                dict(server=dict(name=new_name)),
196
                loads(data))
197

    
198
    def test_reboot_server(self):
199
        vm_id = vm_recv['server']['id']
200
        self.FR.status_code = 202
201
        with patch.object(
202
                self.C,
203
                'perform_request',
204
                return_value=self.FR()) as perform_req:
205
            self.client.reboot_server(vm_id)
206
            self.assertEqual(self.client.http_client.url, self.url)
207
            self.assertEqual(
208
                self.client.http_client.path,
209
                '/servers/%s/action' % vm_id)
210
            (method, data, a_headers, a_params) = perform_req.call_args[0]
211
            self.assert_dicts_are_equal(
212
                dict(reboot=dict(type='SOFT')),
213
                loads(data))
214

    
215
    def test_create_server_metadata(self):
216
        vm_id = vm_recv['server']['id']
217
        metadata = dict(m1='v1', m2='v2', m3='v3')
218
        self.FR.json = dict(meta=vm_recv['server'])
219
        with patch.object(
220
                self.C,
221
                'perform_request',
222
                return_value=self.FR()) as perform_req:
223
            self.assertRaises(
224
                ClientError,
225
                self.client.create_server_metadata,
226
                vm_id, 'key', 'value')
227
            self.FR.status_code = 201
228
            for k, v in metadata.items():
229
                r = self.client.create_server_metadata(vm_id, k, v)
230
                self.assertEqual(self.client.http_client.url, self.url)
231
                self.assertEqual(
232
                    self.client.http_client.path,
233
                    '/servers/%s/meta/%s' % (vm_id, k))
234
                (method, data, a_headers, a_params) = perform_req.call_args[0]
235
                self.assertEqual(dict(meta={k: v}), loads(data))
236
                self.assert_dicts_are_equal(r, vm_recv['server'])
237

    
238
    def test_get_server_metadata(self):
239
        vm_id = vm_recv['server']['id']
240
        metadata = dict(m1='v1', m2='v2', m3='v3')
241
        with patch.object(self.C, 'perform_request', return_value=self.FR()):
242
            self.FR.json = dict(metadata=dict(values=metadata))
243
            r = self.client.get_server_metadata(vm_id)
244
            self.assertEqual(self.client.http_client.url, self.url)
245
            self.assertEqual(
246
                self.client.http_client.path,
247
                '/servers/%s/meta' % vm_id)
248
            self.assert_dicts_are_equal(r, metadata)
249

    
250
            for k, v in metadata.items():
251
                self.FR.json = dict(meta={k: v})
252
                r = self.client.get_server_metadata(vm_id, k)
253
                self.assertEqual(self.client.http_client.url, self.url)
254
                self.assertEqual(
255
                    self.client.http_client.path,
256
                    '/servers/%s/meta/%s' % (vm_id, k))
257
                self.assert_dicts_are_equal(r, {k: v})
258

    
259
    def test_update_server_metadata(self):
260
        vm_id = vm_recv['server']['id']
261
        metadata = dict(m1='v1', m2='v2', m3='v3')
262
        self.FR.json = dict(metadata=metadata)
263
        with patch.object(
264
                CycladesClientApi,
265
                'servers_post',
266
                return_value=self.FR()) as servers_post:
267
            r = self.client.update_server_metadata(vm_id, **metadata)
268
            self.assert_dicts_are_equal(r, metadata)
269
            (called_id, cmd) = servers_post.call_args[0]
270
            self.assertEqual(called_id, vm_id)
271
            self.assertEqual(cmd, 'meta')
272
            data = servers_post.call_args[1]['json_data']
273
            self.assert_dicts_are_equal(data, dict(metadata=metadata))
274

    
275
    def test_delete_server_metadata(self):
276
        vm_id = vm_recv['server']['id']
277
        key = 'metakey'
278
        with patch.object(
279
                CycladesClientApi,
280
                'servers_delete',
281
                return_value=self.FR()) as servers_delete:
282
            self.client.delete_server_metadata(vm_id, key)
283
            self.assertEqual(
284
                (vm_id, 'meta/' + key),
285
                servers_delete.call_args[0])
286

    
287
    """
288
    def test_list_flavors(self):
289
        r = self.client.list_flavors()
290
        self.assertTrue(len(r) > 1)
291
        r = self.client.list_flavors(detail=True)
292
        self.assertTrue('SNF:disk_template' in r[0])
293

294
    def test_get_flavor_details(self):
295
        r = self.client.get_flavor_details(self.flavorid)
296
        self.assert_dicts_are_equal(self._flavor_details, r)
297

298
    def test_list_images(self):
299
        r = self.client.list_images()
300
        self.assertTrue(len(r) > 1)
301
        r = self.client.list_images(detail=True)
302
        for detailed_img in r:
303
            if detailed_img['id'] == self.img:
304
                break
305
        self.assert_dicts_are_equal(detailed_img, self.img_details)
306

307
    def test_get_image_details(self):
308
        r = self.client.get_image_details(self.img)
309
        self.assert_dicts_are_equal(r, self.img_details)
310

311
    def test_get_image_metadata(self):
312
        r = self.client.get_image_metadata(self.img)
313
        self.assert_dicts_are_equal(
314
            self.img_details['metadata']['values'], r)
315
        for key, val in self.img_details['metadata']['values'].items():
316
            r = self.client.get_image_metadata(self.img, key)
317
            self.assertEqual(r[key], val)
318

319
    def test_shutdown_server(self):
320
        self.client.shutdown_server(self.server1['id'])
321
        self._wait_for_status(self.server1['id'], 'ACTIVE')
322
        r = self.client.get_server_details(self.server1['id'])
323
        self.assertEqual(r['status'], 'STOPPED')
324

325
    def test_start_server(self):
326
        self.client.start_server(self.server1['id'])
327
        self._wait_for_status(self.server1['id'], 'STOPPED')
328
        r = self.client.get_server_details(self.server1['id'])
329
        self.assertEqual(r['status'], 'ACTIVE')
330

331
    def test_get_server_console(self):
332
        r = self.client.get_server_console(self.server2['id'])
333
        self.assertTrue('host' in r)
334
        self.assertTrue('password' in r)
335
        self.assertTrue('port' in r)
336
        self.assertTrue('type' in r)
337

338
    def test_get_firewall_profile(self):
339
        self._wait_for_status(self.server1['id'], 'BUILD')
340
        fprofile = self.client.get_firewall_profile(self.server1['id'])
341
        self.assertTrue(fprofile in self.PROFILES)
342

343
    def test_set_firewall_profile(self):
344
        self._wait_for_status(self.server1['id'], 'BUILD')
345
        PROFILES = ['DISABLED', 'ENABLED', 'DISABLED', 'PROTECTED']
346
        fprofile = self.client.get_firewall_profile(self.server1['id'])
347
        print('')
348
        count_success = 0
349
        for counter, fprofile in enumerate(PROFILES):
350
            npos = counter + 1
351
            try:
352
                nprofile = PROFILES[npos]
353
            except IndexError:
354
                nprofile = PROFILES[0]
355
            print('\tprofile swap %s: %s -> %s' % (npos, fprofile, nprofile))
356
            self.client.set_firewall_profile(self.server1['id'], nprofile)
357
            time.sleep(0.5)
358
            self.client.reboot_server(self.server1['id'], hard=True)
359
            time.sleep(1)
360
            self._wait_for_status(self.server1['id'], 'REBOOT')
361
            time.sleep(0.5)
362
            changed = self.client.get_firewall_profile(self.server1['id'])
363
            try:
364
                self.assertEqual(changed, nprofile)
365
            except AssertionError as err:
366
                if count_success:
367
                    print('\tFAIL in swap #%s' % npos)
368
                    break
369
                else:
370
                    raise err
371
            count_success += 1
372

373
    def test_get_server_stats(self):
374
        r = self.client.get_server_stats(self.server1['id'])
375
        it = ('cpuBar', 'cpuTimeSeries', 'netBar', 'netTimeSeries', 'refresh')
376
        for term in it:
377
            self.assertTrue(term in r)
378

379
    def test_create_network(self):
380
        print('\twith no params')
381
        self.network1 = self._create_network(self.netname1)
382
        self._wait_for_network(self.network1['id'], 'ACTIVE')
383
        n1id = self.network1['id']
384
        self.network1 = self.client.get_network_details(n1id)
385
        nets = self.client.list_networks(self.network1['id'])
386
        chosen = [net for net in nets if net['id'] == n1id][0]
387
        chosen.pop('updated')
388
        net1 = dict(self.network1)
389
        net1.pop('updated')
390
        self.assert_dicts_are_equal(chosen, net1)
391
        for param, val in dict(
392
                cidr='192.168.0.0/24',
393
                gateway='192.168.0.1',
394
                type='MAC_FILTERED',
395
                dhcp=True).items():
396
            print('\tdelete %s to avoid max net limit' % n1id)
397
            self._delete_network(n1id)
398
            kwargs = {param: val}
399
            print('\twith %s=%s' % (param, val))
400
            self.network1 = self._create_network(self.netname1, **kwargs)
401
            n1id = self.network1['id']
402
            self._wait_for_network(n1id, 'ACTIVE')
403
            self.network1 = self.client.get_network_details(n1id)
404
            self.assertEqual(self.network1[param], val)
405

406
    def test_connect_server(self):
407
        self.client.connect_server(self.server1['id'], self.network1['id'])
408
        self.assertTrue(self._wait_for_nic(
409
            self.network1['id'],
410
            self.server1['id']))
411

412
    def test_disconnect_server(self):
413
        self.client.disconnect_server(self.server1['id'], self.network1['id'])
414
        self.assertTrue(self._wait_for_nic(
415
            self.network1['id'],
416
            self.server1['id'],
417
            in_creation=False))
418

419
    def _test_0260_wait_for_second_network(self):
420
        self.server1 = self._create_server(
421
            self.servname1,
422
            self.flavorid,
423
            self.img)
424
        self.network2 = self._create_network(self.netname2)
425
        self._wait_for_status(self.server1['id'], 'BUILD')
426
        self._wait_for_network(self.network2['id'], 'ACTIVE')
427
        self._test_0280_list_server_nics()
428

429
    def _test_0280_list_server_nics(self):
430
        r = self.client.list_server_nics(self.server1['id'])
431
        len0 = len(r)
432
        self.client.connect_server(self.server1['id'], self.network2['id'])
433
        self.assertTrue(self._wait_for_nic(
434
            self.network2['id'],
435
            self.server1['id']))
436
        r = self.client.list_server_nics(self.server1['id'])
437
        self.assertTrue(len(r) > len0)
438

439
    def test_list_networks(self):
440
        r = self.client.list_networks()
441
        self.assertTrue(len(r) > 1)
442
        ids = [net['id'] for net in r]
443
        names = [net['name'] for net in r]
444
        self.assertTrue('1' in ids)
445
        #self.assertTrue('public' in names)
446
        self.assertTrue(self.network1['id'] in ids)
447
        self.assertTrue(self.network1['name'] in names)
448

449
        r = self.client.list_networks(detail=True)
450
        ids = [net['id'] for net in r]
451
        names = [net['name'] for net in r]
452
        for net in r:
453
            self.assertTrue(net['id'] in ids)
454
            self.assertTrue(net['name'] in names)
455
            for term in ('status', 'updated', 'created'):
456
                self.assertTrue(term in net.keys())
457

458
    def test_get_network_details(self):
459
        r = self.client.get_network_details(self.network1['id'])
460
        net1 = dict(self.network1)
461
        net1.pop('status')
462
        net1.pop('updated', None)
463
        net1.pop('attachments')
464
        r.pop('status')
465
        r.pop('updated', None)
466
        r.pop('attachments')
467
        self.assert_dicts_are_equal(net1, r)
468

469
    def test_update_network_name(self):
470
        updated_name = self.netname2 + '_upd'
471
        self.client.update_network_name(self.network2['id'], updated_name)
472

473
        def netwait(wait):
474
            r = self.client.get_network_details(self.network2['id'])
475
            if r['name'] == updated_name:
476
                return
477
            time.sleep(wait)
478
        self.do_with_progress_bar(
479
            netwait,
480
            'Network %s name is changing:' % self.network2['id'],
481
            self._waits[:5])
482

483
        r = self.client.get_network_details(self.network2['id'])
484
        self.assertEqual(r['name'], updated_name)
485

486
    def test_delete_image(self):
487
        images = self.client.list_images()
488
        self.client.delete_image(images[2]['id'])
489
        try:
490
            r = self.client.get_image_details(images[2]['id'], success=(400))
491
        except ClientError as err:
492
            self.assertEqual(err.status, 404)
493

494
    def test_create_image_metadata(self):
495
        r = self.client.create_image_metadata(self.img, 'mykey', 'myval')
496
        self.assertEqual(r['mykey'], 'myval')
497

498
    def test_update_image_metadata(self):
499
        r = self.client.create_image_metadata(self.img, 'mykey0', 'myval')
500
        r = self.client.update_image_metadata(self.img, 'mykey0', 'myval0')
501
        self.assertEqual(r['mykey0'], 'myval0')
502

503
    def test_delete_image_metadata(self):
504
        self.client.create_image_metadata(self.img, 'mykey1', 'myval1')
505
        self.client.delete_image_metadata(self.img, 'mykey1')
506
        r = self.client.get_image_metadata(self.img)
507
        self.assertNotEqual('mykey1' in r)
508
    """