@@ -48,6 +48,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy):
4848 columns = (
4949 'description' ,
5050 'id' ,
51+ 'is_default' ,
5152 'name' ,
5253 'project_id' ,
5354 'rules' ,
@@ -57,6 +58,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy):
5758 data = (
5859 new_qos_policy .description ,
5960 new_qos_policy .id ,
61+ new_qos_policy .is_default ,
6062 new_qos_policy .name ,
6163 new_qos_policy .project_id ,
6264 new_qos_policy .rules ,
@@ -106,12 +108,14 @@ def test_create_all_options(self):
106108 '--project' , self .project .name ,
107109 self .new_qos_policy .name ,
108110 '--description' , 'QoS policy description' ,
111+ '--default' ,
109112 ]
110113 verifylist = [
111114 ('share' , True ),
112115 ('project' , self .project .name ),
113116 ('name' , self .new_qos_policy .name ),
114117 ('description' , 'QoS policy description' ),
118+ ('default' , True ),
115119 ]
116120
117121 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
@@ -122,6 +126,28 @@ def test_create_all_options(self):
122126 'tenant_id' : self .project .id ,
123127 'name' : self .new_qos_policy .name ,
124128 'description' : 'QoS policy description' ,
129+ 'is_default' : True ,
130+ })
131+ self .assertEqual (self .columns , columns )
132+ self .assertEqual (self .data , data )
133+
134+ def test_create_no_default (self ):
135+ arglist = [
136+ self .new_qos_policy .name ,
137+ '--no-default'
138+ ]
139+ verifylist = [
140+ ('project' , None ),
141+ ('name' , self .new_qos_policy .name ),
142+ ('default' , False ),
143+ ]
144+
145+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
146+ columns , data = (self .cmd .take_action (parsed_args ))
147+
148+ self .network .create_qos_policy .assert_called_once_with (** {
149+ 'name' : self .new_qos_policy .name ,
150+ 'is_default' : False ,
125151 })
126152 self .assertEqual (self .columns , columns )
127153 self .assertEqual (self .data , data )
@@ -220,6 +246,7 @@ class TestListNetworkQosPolicy(TestQosPolicy):
220246 'ID' ,
221247 'Name' ,
222248 'Shared' ,
249+ 'Default' ,
223250 'Project' ,
224251 )
225252 data = []
@@ -228,6 +255,7 @@ class TestListNetworkQosPolicy(TestQosPolicy):
228255 qos_policy .id ,
229256 qos_policy .name ,
230257 qos_policy .shared ,
258+ qos_policy .is_default ,
231259 qos_policy .project_id ,
232260 ))
233261
@@ -333,17 +361,19 @@ def test_set_nothing(self):
333361 self ._qos_policy , ** attrs )
334362 self .assertIsNone (result )
335363
336- def test_set_name_share_description (self ):
364+ def test_set_name_share_description_default (self ):
337365 arglist = [
338366 '--name' , 'new_qos_policy' ,
339367 '--share' ,
340368 '--description' , 'QoS policy description' ,
369+ '--default' ,
341370 self ._qos_policy .name ,
342371 ]
343372 verifylist = [
344373 ('name' , 'new_qos_policy' ),
345374 ('share' , True ),
346375 ('description' , 'QoS policy description' ),
376+ ('default' , True ),
347377 ('policy' , self ._qos_policy .name ),
348378 ]
349379
@@ -353,25 +383,29 @@ def test_set_name_share_description(self):
353383 'name' : 'new_qos_policy' ,
354384 'description' : 'QoS policy description' ,
355385 'shared' : True ,
386+ 'is_default' : True ,
356387 }
357388 self .network .update_qos_policy .assert_called_with (
358389 self ._qos_policy , ** attrs )
359390 self .assertIsNone (result )
360391
361- def test_set_no_share (self ):
392+ def test_set_no_share_no_default (self ):
362393 arglist = [
363394 '--no-share' ,
395+ '--no-default' ,
364396 self ._qos_policy .name ,
365397 ]
366398 verifylist = [
367399 ('no_share' , True ),
400+ ('no_default' , True ),
368401 ('policy' , self ._qos_policy .name ),
369402 ]
370403
371404 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
372405 result = self .cmd .take_action (parsed_args )
373406 attrs = {
374- 'shared' : False
407+ 'shared' : False ,
408+ 'is_default' : False
375409 }
376410 self .network .update_qos_policy .assert_called_with (
377411 self ._qos_policy , ** attrs )
@@ -386,6 +420,7 @@ class TestShowNetworkQosPolicy(TestQosPolicy):
386420 columns = (
387421 'description' ,
388422 'id' ,
423+ 'is_default' ,
389424 'name' ,
390425 'project_id' ,
391426 'rules' ,
@@ -394,6 +429,7 @@ class TestShowNetworkQosPolicy(TestQosPolicy):
394429 data = (
395430 _qos_policy .description ,
396431 _qos_policy .id ,
432+ _qos_policy .is_default ,
397433 _qos_policy .name ,
398434 _qos_policy .project_id ,
399435 _qos_policy .rules ,
0 commit comments