Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / api / tests / images.py @ d984eedc

History | View | Annotate | Download (16.5 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.lib.api import faults
37
from snf_django.utils.testing import BaseAPITest
38
from synnefo.lib.services import get_service_path
39
from synnefo.cyclades_settings import cyclades_services
40
from synnefo.lib import join_urls
41

    
42
from mock import patch
43
from functools import wraps
44

    
45

    
46
def assert_backend_closed(func):
47
    """Decorator for ensuring that ImageBackend is returned to pool."""
48
    @wraps(func)
49
    def wrapper(self, backend):
50
        result = func(self, backend)
51
        if backend.called is True:
52
            backend.return_value.close.asssert_called
53
        return result
54
    return wrapper
55

    
56

    
57
class ComputeAPITest(BaseAPITest):
58
    def setUp(self, *args, **kwargs):
59
        super(ComputeAPITest, self).setUp(*args, **kwargs)
60
        self.compute_path = get_service_path(cyclades_services, 'compute',
61
                                             version='v2.0')
62
    def myget(self, path, *args, **kwargs):
63
        path = join_urls(self.compute_path, path)
64
        return self.get(path, *args, **kwargs)
65

    
66
    def myput(self, path, *args, **kwargs):
67
        path = join_urls(self.compute_path, path)
68
        return self.put(path, *args, **kwargs)
69

    
70
    def mypost(self, path, *args, **kwargs):
71
        path = join_urls(self.compute_path, path)
72
        return self.post(path, *args, **kwargs)
73

    
74
    def mydelete(self, path, *args, **kwargs):
75
        path = join_urls(self.compute_path, path)
76
        return self.delete(path, *args, **kwargs)
77

    
78

    
79
@patch('synnefo.plankton.backend.ImageBackend')
80
class ImageAPITest(ComputeAPITest):
81
    @assert_backend_closed
82
    def test_create_image(self, mimage):
83
        """Test that create image is not implemented"""
84
        response = self.mypost('images/', 'user', json.dumps(''), 'json')
85
        self.assertEqual(response.status_code, 501)
86

    
87
    @assert_backend_closed
88
    def test_list_images(self, mimage):
89
        """Test that expected list of images is returned"""
90
        images = [{'id': 1, 'name': 'image-1'},
91
                  {'id': 2, 'name': 'image-2'},
92
                  {'id': 3, 'name': 'image-3'}]
93
        mimage().list_images.return_value = images
94
        response = self.myget('images', 'user')
95
        self.assertSuccess(response)
96
        api_images = json.loads(response.content)['images']
97
        self.assertEqual(images, api_images)
98

    
99
    @assert_backend_closed
100
    def test_list_images_detail(self, mimage):
101
        self.maxDiff = None
102
        images = [{'id': 1,
103
                   'name': 'image-1',
104
                   'status': 'available',
105
                   'created_at': '2012-11-26 11:52:54',
106
                   'updated_at': '2012-12-26 11:52:54',
107
                   'owner': 'user1',
108
                   'deleted_at': '',
109
                   'is_snapshot': False,
110
                   'is_public': True,
111
                   'properties': {'foo': 'bar'}},
112
                  {'id': 2,
113
                   'name': 'image-2',
114
                   'status': 'deleted',
115
                   'created_at': '2012-11-26 11:52:54',
116
                   'updated_at': '2012-12-26 11:52:54',
117
                   'owner': 'user1',
118
                   'deleted_at': '2012-12-27 11:52:54',
119
                   'is_snapshot': False,
120
                   'is_public': True,
121
                   'properties': ''},
122
                  {'id': 3,
123
                   'name': 'image-3',
124
                   'status': 'available',
125
                   'created_at': '2012-11-26 11:52:54',
126
                   'deleted_at': '',
127
                   'updated_at': '2012-12-26 11:52:54',
128
                   'owner': 'user1',
129
                   'is_snapshot': False,
130
                   'is_public': False,
131
                   'properties': ''}]
132
        result_images = [
133
                  {'id': 1,
134
                   'name': 'image-1',
135
                   'status': 'ACTIVE',
136
                   'progress': 100,
137
                   'created': '2012-11-26T11:52:54+00:00',
138
                   'updated': '2012-12-26T11:52:54+00:00',
139
                   'user_id': 'user1',
140
                   'tenant_id': 'user1',
141
                   'is_snapshot': False,
142
                   'public': True,
143
                   'metadata': {'foo': 'bar'}},
144
                  {'id': 2,
145
                   'name': 'image-2',
146
                   'status': 'DELETED',
147
                   'progress': 0,
148
                   'user_id': 'user1',
149
                   'tenant_id': 'user1',
150
                   'created': '2012-11-26T11:52:54+00:00',
151
                   'updated': '2012-12-26T11:52:54+00:00',
152
                   'is_snapshot': False,
153
                   'public': True,
154
                   'metadata': {}},
155
                  {'id': 3,
156
                   'name': 'image-3',
157
                   'status': 'ACTIVE',
158
                   'progress': 100,
159
                   'user_id': 'user1',
160
                   'tenant_id': 'user1',
161
                   'created': '2012-11-26T11:52:54+00:00',
162
                   'updated': '2012-12-26T11:52:54+00:00',
163
                   'is_snapshot': False,
164
                   'public': False,
165
                   'metadata': {}}]
166
        mimage().list_images.return_value = images
167
        response = self.myget('images/detail', 'user')
168
        self.assertSuccess(response)
169
        api_images = json.loads(response.content)['images']
170
        self.assertEqual(len(result_images), len(api_images))
171
        map(lambda image: image.pop("links"), api_images)
172
        self.assertEqual(result_images, api_images)
173

    
174
    @assert_backend_closed
175
    def test_list_images_detail_since(self, mimage):
176
        from datetime import datetime, timedelta
177
        from time import sleep
178
        old_time = datetime.now()
179
        new_time = old_time + timedelta(seconds=0.1)
180
        sleep(0.1)
181
        images = [
182
                  {'id': 1,
183
                   'name': 'image-1',
184
                   'status': 'available',
185
                   'progress': 100,
186
                   'created_at': old_time.isoformat(),
187
                   'deleted_at': '',
188
                   'updated_at': old_time.isoformat(),
189
                   'owner': 'user1',
190
                   'is_snapshot': False,
191
                   'is_public': True,
192
                   'properties': ''},
193
                  {'id': 2,
194
                   'name': 'image-2',
195
                   'status': 'deleted',
196
                   'progress': 0,
197
                   'owner': 'user2',
198
                   'created_at': new_time.isoformat(),
199
                   'updated_at': new_time.isoformat(),
200
                   'deleted_at': new_time.isoformat(),
201
                   'is_snapshot': False,
202
                   'is_public': False,
203
                   'properties': ''}]
204
        mimage().list_images.return_value = images
205
        response =\
206
            self.myget('images/detail?changes-since=%sUTC' % new_time)
207
        self.assertSuccess(response)
208
        api_images = json.loads(response.content)['images']
209
        self.assertEqual(1, len(api_images))
210

    
211
    @assert_backend_closed
212
    def test_get_image_details(self, mimage):
213
        self.maxDiff = None
214
        image = {'id': 42,
215
                 'name': 'image-1',
216
                 'status': 'available',
217
                 'created_at': '2012-11-26 11:52:54',
218
                 'updated_at': '2012-12-26 11:52:54',
219
                 'deleted_at': '',
220
                 'owner': 'user1',
221
                 'is_snapshot': False,
222
                 'is_public': True,
223
                 'properties': {'foo': 'bar'}}
224
        result_image = \
225
                  {'id': 42,
226
                   'name': 'image-1',
227
                   'status': 'ACTIVE',
228
                   'progress': 100,
229
                   'created': '2012-11-26T11:52:54+00:00',
230
                   'updated': '2012-12-26T11:52:54+00:00',
231
                   'user_id': 'user1',
232
                   'tenant_id': 'user1',
233
                   'is_snapshot': False,
234
                   'public': True,
235
                   'metadata': {'foo': 'bar'}}
236
        mimage.return_value.get_image.return_value = image
237
        response = self.myget('images/42', 'user')
238
        self.assertSuccess(response)
239
        api_image = json.loads(response.content)['image']
240
        api_image.pop("links")
241
        self.assertEqual(api_image, result_image)
242

    
243
    @assert_backend_closed
244
    def test_invalid_image(self, mimage):
245
        mimage.return_value.get_image.side_effect = \
246
            faults.ItemNotFound('Image not found')
247
        response = self.myget('images/42', 'user')
248
        self.assertItemNotFound(response)
249

    
250
    @assert_backend_closed
251
    def test_delete_image(self, mimage):
252
        response = self.mydelete("images/42", "user")
253
        self.assertEqual(response.status_code, 204)
254
        mimage.return_value.unregister.assert_called_once_with('42')
255
        mimage.return_value._delete.assert_not_called('42')
256

    
257
    @assert_backend_closed
258
    def test_catch_wrong_api_paths(self, *args):
259
        response = self.myget('nonexistent')
260
        self.assertEqual(response.status_code, 400)
261
        try:
262
            json.loads(response.content)
263
        except ValueError:
264
            self.assertTrue(False)
265

    
266
    @assert_backend_closed
267
    def test_method_not_allowed(self, *args):
268
        # /images/ allows only POST, GET
269
        response = self.myput('images', '', '')
270
        self.assertMethodNotAllowed(response)
271
        response = self.mydelete('images')
272
        self.assertMethodNotAllowed(response)
273

    
274
        # /images/<imgid>/ allows only GET, DELETE
275
        response = self.mypost("images/42")
276
        self.assertMethodNotAllowed(response)
277
        response = self.myput('images/42', '', '')
278
        self.assertMethodNotAllowed(response)
279

    
280
        # /images/<imgid>/metadata/ allows only POST, GET
281
        response = self.myput('images/42/metadata', '', '')
282
        self.assertMethodNotAllowed(response)
283
        response = self.mydelete('images/42/metadata')
284
        self.assertMethodNotAllowed(response)
285

    
286
        # /images/<imgid>/metadata/ allows only POST, GET
287
        response = self.myput('images/42/metadata', '', '')
288
        self.assertMethodNotAllowed(response)
289
        response = self.mydelete('images/42/metadata')
290
        self.assertMethodNotAllowed(response)
291

    
292
        # /images/<imgid>/metadata/<key> allows only PUT, GET, DELETE
293
        response = self.mypost('images/42/metadata/foo')
294
        self.assertMethodNotAllowed(response)
295

    
296

    
297
@patch('synnefo.plankton.backend.ImageBackend')
298
class ImageMetadataAPITest(ComputeAPITest):
299
    def setUp(self):
300
        self.image = {'id': 42,
301
                      'name': 'image-1',
302
                      'status': 'available',
303
                      'created_at': '2012-11-26 11:52:54',
304
                      'updated_at': '2012-12-26 11:52:54',
305
                      'deleted_at': '',
306
                      'properties': {'foo': 'bar', 'foo2': 'bar2'}}
307
        self.result_image = \
308
            {'id': 42,
309
             'name': 'image-1',
310
             'status': 'ACTIVE',
311
             'progress': 100,
312
             'created': '2012-11-26T11:52:54+00:00',
313
             'updated': '2012-12-26T11:52:54+00:00',
314
             'metadata': {'foo': 'bar'}}
315
        super(ImageMetadataAPITest, self).setUp()
316

    
317
    @assert_backend_closed
318
    def test_list_metadata(self, backend):
319
        backend.return_value.get_image.return_value = self.image
320
        response = self.myget('images/42/metadata', 'user')
321
        self.assertSuccess(response)
322
        meta = json.loads(response.content)['metadata']
323
        self.assertEqual(meta, self.image['properties'])
324

    
325
    @assert_backend_closed
326
    def test_get_metadata(self, backend):
327
        backend.return_value.get_image.return_value = self.image
328
        response = self.myget('images/42/metadata/foo', 'user')
329
        self.assertSuccess(response)
330
        meta = json.loads(response.content)['meta']
331
        self.assertEqual(meta['foo'], 'bar')
332

    
333
    @assert_backend_closed
334
    def test_get_invalid_metadata(self, backend):
335
        backend.return_value.get_image.return_value = self.image
336
        response = self.myget('images/42/metadata/not_found', 'user')
337
        self.assertItemNotFound(response)
338

    
339
    def test_delete_metadata_item(self, backend):
340
        backend.return_value.get_image.return_value = self.image
341
        response = self.mydelete('images/42/metadata/foo', 'user')
342
        self.assertEqual(response.status_code, 204)
343
        backend.return_value.update_metadata\
344
               .assert_called_once_with('42', {'properties': {'foo2': 'bar2'}})
345

    
346
    @assert_backend_closed
347
    def test_create_metadata_item(self, backend):
348
        backend.return_value.get_image.return_value = self.image
349
        request = {'meta': {'foo3': 'bar3'}}
350
        response = self.myput('images/42/metadata/foo3', 'user',
351
                              json.dumps(request), 'json')
352
        self.assertEqual(response.status_code, 201)
353
        backend.return_value.update_metadata.assert_called_once_with('42',
354
                {'properties':
355
                    {'foo': 'bar', 'foo2': 'bar2', 'foo3': 'bar3'}})
356

    
357
    @assert_backend_closed
358
    def test_create_metadata_malformed_1(self, backend):
359
        backend.return_value.get_image.return_value = self.image
360
        request = {'met': {'foo3': 'bar3'}}
361
        response = self.myput('images/42/metadata/foo3', 'user',
362
                              json.dumps(request), 'json')
363
        self.assertBadRequest(response)
364

    
365
    @assert_backend_closed
366
    def test_create_metadata_malformed_2(self, backend):
367
        backend.return_value.get_image.return_value = self.image
368
        request = {'metadata': [('foo3', 'bar3')]}
369
        response = self.myput('images/42/metadata/foo3', 'user',
370
                              json.dumps(request), 'json')
371
        self.assertBadRequest(response)
372

    
373
    @assert_backend_closed
374
    def test_create_metadata_malformed_3(self, backend):
375
        backend.return_value.get_image.return_value = self.image
376
        request = {'met': {'foo3': 'bar3', 'foo4': 'bar4'}}
377
        response = self.myput('images/42/metadata/foo3', 'user',
378
                              json.dumps(request), 'json')
379
        self.assertBadRequest(response)
380

    
381
    @assert_backend_closed
382
    def test_create_metadata_malformed_4(self, backend):
383
        backend.return_value.get_image.return_value = self.image
384
        request = {'met': {'foo3': 'bar3'}}
385
        response = self.myput('images/42/metadata/foo4', 'user',
386
                              json.dumps(request), 'json')
387
        self.assertBadRequest(response)
388

    
389
    @assert_backend_closed
390
    def test_update_metadata_item(self, backend):
391
        backend.return_value.get_image.return_value = self.image
392
        request = {'metadata': {'foo': 'bar_new', 'foo4': 'bar4'}}
393
        response = self.mypost('images/42/metadata', 'user',
394
                               json.dumps(request), 'json')
395
        self.assertEqual(response.status_code, 201)
396
        backend.return_value.update_metadata.assert_called_once_with('42',
397
                {'properties':
398
                    {'foo': 'bar_new', 'foo2': 'bar2', 'foo4': 'bar4'}
399
                })
400

    
401
    @assert_backend_closed
402
    def test_update_metadata_malformed(self, backend):
403
        backend.return_value.get_image.return_value = self.image
404
        request = {'meta': {'foo': 'bar_new', 'foo4': 'bar4'}}
405
        response = self.mypost('images/42/metadata', 'user',
406
                               json.dumps(request), 'json')
407
        self.assertBadRequest(response)