def test_msg_json_merge_object(self): output = StringIO() console = ConsoleOutput(output, json=True) test_obj1 = {"test": 1, "foo": "foo"} test_obj2 = Mock(__json__=Mock(return_value={"test": 2})) console.msg_json(test_obj1, test_obj2, ["qux"], foo="bar", baz="qux") self.assertEqual( '{\n' ' "test": 2,\n' ' "foo": "bar",\n' ' "baz": "qux"\n' '}\n', output.getvalue()) self.assertEqual([("test", 1), ("foo", "foo")], list(test_obj1.items()))
def test_msg_json_merge_list(self): output = StringIO() console = ConsoleOutput(output, json=True) test_list1 = ["foo", "bar"] test_list2 = Mock(__json__=Mock(return_value={"foo": "bar"})) console.msg_json(test_list1, ["baz"], test_list2, {"foo": "bar"}, foo="bar", baz="qux") self.assertEqual( '[\n' ' "foo",\n' ' "bar",\n' ' "baz",\n' ' {\n "foo": "bar"\n },\n' ' {\n "foo": "bar"\n },\n' ' {\n "foo": "bar",\n "baz": "qux"\n }\n' ']\n', output.getvalue()) self.assertEqual(["foo", "bar"], test_list1)
def test_msg_json_error(self, mock_exit): output = StringIO() console = ConsoleOutput(output, json=True) console.msg_json({"error": "bad"}) self.assertEqual('''{\n "error": "bad"\n}\n''', output.getvalue()) mock_exit.assert_called_with(1)
def test_msg_json_object(self): output = StringIO() test_obj = _TestObj() console = ConsoleOutput(output, json=True) console.msg_json(test_obj) self.assertEqual('''{\n "test": 1\n}\n''', output.getvalue())
def test_msg_json(self): output = StringIO() console = ConsoleOutput(output, json=True) console.msg_json({"test": 1}) self.assertEqual('''{\n "test": 1\n}\n''', output.getvalue())
def test_msg_json_not_set(self): output = StringIO() console = ConsoleOutput(output) self.assertEqual(None, console.msg_json({"test": 1})) self.assertEqual("", output.getvalue())
def test_msg(self): output = StringIO() console = ConsoleOutput(output) console.msg("foo") console.msg_json({"test": 1}) self.assertEqual("foo\n", output.getvalue())
def test_msg_json_list(self): output = StringIO() console = ConsoleOutput(output, json=True) test_list = ["foo", "bar"] console.msg_json(test_list) self.assertEqual('[\n "foo",\n "bar"\n]\n', output.getvalue())
def test_msg_json_object(self): output = StringIO() console = ConsoleOutput(output, json=True) console.msg_json(Mock(__json__=Mock(return_value={"test": 1}))) self.assertEqual('{\n "test": 1\n}\n', output.getvalue())