def get_args(choices): parser = base_parser( api_client.build_os_option_parser( argparse.ArgumentParser(description='Freezer Scheduler') )) parser.add_argument( 'action', action='store', default=None, choices=choices, help='') return parser.parse_args()
def test_build_os_option_parser(self, mock_env): mock_env.return_value = '' mock_parser = Mock() mock_parser._me = 'test12345' retval = client.build_os_option_parser(mock_parser) self.assertEquals(retval._me, 'test12345') call_count = mock_parser.add_argument.call_count self.assertGreater(call_count, 10)