# Copyright (c) 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved # # Permission is hereby granted, free of charge, to any person obtaining a # copy of this software and associated documentation files (the # "Software"), to deal in the Software without restriction, including # without limitation the rights to use, copy, modify, merge, publish, dis- # tribute, sublicense, and/or sell copies of the Software, and to permit # persons to whom the Software is furnished to do so, subject to the fol- # lowing conditions: # # The above copyright notice and this permission notice shall be included # in all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS # IN THE SOFTWARE. # import unittest import sys from awscli.compat import six import mock from awscli import text class TestSection(unittest.TestCase): def format_text(self, data, stream=None): if stream is None: stream = six.StringIO() text.format_text(data, stream=stream) return stream.getvalue() def assert_text_renders_to(self, data, expected_rendering): rendered = self.format_text(data) self.assertEqual(rendered, expected_rendering) def test_dict_format(self): self.assert_text_renders_to(dict(a=1, b=2, c=3), "1\t2\t3\n") def test_list_format(self): self.assert_text_renders_to([1, 2, 3], "1\t2\t3\n") def test_list_of_dicts(self): self.assert_text_renders_to( {'foo': [dict(a=1, b=2, c=3), dict(a=4, b=5, c=6)]}, 'FOO\t1\t2\t3\n' 'FOO\t4\t5\t6\n') def test_multiple_list_of_dicts(self): self.assert_text_renders_to( {'foo': [dict(a=1, b=2, c=3), dict(a=4, b=5, c=6)], 'zoo': [dict(a=7, b=8, c=9), dict(a=0, b=1, c=2)]}, 'FOO\t1\t2\t3\n' 'FOO\t4\t5\t6\n' 'ZOO\t7\t8\t9\n' 'ZOO\t0\t1\t2\n' ) def test_single_scalar_number(self): self.assert_text_renders_to(10, '10\n') def test_list_of_single_number(self): self.assert_text_renders_to([10], '10\n') def test_list_of_multiple_numbers(self): self.assert_text_renders_to([10, 10, 10], '10\t10\t10\n') def test_different_keys_in_sublists(self): self.assert_text_renders_to( # missing "b" adds "d" {'foo': [dict(a=1, b=2, c=3), dict(a=4, c=5), dict(a=6, d=7)]}, 'FOO\t1\t2\t3\t\n' 'FOO\t4\t\t5\t\n' 'FOO\t6\t\t\t7\n' ) def test_different_keys_in_nested_sublists(self): self.assert_text_renders_to({'bar':[ {'foo': [dict(a=1, b=2, c=3), dict(a=4, c=5)]}, {'foo': [dict(b=6, d=7), dict(b=8, c=9)]}, ]}, 'FOO\t1\t2\t3\n' 'FOO\t4\t\t5\n' 'FOO\t6\t\t7\n' 'FOO\t8\t9\t\n' ) def test_different_keys_in_deeply_nested_sublists(self): self.assert_text_renders_to({'bar':[ {'foo': [[[dict(a=1, b=2, c=3), dict(a=4, c=5)]]]}, {'foo': [[[dict(b=6, d=7), dict(b=8, c=9)]]]}, ]}, 'FOO\t1\t2\t3\n' 'FOO\t4\t\t5\n' 'FOO\t6\t\t7\n' 'FOO\t8\t9\t\n' ) def test_scalars_and_complex_types(self): self.assert_text_renders_to( {'foo': [dict(a=1, b=dict(y='y', z='z'), c=3), dict(a=4, b=dict(y='y', z='z'), c=6)]}, 'FOO\t1\t3\n' 'B\ty\tz\n' 'FOO\t4\t6\n' 'B\ty\tz\n') def test_nested_list_of_lists(self): self.assert_text_renders_to( [['1', '2', '3'], ['4', '5', '6']], '1\t2\t3\n' '4\t5\t6\n' ) def test_deeply_nested_lists(self): self.assert_text_renders_to( [ [['1', '2', '3'], ['4', '5', '6']], [['7', '8', '9'], ['0', '1', '2']], ], '1\t2\t3\n' '4\t5\t6\n' '7\t8\t9\n' '0\t1\t2\n' ) def test_unicode_text(self): self.assert_text_renders_to([['1', '2', u'\u2713']], u'1\t2\t\u2713\n') def test_single_scalar_value(self): self.assert_text_renders_to('foobarbaz', 'foobarbaz\n') def test_empty_list(self): self.assert_text_renders_to([], '') def test_empty_inner_list(self): self.assert_text_renders_to([[]], '') def test_deeploy_nested_empty_list(self): self.assert_text_renders_to([[[[]]]], '') def test_deeploy_nested_single_scalar(self): self.assert_text_renders_to([[[['a']]]], 'a\n') def test_empty_list_mock_calls(self): # We also need this test as well as test_empty_list # because we want to ensure that write() is never called with # a list object. fake_stream = mock.Mock() self.format_text(data=[], stream=fake_stream) # We should not call .write() at all for an empty list. self.assertFalse(fake_stream.write.called) def test_list_of_strings_in_dict(self): self.assert_text_renders_to( {'KeyName': ['a', 'b', 'c']}, 'KEYNAME\ta\n' 'KEYNAME\tb\n' 'KEYNAME\tc\n') def test_inconsistent_sublists(self): self.assert_text_renders_to( [ [['1', '2'], ['3', '4', '5', '6']], [['7', '8', '9'], ['0']] ], '1\t2\n' '3\t4\t5\t6\n' '7\t8\t9\n' '0\n' ) def test_lists_mixed_with_scalars(self): self.assert_text_renders_to( [ ['a', 'b', ['c', 'd']], ['e', 'f', ['g', 'h']] ], 'a\tb\n' 'c\td\n' 'e\tf\n' 'g\th\n' ) def test_deeply_nested_with_scalars(self): self.assert_text_renders_to( [ ['a', 'b', ['c', 'd', ['e', 'f', ['g', 'h']]]], ['i', 'j', ['k', 'l', ['m', 'n', ['o', 'p']]]], ], 'a\tb\n' 'c\td\n' 'e\tf\n' 'g\th\n' 'i\tj\n' 'k\tl\n' 'm\tn\n' 'o\tp\n' ) def test_deeply_nested_with_identifier(self): self.assert_text_renders_to( {'foo': [ ['a', 'b', ['c', 'd']], ['e', 'f', ['g', 'h']] ]}, 'FOO\ta\n' 'FOO\tb\n' 'FOO\tc\n' 'FOO\td\n' 'FOO\te\n' 'FOO\tf\n' 'FOO\tg\n' 'FOO\th\n' ) if __name__ == '__main__': unittest.main()