Revision 1df5305d kamaki/clients/compute/test.py

b/kamaki/clients/compute/test.py
285 285
    def test_delete_server(self, SD):
286 286
        vm_id = vm_recv['server']['id']
287 287
        self.client.delete_server(vm_id)
288
        self.assertEqual(SD.mock_calls[-1], call(vm_id))
288
        SD.assert_called_once_with(vm_id)
289 289

  
290 290
    @patch('%s.images_delete' % compute_pkg, return_value=FR())
291 291
    def test_delete_image(self, ID):
292 292
        self.client.delete_image(img_ref)
293
        self.assertEqual(ID.mock_calls[-1], call(img_ref))
293
        ID.assert_called_once_with(img_ref)
294 294

  
295 295
    @patch('%s.images_put' % compute_pkg, return_value=FR())
296 296
    def test_create_image_metadata(self, IP):
297 297
        (key, val) = ('k1', 'v1')
298 298
        FR.json = dict(meta=img_recv['image'])
299 299
        r = self.client.create_image_metadata(img_ref, key, val)
300
        self.assertEqual(IP.mock_calls[-1], call(
300
        IP.assert_called_once_with(
301 301
            img_ref, 'meta/%s' % key,
302
            json_data=dict(meta={key: val})))
302
            json_data=dict(meta={key: val}))
303 303
        self.assert_dicts_are_equal(r, img_recv['image'])
304 304

  
305 305
    @patch('%s.images_post' % compute_pkg, return_value=FR())
......
307 307
        metadata = dict(m1='v1', m2='v2', m3='v3')
308 308
        FR.json = dict(metadata=metadata)
309 309
        r = self.client.update_image_metadata(img_ref, **metadata)
310
        self.assertEqual(
311
            IP.mock_calls[-1],
312
            call(img_ref, 'meta', json_data=dict(metadata=metadata)))
310
        IP.assert_called_once_with(
311
            img_ref, 'meta',
312
            json_data=dict(metadata=metadata))
313 313
        self.assert_dicts_are_equal(r, metadata)
314 314

  
315 315
    @patch('%s.images_delete' % compute_pkg, return_value=FR())
316 316
    def test_delete_image_metadata(self, ID):
317 317
        key = 'metakey'
318 318
        self.client.delete_image_metadata(img_ref, key)
319
        self.assertEqual(ID.mock_calls[-1], call(img_ref, '/meta/%s' % key))
319
        ID.assert_called_once_with(img_ref, '/meta/%s' % key)
320 320

  
321 321
if __name__ == '__main__':
322 322
    from sys import argv

Also available in: Unified diff