@@ -199,23 +199,50 @@ def setUp(self):
199199 def test_type_list_without_options (self ):
200200 arglist = []
201201 verifylist = [
202- ("long" , False )
202+ ("long" , False ),
203+ ("private" , False ),
204+ ("public" , False ),
203205 ]
204206 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
205207
206208 columns , data = self .cmd .take_action (parsed_args )
209+ self .types_mock .list .assert_called_once_with (is_public = None )
207210 self .assertEqual (self .columns , columns )
208211 self .assertEqual (self .data , list (data ))
209212
210213 def test_type_list_with_options (self ):
211- arglist = ["--long" ]
212- verifylist = [("long" , True )]
214+ arglist = [
215+ "--long" ,
216+ "--public" ,
217+ ]
218+ verifylist = [
219+ ("long" , True ),
220+ ("private" , False ),
221+ ("public" , True ),
222+ ]
213223 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
214224
215225 columns , data = self .cmd .take_action (parsed_args )
226+ self .types_mock .list .assert_called_once_with (is_public = True )
216227 self .assertEqual (self .columns_long , columns )
217228 self .assertEqual (self .data_long , list (data ))
218229
230+ def test_type_list_with_private_option (self ):
231+ arglist = [
232+ "--private" ,
233+ ]
234+ verifylist = [
235+ ("long" , False ),
236+ ("private" , True ),
237+ ("public" , False ),
238+ ]
239+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
240+
241+ columns , data = self .cmd .take_action (parsed_args )
242+ self .types_mock .list .assert_called_once_with (is_public = False )
243+ self .assertEqual (self .columns , columns )
244+ self .assertEqual (self .data , list (data ))
245+
219246
220247class TestTypeSet (TestType ):
221248
0 commit comments