1313# under the License.
1414#
1515
16- import copy
1716import mock
1817import testtools
1918
2221from openstackclient .common import utils as common_utils
2322from openstackclient .compute .v2 import server
2423from openstackclient .tests .compute .v2 import fakes as compute_fakes
25- from openstackclient .tests import fakes
2624from openstackclient .tests .image .v2 import fakes as image_fakes
2725from openstackclient .tests import utils
2826from openstackclient .tests .volume .v2 import fakes as volume_fakes
@@ -113,12 +111,9 @@ def setUp(self):
113111 self .flavor = compute_fakes .FakeFlavor .create_one_flavor ()
114112 self .flavors_mock .get .return_value = self .flavor
115113
116- self .volume = fakes .FakeResource (
117- None ,
118- copy .deepcopy (volume_fakes .VOLUME ),
119- loaded = True ,
120- )
114+ self .volume = volume_fakes .FakeVolume .create_one_volume ()
121115 self .volumes_mock .get .return_value = self .volume
116+ self .block_device_mapping = 'vda=' + self .volume .name + ':::0'
122117
123118 # Get the command object to test
124119 self .cmd = server .CreateServer (self .app , None )
@@ -357,13 +352,13 @@ def test_server_create_with_block_device_mapping(self):
357352 arglist = [
358353 '--image' , 'image1' ,
359354 '--flavor' , self .flavor .id ,
360- '--block-device-mapping' , compute_fakes .block_device_mapping ,
355+ '--block-device-mapping' , self .block_device_mapping ,
361356 self .new_server .name ,
362357 ]
363358 verifylist = [
364359 ('image' , 'image1' ),
365360 ('flavor' , self .flavor .id ),
366- ('block_device_mapping' , [compute_fakes .block_device_mapping ]),
361+ ('block_device_mapping' , [self .block_device_mapping ]),
367362 ('config_drive' , False ),
368363 ('server_name' , self .new_server .name ),
369364 ]
@@ -373,9 +368,9 @@ def test_server_create_with_block_device_mapping(self):
373368 columns , data = self .cmd .take_action (parsed_args )
374369
375370 real_volume_mapping = (
376- (compute_fakes .block_device_mapping .split ('=' , 1 )[1 ]).replace (
377- volume_fakes . volume_name ,
378- volume_fakes . volume_id ))
371+ (self .block_device_mapping .split ('=' , 1 )[1 ]).replace (
372+ self . volume . name ,
373+ self . volume . id ))
379374
380375 # Set expected values
381376 kwargs = dict (
0 commit comments