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 )
@@ -369,13 +364,13 @@ def test_server_create_with_block_device_mapping(self):
369364 arglist = [
370365 '--image' , 'image1' ,
371366 '--flavor' , self .flavor .id ,
372- '--block-device-mapping' , compute_fakes .block_device_mapping ,
367+ '--block-device-mapping' , self .block_device_mapping ,
373368 self .new_server .name ,
374369 ]
375370 verifylist = [
376371 ('image' , 'image1' ),
377372 ('flavor' , self .flavor .id ),
378- ('block_device_mapping' , [compute_fakes .block_device_mapping ]),
373+ ('block_device_mapping' , [self .block_device_mapping ]),
379374 ('config_drive' , False ),
380375 ('server_name' , self .new_server .name ),
381376 ]
@@ -385,9 +380,9 @@ def test_server_create_with_block_device_mapping(self):
385380 columns , data = self .cmd .take_action (parsed_args )
386381
387382 real_volume_mapping = (
388- (compute_fakes .block_device_mapping .split ('=' , 1 )[1 ]).replace (
389- volume_fakes . volume_name ,
390- volume_fakes . volume_id ))
383+ (self .block_device_mapping .split ('=' , 1 )[1 ]).replace (
384+ self . volume . name ,
385+ self . volume . id ))
391386
392387 # Set expected values
393388 kwargs = dict (
0 commit comments