class TestArgLoader:
def setup(self):
self.args = ["1.2.3.4"]
self.sut = ArgLoader(self.args)
def test_init_sets_expected_properties(self):
eq_(self.args, self.sut.in_args)
@mock.patch('lib.arg_loader.argparse')
def test_get_args_operates_as_expected(self, mock_argparse):
mock_parser = mock.MagicMock()
mock_argparse.ArgumentParser.return_value = mock_parser
mock_parser.parse_args.return_value = "Parsed Args"
actual = self.sut.get_args()
mock_argparse.ArgumentParser.assert_called_once_with(
description=DESCRIPTION)
mock_parser.add_argument.assert_called_once_with(
"version_number", help=VNUM_HELP_TEXT)
eq_("Parsed Args", actual)
def setup(self):
self.args = ["1.2.3.4"]
self.sut = ArgLoader(self.args)
