def test_argument_parser_for_custom_parameters(self): """ Check if the parser is correctly setting the user specified parameters """ os.environ['MY_SIGNALFX_API_TOKEN'] = 'abcdefghijk' # simulate user arguments as a list custom_argv = ['--config-file', 'my_configuration.txt', '--log-file', '/tmp/dummy_log_file', '--signalfx-rest-api', 'http://lab-api.corp.signalfuse.com:8080', '--pickle-file', 'my_pk_metadata.pk', '--sleep-duration', '10', '--env-variable-name', 'MY_SIGNALFX_API_TOKEN', '--log-handler', 'stdout' ] parser = collect_chef_metadata.get_argument_parser() args = vars(parser.parse_args(custom_argv)) self.assertNotEqual(args, None) self.assertEqual(args['CONFIG_FILE'], 'my_configuration.txt') self.assertEqual(args['LOG_FILE'], '/tmp/dummy_log_file') self.assertEqual(args['SIGNALFX_REST_API'], 'http://lab-api.corp.signalfuse.com:8080') self.assertEqual(args['PICKLE_FILE'], 'my_pk_metadata.pk') self.assertEqual(args['SLEEP_DURATION'], 10) self.assertEqual(args['ENV_VARIABLE_NAME'], 'MY_SIGNALFX_API_TOKEN') self.assertEqual(args['LOG_HANDLER'], 'stdout')
def test_argument_parser_for_default_parameters(self): """ Check if the parser is correctly setting the default parameters """ os.environ['SIGNALFX_API_TOKEN'] = 'abcdefghijk' argv = [] parser = collect_chef_metadata.get_argument_parser() args = vars(parser.parse_args(argv)) self.assertNotEqual(args, None) self.assertEqual(args['CONFIG_FILE'], collect_chef_metadata.DEFAULT_CONFIG_FILE) self.assertEqual(args['LOG_FILE'], collect_chef_metadata.DEFAULT_LOG_FILE) self.assertEqual(args['SIGNALFX_REST_API'], collect_chef_metadata.DEFAULT_SIGNALFX_REST_API) self.assertEqual(args['PICKLE_FILE'], collect_chef_metadata.DEFAULT_PICKLE_FILE) self.assertEqual(args['SLEEP_DURATION'], collect_chef_metadata.DEFAULT_SLEEP_DURATION) self.assertEqual(args['ENV_VARIABLE_NAME'], collect_chef_metadata.DEFAULT_ENV_VARIABLE_NAME) self.assertEqual(args['LOG_HANDLER'], collect_chef_metadata.DEFAULT_LOG_HANDLER)