def test_list(self): sf = json_format.JSONFormatter() c = ('a', 'b', 'c') d = (('A1', 'B1', 'C1'), ('A2', 'B2', 'C2'), ('A3', 'B3', 'C3')) expected = [{ 'a': 'A1', 'b': 'B1', 'c': 'C1' }, { 'a': 'A2', 'b': 'B2', 'c': 'C2' }, { 'a': 'A3', 'b': 'B3', 'c': 'C3' }] args = mock.Mock() sf.add_argument_group(args) args.noindent = True output = six.StringIO() sf.emit_list(c, d, output, args) value = output.getvalue() self.assertEqual(1, len(value.splitlines())) actual = json.loads(value) self.assertEqual(expected, actual) args.noindent = False output = six.StringIO() sf.emit_list(c, d, output, args) value = output.getvalue() self.assertEqual(17, len(value.splitlines())) actual = json.loads(value) self.assertEqual(expected, actual)
def test_formattablecolumn_one(self): sf = json_format.JSONFormatter() c = ('a', 'b', 'c', 'd') d = ('A', 'B', 'C', test_columns.FauxColumn(['the', 'value'])) expected = { 'a': 'A', 'b': 'B', 'c': 'C', 'd': ['the', 'value'], } args = mock.Mock() sf.add_argument_group(args) args.noindent = True output = six.StringIO() sf.emit_one(c, d, output, args) value = output.getvalue() print(len(value.splitlines())) self.assertEqual(1, len(value.splitlines())) actual = json.loads(value) self.assertEqual(expected, actual)
def test_json_format_formattablecolumn_list(): sf = json_format.JSONFormatter() c = ('a', 'b', 'c') d = (('A1', 'B1', test_columns.FauxColumn(['the', 'value'])), ) expected = [ { 'a': 'A1', 'b': 'B1', 'c': ['the', 'value'] }, ] args = mock.Mock() sf.add_argument_group(args) args.noindent = True output = six.StringIO() sf.emit_list(c, d, output, args) value = output.getvalue() assert 1 == len(value.splitlines()) actual = json.loads(value) assert expected == actual
def test_one(self): sf = json_format.JSONFormatter() c = ('a', 'b', 'c', 'd') d = ('A', 'B', 'C', '"escape me"') expected = {'a': 'A', 'b': 'B', 'c': 'C', 'd': '"escape me"'} args = mock.Mock() sf.add_argument_group(args) args.noindent = True output = six.StringIO() sf.emit_one(c, d, output, args) value = output.getvalue() print(len(value.splitlines())) self.assertEqual(1, len(value.splitlines())) actual = json.loads(value) self.assertEqual(expected, actual) args.noindent = False output = six.StringIO() sf.emit_one(c, d, output, args) value = output.getvalue() self.assertEqual(6, len(value.splitlines())) actual = json.loads(value) self.assertEqual(expected, actual)