@@ -38,41 +38,28 @@ def _build_compute_az_datalist(compute_az, long_datalist=False):
3838 return (datalist ,)
3939
4040
41- def _build_volume_az_datalist ( volume_az , long_datalist = False ):
41+ def _build_network_az_datalist ( network_az , long_datalist = False ):
4242 datalist = ()
4343 if not long_datalist :
44- datalist = (
45- volume_az .name ,
46- 'available' ,
47- )
44+ datalist = (network_az .name , network_az .state )
4845 else :
4946 datalist = (
50- volume_az .name ,
51- 'available' ,
52- '' ,
47+ network_az .name ,
48+ network_az . state ,
49+ network_az . resource ,
5350 '' ,
5451 '' ,
5552 '' ,
5653 )
5754 return (datalist ,)
5855
5956
60- def _build_network_az_datalist ( network_az , long_datalist = False ):
57+ def _build_volume_az_datalist ( volume_az , long_datalist = False ):
6158 datalist = ()
6259 if not long_datalist :
63- datalist = (
64- network_az .name ,
65- network_az .state ,
66- )
60+ datalist = (volume_az .name , 'available' )
6761 else :
68- datalist = (
69- network_az .name ,
70- network_az .state ,
71- network_az .resource ,
72- '' ,
73- '' ,
74- '' ,
75- )
62+ datalist = (volume_az .name , 'available' , '' , '' , '' , '' )
7663 return (datalist ,)
7764
7865
@@ -82,10 +69,6 @@ class TestAvailabilityZoneList(
8269 compute_fakes .FakeClientMixin ,
8370 utils .TestCommand ,
8471):
85- compute_azs = compute_fakes .create_availability_zones ()
86- volume_azs = volume_fakes .create_availability_zones (count = 1 )
87- network_azs = network_fakes .create_availability_zones ()
88-
8972 short_columnslist = ('Zone Name' , 'Zone Status' )
9073 long_columnslist = (
9174 'Zone Name' ,
@@ -99,11 +82,14 @@ class TestAvailabilityZoneList(
9982 def setUp (self ):
10083 super ().setUp ()
10184
85+ self .compute_azs = compute_fakes .create_availability_zones ()
10286 self .compute_client .availability_zones .return_value = self .compute_azs
87+ self .network_azs = network_fakes .create_availability_zones ()
88+ self .network_client .availability_zones .return_value = self .network_azs
89+ self .volume_azs = volume_fakes .create_availability_zones (count = 1 )
10390 self .volume_sdk_client .availability_zones .return_value = (
10491 self .volume_azs
10592 )
106- self .network_client .availability_zones .return_value = self .network_azs
10793
10894 # Get the command object to test
10995 self .cmd = availability_zone .ListAvailabilityZone (self .app , None )
@@ -119,17 +105,17 @@ def test_availability_zone_list_no_options(self):
119105 columns , data = self .cmd .take_action (parsed_args )
120106
121107 self .compute_client .availability_zones .assert_called_with (details = True )
122- self .volume_sdk_client .availability_zones .assert_called_with ()
123108 self .network_client .availability_zones .assert_called_with ()
109+ self .volume_sdk_client .availability_zones .assert_called_with ()
124110
125111 self .assertEqual (self .short_columnslist , columns )
126112 datalist = ()
127113 for compute_az in self .compute_azs :
128114 datalist += _build_compute_az_datalist (compute_az )
129- for volume_az in self .volume_azs :
130- datalist += _build_volume_az_datalist (volume_az )
131115 for network_az in self .network_azs :
132116 datalist += _build_network_az_datalist (network_az )
117+ for volume_az in self .volume_azs :
118+ datalist += _build_volume_az_datalist (volume_az )
133119 self .assertEqual (datalist , tuple (data ))
134120
135121 def test_availability_zone_list_long (self ):
@@ -147,23 +133,23 @@ def test_availability_zone_list_long(self):
147133 columns , data = self .cmd .take_action (parsed_args )
148134
149135 self .compute_client .availability_zones .assert_called_with (details = True )
150- self .volume_sdk_client .availability_zones .assert_called_with ()
151136 self .network_client .availability_zones .assert_called_with ()
137+ self .volume_sdk_client .availability_zones .assert_called_with ()
152138
153139 self .assertEqual (self .long_columnslist , columns )
154140 datalist = ()
155141 for compute_az in self .compute_azs :
156142 datalist += _build_compute_az_datalist (
157143 compute_az , long_datalist = True
158144 )
159- for volume_az in self .volume_azs :
160- datalist += _build_volume_az_datalist (
161- volume_az , long_datalist = True
162- )
163145 for network_az in self .network_azs :
164146 datalist += _build_network_az_datalist (
165147 network_az , long_datalist = True
166148 )
149+ for volume_az in self .volume_azs :
150+ datalist += _build_volume_az_datalist (
151+ volume_az , long_datalist = True
152+ )
167153 self .assertEqual (datalist , tuple (data ))
168154
169155 def test_availability_zone_list_compute (self ):
@@ -181,21 +167,21 @@ def test_availability_zone_list_compute(self):
181167 columns , data = self .cmd .take_action (parsed_args )
182168
183169 self .compute_client .availability_zones .assert_called_with (details = True )
184- self .volume_sdk_client .availability_zones .assert_not_called ()
185170 self .network_client .availability_zones .assert_not_called ()
171+ self .volume_sdk_client .availability_zones .assert_not_called ()
186172
187173 self .assertEqual (self .short_columnslist , columns )
188174 datalist = ()
189175 for compute_az in self .compute_azs :
190176 datalist += _build_compute_az_datalist (compute_az )
191177 self .assertEqual (datalist , tuple (data ))
192178
193- def test_availability_zone_list_volume (self ):
179+ def test_availability_zone_list_network (self ):
194180 arglist = [
195- '--volume ' ,
181+ '--network ' ,
196182 ]
197183 verifylist = [
198- ('volume ' , True ),
184+ ('network ' , True ),
199185 ]
200186 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
201187
@@ -205,21 +191,21 @@ def test_availability_zone_list_volume(self):
205191 columns , data = self .cmd .take_action (parsed_args )
206192
207193 self .compute_client .availability_zones .assert_not_called ()
208- self .volume_sdk_client .availability_zones .assert_called_with ()
209- self .network_client .availability_zones .assert_not_called ()
194+ self .network_client .availability_zones .assert_called_with ()
195+ self .volume_sdk_client .availability_zones .assert_not_called ()
210196
211197 self .assertEqual (self .short_columnslist , columns )
212198 datalist = ()
213- for volume_az in self .volume_azs :
214- datalist += _build_volume_az_datalist ( volume_az )
199+ for network_az in self .network_azs :
200+ datalist += _build_network_az_datalist ( network_az )
215201 self .assertEqual (datalist , tuple (data ))
216202
217- def test_availability_zone_list_network (self ):
203+ def test_availability_zone_list_volume (self ):
218204 arglist = [
219- '--network ' ,
205+ '--volume ' ,
220206 ]
221207 verifylist = [
222- ('network ' , True ),
208+ ('volume ' , True ),
223209 ]
224210 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
225211
@@ -229,11 +215,11 @@ def test_availability_zone_list_network(self):
229215 columns , data = self .cmd .take_action (parsed_args )
230216
231217 self .compute_client .availability_zones .assert_not_called ()
232- self .volume_sdk_client .availability_zones .assert_not_called ()
233- self .network_client .availability_zones .assert_called_with ()
218+ self .network_client .availability_zones .assert_not_called ()
219+ self .volume_sdk_client .availability_zones .assert_called_with ()
234220
235221 self .assertEqual (self .short_columnslist , columns )
236222 datalist = ()
237- for network_az in self .network_azs :
238- datalist += _build_network_az_datalist ( network_az )
223+ for volume_az in self .volume_azs :
224+ datalist += _build_volume_az_datalist ( volume_az )
239225 self .assertEqual (datalist , tuple (data ))
0 commit comments