class TestOverrideRequiredArgsArgument(unittest.TestCase): def setUp(self): self.session = mock.Mock() self.argument = OverrideRequiredArgsArgument(self.session) # Set up a sample argument_table self.argument_table = {} self.mock_arg = mock.Mock() self.mock_arg.required = True self.argument_table['mock-arg'] = self.mock_arg def test_register_argument_action(self): register_args = self.session.register.call_args self.assertEqual(register_args[0][0], 'before-building-argument-table-parser') self.assertEqual(register_args[0][1], self.argument.override_required_args) def test_override_required_args_if_in_cmdline(self): args = ['--no-required-args'] self.argument.override_required_args(self.argument_table, args) self.assertFalse(self.mock_arg.required) def test_no_override_required_args_if_not_in_cmdline(self): args = [] self.argument.override_required_args(self.argument_table, args) self.assertTrue(self.mock_arg.required)
class TestOverrideRequiredArgsArgument(unittest.TestCase): def setUp(self): self.session = mock.Mock() self.argument = OverrideRequiredArgsArgument(self.session) # Set up a sample argument_table self.argument_table = {} self.mock_arg = mock.Mock() self.mock_arg.required = True self.argument_table["mock-arg"] = self.mock_arg def test_register_argument_action(self): register_args = self.session.register.call_args self.assertEqual(register_args[0][0], "before-building-argument-table-parser") self.assertEqual(register_args[0][1], self.argument.override_required_args) def test_override_required_args_if_in_cmdline(self): args = ["--no-required-args"] self.argument.override_required_args(self.argument_table, args) self.assertFalse(self.mock_arg.required) def test_no_override_required_args_if_not_in_cmdline(self): args = [] self.argument.override_required_args(self.argument_table, args) self.assertTrue(self.mock_arg.required)
def setUp(self): self.session = mock.Mock() self.argument = OverrideRequiredArgsArgument(self.session) # Set up a sample argument_table self.argument_table = {} self.mock_arg = mock.Mock() self.mock_arg.required = True self.argument_table['mock-arg'] = self.mock_arg
def setUp(self): self.session = mock.Mock() self.argument = OverrideRequiredArgsArgument(self.session) # Set up a sample argument_table self.argument_table = {} self.mock_arg = mock.Mock() self.mock_arg.required = True self.argument_table["mock-arg"] = self.mock_arg