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())