@@ -77,6 +77,12 @@ class TestCreateNetworkIdentityV3(TestNetwork):
7777 def setUp (self ):
7878 super (TestCreateNetworkIdentityV3 , self ).setUp ()
7979
80+ self .new_network = mock .Mock (return_value = copy .deepcopy (RESPONSE ))
81+ self .network .create_network = self .new_network
82+
83+ # Get the command object to test
84+ self .cmd = network .CreateNetwork (self .app , self .namespace )
85+
8086 def test_create_no_options (self ):
8187 arglist = [
8288 FAKE_NAME ,
@@ -87,14 +93,11 @@ def test_create_no_options(self):
8793 ('shared' , None ),
8894 ('project' , None ),
8995 ]
90- mocker = mock .Mock (return_value = copy .deepcopy (RESPONSE ))
91- self .app .client_manager .network .create_network = mocker
92- cmd = network .CreateNetwork (self .app , self .namespace )
9396
94- parsed_args = self .check_parser (cmd , arglist , verifylist )
95- result = list (cmd .take_action (parsed_args ))
97+ parsed_args = self .check_parser (self . cmd , arglist , verifylist )
98+ result = list (self . cmd .take_action (parsed_args ))
9699
97- mocker .assert_called_with ({
100+ self . network . create_network .assert_called_with ({
98101 RESOURCE : {
99102 'admin_state_up' : True ,
100103 'name' : FAKE_NAME ,
@@ -117,8 +120,6 @@ def test_create_all_options(self):
117120 ('project_domain' , identity_fakes_v3 .domain_name ),
118121 ('name' , FAKE_NAME ),
119122 ]
120- mocker = mock .Mock (return_value = copy .deepcopy (RESPONSE ))
121- self .app .client_manager .network .create_network = mocker
122123 identity_client = identity_fakes_v3 .FakeIdentityv3Client (
123124 endpoint = fakes .AUTH_URL ,
124125 token = fakes .AUTH_TOKEN ,
@@ -136,12 +137,11 @@ def test_create_all_options(self):
136137 copy .deepcopy (identity_fakes_v3 .DOMAIN ),
137138 loaded = True ,
138139 )
139- cmd = network .CreateNetwork (self .app , self .namespace )
140140
141- parsed_args = self .check_parser (cmd , arglist , verifylist )
142- result = list (cmd .take_action (parsed_args ))
141+ parsed_args = self .check_parser (self . cmd , arglist , verifylist )
142+ result = list (self . cmd .take_action (parsed_args ))
143143
144- mocker .assert_called_with ({
144+ self . network . create_network .assert_called_with ({
145145 RESOURCE : {
146146 'admin_state_up' : False ,
147147 'name' : FAKE_NAME ,
@@ -162,14 +162,11 @@ def test_create_other_options(self):
162162 ('shared' , False ),
163163 ('name' , FAKE_NAME ),
164164 ]
165- mocker = mock .Mock (return_value = copy .deepcopy (RESPONSE ))
166- self .app .client_manager .network .create_network = mocker
167- cmd = network .CreateNetwork (self .app , self .namespace )
168165
169- parsed_args = self .check_parser (cmd , arglist , verifylist )
170- result = list (cmd .take_action (parsed_args ))
166+ parsed_args = self .check_parser (self . cmd , arglist , verifylist )
167+ result = list (self . cmd .take_action (parsed_args ))
171168
172- mocker .assert_called_with ({
169+ self . network . create_network .assert_called_with ({
173170 RESOURCE : {
174171 'admin_state_up' : True ,
175172 'name' : FAKE_NAME ,
0 commit comments