def test_exit_json(self, mock_exit): output = StringIO() console = ConsoleOutput(output, json=True) console.exit("error") self.assertEqual('''{\n "error": "error"\n}\n''', output.getvalue()) mock_exit.assert_called_with(1)
def test_exit(self, mock_exit): output = StringIO() console = ConsoleOutput(output) console.exit("error") self.assertEqual("error: error\n", output.getvalue()) mock_exit.assert_called_with(1)