1414from unittest .mock import call
1515
1616from openstack .block_storage .v3 import type as _type
17+ from openstack .identity .v3 import project as _project
1718from openstack .test import fakes as sdk_fakes
1819from osc_lib .cli import format_columns
1920from osc_lib import exceptions
2021from osc_lib import utils
2122
22- from openstackclient .tests .unit .identity .v3 import fakes as identity_fakes
2323from openstackclient .tests .unit import utils as tests_utils
2424from openstackclient .tests .unit .volume .v2 import fakes as volume_fakes
2525from openstackclient .volume .v2 import volume_type
@@ -30,7 +30,7 @@ def setUp(self):
3030 super ().setUp ()
3131
3232 self .new_volume_type = sdk_fakes .generate_fake_resource (_type .Type )
33- self .project = identity_fakes . FakeProject . create_one_project ( )
33+ self .project = sdk_fakes . generate_fake_resource ( _project . Project )
3434 self .columns = (
3535 'description' ,
3636 'id' ,
@@ -47,9 +47,7 @@ def setUp(self):
4747 )
4848
4949 self .volume_sdk_client .create_type .return_value = self .new_volume_type
50-
51- self .projects_mock = self .identity_client .projects
52- self .projects_mock .get .return_value = self .project
50+ self .identity_sdk_client .find_project .return_value = self .project
5351
5452 self .cmd = volume_type .CreateVolumeType (self .app , None )
5553
@@ -490,9 +488,8 @@ class TestTypeSet(volume_fakes.TestVolume):
490488 def setUp (self ):
491489 super ().setUp ()
492490
493- self .project = identity_fakes .FakeProject .create_one_project ()
494- self .projects_mock = self .identity_client .projects
495- self .projects_mock .get .return_value = self .project
491+ self .project = sdk_fakes .generate_fake_resource (_project .Project )
492+ self .identity_sdk_client .find_project .return_value = self .project
496493
497494 self .volume_type = sdk_fakes .generate_fake_resource (_type .Type )
498495 self .volume_sdk_client .find_type .return_value = self .volume_type
@@ -897,16 +894,15 @@ def test_type_show_with_encryption(self):
897894
898895
899896class TestTypeUnset (volume_fakes .TestVolume ):
900- project = identity_fakes .FakeProject .create_one_project ()
901897 volume_type = sdk_fakes .generate_fake_resource (_type .Type )
902898
903899 def setUp (self ):
904900 super ().setUp ()
905901
906902 self .volume_sdk_client .find_type .return_value = self .volume_type
907903
908- self .projects_mock = self . identity_client . projects
909- self .projects_mock . get .return_value = self .project
904+ self .project = sdk_fakes . generate_fake_resource ( _project . Project )
905+ self .identity_sdk_client . find_project .return_value = self .project
910906
911907 self .cmd = volume_type .UnsetVolumeType (self .app , None )
912908
@@ -973,10 +969,10 @@ def test_type_unset_not_called_without_project_argument(self):
973969 def test_type_unset_failed_with_missing_volume_type_argument (self ):
974970 arglist = [
975971 '--project' ,
976- 'identity_fakes.project_id ' ,
972+ 'foo ' ,
977973 ]
978974 verifylist = [
979- ('project' , 'identity_fakes.project_id ' ),
975+ ('project' , 'foo ' ),
980976 ]
981977
982978 self .assertRaises (
0 commit comments