def test_main(self, mock_get_template):
utils.main("render", "somepath", "a=1", "b=2")
mock_get_template.assert_called_once_with("somepath")
mock_get_template.return_value.render.assert_called_once_with(
a="1", b="2"
)
def test_main_bad_input(self, mock_stderr, mock_print):
self.assertTrue(utils.HELP_MESSAGE.startswith("Usage:"))
for args in ([], ["not_a_render"], ["render"],
["render", "expected_arg", "unexpected_arg"]):
self.assertEqual(1, utils.main(*args))
mock_print.assert_called_once_with(utils.HELP_MESSAGE,
file=mock_stderr)
mock_print.reset_mock()
def test_main_bad_input(self, mock_stderr, mock_print):
self.assertTrue(utils.HELP_MESSAGE.startswith("Usage:"))
for args in ([], ["not_a_render"], ["render"],
["render", "expected_arg", "unexpected_arg"]):
self.assertEqual(1, utils.main(*args))
mock_print.assert_called_once_with(utils.HELP_MESSAGE,
file=mock_stderr)
mock_print.reset_mock()
def test_main(self, mock_get_template):
utils.main("render", "somepath", "a=1", "b=2")
mock_get_template.assert_called_once_with("somepath")
mock_get_template.return_value.render.assert_called_once_with(a="1",
b="2")
