@@ -227,58 +227,61 @@ def test_delete_multiple_agggregates_with_exception(self):
227227
228228
229229class TestAggregateList (TestAggregate ):
230- list_columns = (
231- "ID" ,
232- "Name" ,
233- "Availability Zone" ,
234- )
235-
236- list_columns_long = (
237- "ID" ,
238- "Name" ,
239- "Availability Zone" ,
240- "Properties" ,
241- "Hosts" ,
242- )
243-
244- list_data = (
245- (
246- TestAggregate .fake_ag .id ,
247- TestAggregate .fake_ag .name ,
248- TestAggregate .fake_ag .availability_zone ,
249- ),
250- )
251-
252- list_data_long = (
253- (
254- TestAggregate .fake_ag .id ,
255- TestAggregate .fake_ag .name ,
256- TestAggregate .fake_ag .availability_zone ,
257- format_columns .DictColumn (
258- {
259- key : value
260- for key , value in TestAggregate .fake_ag .metadata .items ()
261- if key != 'availability_zone'
262- }
263- ),
264- format_columns .ListColumn (TestAggregate .fake_ag .hosts ),
265- ),
266- )
267-
268230 def setUp (self ):
269231 super ().setUp ()
270232
271233 self .compute_sdk_client .aggregates .return_value = [self .fake_ag ]
272234 self .cmd = aggregate .ListAggregate (self .app , None )
273235
274236 def test_aggregate_list (self ):
237+ self .set_compute_api_version ('2.41' )
238+
239+ parsed_args = self .check_parser (self .cmd , [], [])
240+ columns , data = self .cmd .take_action (parsed_args )
241+
242+ expected_columns = (
243+ "ID" ,
244+ "UUID" ,
245+ "Name" ,
246+ "Availability Zone" ,
247+ )
248+ expected_data = (
249+ (
250+ self .fake_ag .id ,
251+ self .fake_ag .uuid ,
252+ self .fake_ag .name ,
253+ self .fake_ag .availability_zone ,
254+ ),
255+ )
256+
257+ self .assertEqual (expected_columns , columns )
258+ self .assertCountEqual (expected_data , tuple (data ))
259+
260+ def test_aggregate_list_pre_v241 (self ):
261+ self .set_compute_api_version ('2.40' )
262+
275263 parsed_args = self .check_parser (self .cmd , [], [])
276264 columns , data = self .cmd .take_action (parsed_args )
277265
278- self .assertEqual (self .list_columns , columns )
279- self .assertCountEqual (self .list_data , tuple (data ))
266+ expected_columns = (
267+ "ID" ,
268+ "Name" ,
269+ "Availability Zone" ,
270+ )
271+ expected_data = (
272+ (
273+ self .fake_ag .id ,
274+ self .fake_ag .name ,
275+ self .fake_ag .availability_zone ,
276+ ),
277+ )
278+
279+ self .assertEqual (expected_columns , columns )
280+ self .assertCountEqual (expected_data , tuple (data ))
280281
281282 def test_aggregate_list_with_long (self ):
283+ self .set_compute_api_version ('2.41' )
284+
282285 arglist = [
283286 '--long' ,
284287 ]
@@ -288,8 +291,33 @@ def test_aggregate_list_with_long(self):
288291 parsed_args = self .check_parser (self .cmd , arglist , vertifylist )
289292 columns , data = self .cmd .take_action (parsed_args )
290293
291- self .assertEqual (self .list_columns_long , columns )
292- self .assertCountEqual (self .list_data_long , tuple (data ))
294+ expected_columns = (
295+ "ID" ,
296+ "UUID" ,
297+ "Name" ,
298+ "Availability Zone" ,
299+ "Properties" ,
300+ "Hosts" ,
301+ )
302+ expected_data = (
303+ (
304+ self .fake_ag .id ,
305+ self .fake_ag .uuid ,
306+ self .fake_ag .name ,
307+ self .fake_ag .availability_zone ,
308+ format_columns .DictColumn (
309+ {
310+ key : value
311+ for key , value in self .fake_ag .metadata .items ()
312+ if key != 'availability_zone'
313+ }
314+ ),
315+ format_columns .ListColumn (self .fake_ag .hosts ),
316+ ),
317+ )
318+
319+ self .assertEqual (expected_columns , columns )
320+ self .assertCountEqual (expected_data , tuple (data ))
293321
294322
295323class TestAggregateRemoveHost (TestAggregate ):
0 commit comments