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'))
