def test_argument_process(self): def callback(settings, value): settings['TEST'] = value argument = Argument( Config( env='CONFIG', arg='--config', default=None ), callback=callback ) test_settings = {} argument.process([], {}, test_settings) self.assertEqual(test_settings, {'TEST': None}) test_settings = {} argument.process(['--config', 'value'], {}, test_settings) self.assertEqual(test_settings, {'TEST': 'value'}) test_settings = {} argument.process([], {'CONFIG': 'value'}, test_settings) self.assertEqual(test_settings, {'TEST': 'value'})
def test_argument(self): callback = Callback() argument = Argument( Config( env='CONFIG', arg='--config', default=None ), callback=callback ) argument.process([], {}, {}) self.assertEqual(callback.num_calls, 1) self.assertEqual(callback.get_call(0).args, ({}, None)) callback.reset() argument.process(['--config', 'value'], {}, {}) self.assertEqual(callback.num_calls, 1) self.assertEqual(callback.get_call(0).args, ({}, 'value')) callback.reset() argument.process([], {'CONFIG': 'value'}, {}) self.assertEqual(callback.num_calls, 1) self.assertEqual(callback.get_call(0).args, ({}, 'value'))