1414#
1515
1616import argparse
17- import copy
1817import mock
1918from mock import call
2019
2120from osc_lib import exceptions
2221from osc_lib import utils
2322
24- from openstackclient .tests .unit import fakes
2523from openstackclient .tests .unit .identity .v2_0 import fakes as identity_fakes
24+ from openstackclient .tests .unit .image .v1 import fakes as image_fakes
2625from openstackclient .tests .unit import utils as tests_utils
2726from openstackclient .tests .unit .volume .v1 import fakes as volume_fakes
2827from openstackclient .volume .v1 import volume
@@ -321,19 +320,16 @@ def test_volume_create_properties(self):
321320 self .assertEqual (self .datalist , data )
322321
323322 def test_volume_create_image_id (self ):
324- self .images_mock .get .return_value = fakes .FakeResource (
325- None ,
326- copy .deepcopy (volume_fakes .IMAGE ),
327- loaded = True ,
328- )
323+ image = image_fakes .FakeImage .create_one_image ()
324+ self .images_mock .get .return_value = image
329325
330326 arglist = [
331- '--image' , volume_fakes . image_id ,
327+ '--image' , image . id ,
332328 '--size' , str (self .new_volume .size ),
333329 self .new_volume .display_name ,
334330 ]
335331 verifylist = [
336- ('image' , volume_fakes . image_id ),
332+ ('image' , image . id ),
337333 ('size' , self .new_volume .size ),
338334 ('name' , self .new_volume .display_name ),
339335 ]
@@ -360,26 +356,23 @@ def test_volume_create_image_id(self):
360356 None ,
361357 None ,
362358 None ,
363- volume_fakes . image_id ,
359+ image . id ,
364360 )
365361
366362 self .assertEqual (self .columns , columns )
367363 self .assertEqual (self .datalist , data )
368364
369365 def test_volume_create_image_name (self ):
370- self .images_mock .get .return_value = fakes .FakeResource (
371- None ,
372- copy .deepcopy (volume_fakes .IMAGE ),
373- loaded = True ,
374- )
366+ image = image_fakes .FakeImage .create_one_image ()
367+ self .images_mock .get .return_value = image
375368
376369 arglist = [
377- '--image' , volume_fakes . image_name ,
370+ '--image' , image . name ,
378371 '--size' , str (self .new_volume .size ),
379372 self .new_volume .display_name ,
380373 ]
381374 verifylist = [
382- ('image' , volume_fakes . image_name ),
375+ ('image' , image . name ),
383376 ('size' , self .new_volume .size ),
384377 ('name' , self .new_volume .display_name ),
385378 ]
@@ -406,7 +399,7 @@ def test_volume_create_image_name(self):
406399 None ,
407400 None ,
408401 None ,
409- volume_fakes . image_id ,
402+ image . id ,
410403 )
411404
412405 self .assertEqual (self .columns , columns )
0 commit comments