Skip to content

Commit 364e934

Browse files
committed
trivial: Sort AZ helpers alphabetically
Change-Id: Ia99abffaa6f34e3ec588695cc88328d8a8acb365 Signed-off-by: Stephen Finucane <stephenfin@redhat.com>
1 parent 34d06db commit 364e934

2 files changed

Lines changed: 71 additions & 85 deletions

File tree

openstackclient/common/availability_zone.py

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -62,18 +62,6 @@ def _xform_compute_availability_zone(az, include_extra):
6262
return result
6363

6464

65-
def _xform_volume_availability_zone(az):
66-
result = []
67-
zone_info = {
68-
'zone_name': az.name,
69-
'zone_status': (
70-
'available' if az.state['available'] else 'not available'
71-
),
72-
}
73-
result.append(zone_info)
74-
return result
75-
76-
7765
def _xform_network_availability_zone(az):
7866
result = []
7967
zone_info = {}
@@ -86,6 +74,18 @@ def _xform_network_availability_zone(az):
8674
return result
8775

8876

77+
def _xform_volume_availability_zone(az):
78+
result = []
79+
zone_info = {
80+
'zone_name': az.name,
81+
'zone_status': (
82+
'available' if az.state['available'] else 'not available'
83+
),
84+
}
85+
result.append(zone_info)
86+
return result
87+
88+
8989
class ListAvailabilityZone(command.Lister):
9090
_description = _("List availability zones and their status")
9191

@@ -132,27 +132,6 @@ def _get_compute_availability_zones(self, parsed_args):
132132
result += _xform_compute_availability_zone(zone, parsed_args.long)
133133
return result
134134

135-
def _get_volume_availability_zones(self, parsed_args):
136-
volume_client = sdk_utils.ensure_service_version(
137-
self.app.client_manager.sdk_connection.volume, '3'
138-
)
139-
data = []
140-
try:
141-
data = list(volume_client.availability_zones())
142-
except Exception as e:
143-
LOG.debug('Volume availability zone exception: %s', e)
144-
if parsed_args.volume:
145-
message = _(
146-
"Availability zones list not supported by "
147-
"Block Storage API"
148-
)
149-
LOG.warning(message)
150-
151-
result = []
152-
for zone in data:
153-
result += _xform_volume_availability_zone(zone)
154-
return result
155-
156135
def _get_network_availability_zones(self, parsed_args):
157136
network_client = self.app.client_manager.network
158137
try:
@@ -174,6 +153,27 @@ def _get_network_availability_zones(self, parsed_args):
174153
result += _xform_network_availability_zone(zone)
175154
return result
176155

156+
def _get_volume_availability_zones(self, parsed_args):
157+
volume_client = sdk_utils.ensure_service_version(
158+
self.app.client_manager.sdk_connection.volume, '3'
159+
)
160+
data = []
161+
try:
162+
data = list(volume_client.availability_zones())
163+
except Exception as e:
164+
LOG.debug('Volume availability zone exception: %s', e)
165+
if parsed_args.volume:
166+
message = _(
167+
"Availability zones list not supported by "
168+
"Block Storage API"
169+
)
170+
LOG.warning(message)
171+
172+
result = []
173+
for zone in data:
174+
result += _xform_volume_availability_zone(zone)
175+
return result
176+
177177
def take_action(self, parsed_args):
178178
columns: tuple[str, ...] = ('Zone Name', 'Zone Status')
179179
if parsed_args.long:
@@ -187,17 +187,17 @@ def take_action(self, parsed_args):
187187
# Show everything by default.
188188
show_all = (
189189
not parsed_args.compute
190-
and not parsed_args.volume
191190
and not parsed_args.network
191+
and not parsed_args.volume
192192
)
193193

194194
result = []
195195
if parsed_args.compute or show_all:
196196
result += self._get_compute_availability_zones(parsed_args)
197-
if parsed_args.volume or show_all:
198-
result += self._get_volume_availability_zones(parsed_args)
199197
if parsed_args.network or show_all:
200198
result += self._get_network_availability_zones(parsed_args)
199+
if parsed_args.volume or show_all:
200+
result += self._get_volume_availability_zones(parsed_args)
201201

202202
return (
203203
columns,

openstackclient/tests/unit/common/test_availability_zone.py

Lines changed: 35 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)