def test_cmd_list_if_invalid_command_given(self): self._replace_stdout_with_stringio() ret = admin.main(["this is not a valid command"]) self.assertEquals(ret, False) stdout = sys.stdout.getvalue() self._check_cmd_list(stdout)
def test_main(self): saved_commands = admin.commands def restore_commands(): admin.commands = saved_commands testcmdclass = mock.Mock() testcmd = mock.Mock() testcmdclass.return_value = testcmd admin.commands = {"Test": testcmdclass} self.addCleanup(restore_commands) admin.main(["Test", "arg2", "arg3"]) testcmd.init.assert_called_with(["arg2", "arg3"]) testcmd.assert_called_with()
def test_unauthorized_error(self): self._replace_stdout_with_stringio() self._define_test_cmd() ret = admin.main(["Foo", "-r"]) self.assertFalse(ret) self.assertEquals(sys.stdout.getvalue(), "Action not permitted\n")
def test_command_long_option(self): self._define_test_cmd() self.succesful = False admin.main(["Foo", "--mark_succesful"]) self.assertTrue(self.succesful)
def test_command_short_option(self): self._define_test_cmd() self.succesful = False admin.main(["Foo", "-m"]) self.assertTrue(self.succesful)