101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
#
|
||
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
# not use this file except in compliance with the License. You may obtain
|
||
|
# a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
# License for the specific language governing permissions and limitations
|
||
|
# under the License.
|
||
|
|
||
|
from io import StringIO
|
||
|
import yaml
|
||
|
|
||
|
from unittest import mock
|
||
|
|
||
|
from cliff.formatters import yaml_format
|
||
|
from cliff.tests import base
|
||
|
from cliff.tests import test_columns
|
||
|
|
||
|
|
||
|
class TestYAMLFormatter(base.TestBase):
|
||
|
|
||
|
def test_format_one(self):
|
||
|
sf = yaml_format.YAMLFormatter()
|
||
|
c = ('a', 'b', 'c', 'd')
|
||
|
d = ('A', 'B', 'C', '"escape me"')
|
||
|
expected = {
|
||
|
'a': 'A',
|
||
|
'b': 'B',
|
||
|
'c': 'C',
|
||
|
'd': '"escape me"'
|
||
|
}
|
||
|
output = StringIO()
|
||
|
args = mock.Mock()
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
actual = yaml.safe_load(output.getvalue())
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_formattablecolumn_one(self):
|
||
|
sf = yaml_format.YAMLFormatter()
|
||
|
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 = StringIO()
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
value = output.getvalue()
|
||
|
print(len(value.splitlines()))
|
||
|
actual = yaml.safe_load(output.getvalue())
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_list(self):
|
||
|
sf = yaml_format.YAMLFormatter()
|
||
|
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'}
|
||
|
]
|
||
|
output = StringIO()
|
||
|
args = mock.Mock()
|
||
|
sf.add_argument_group(args)
|
||
|
sf.emit_list(c, d, output, args)
|
||
|
actual = yaml.safe_load(output.getvalue())
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_formattablecolumn_list(self):
|
||
|
sf = yaml_format.YAMLFormatter()
|
||
|
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 = StringIO()
|
||
|
sf.emit_list(c, d, output, args)
|
||
|
actual = yaml.safe_load(output.getvalue())
|
||
|
self.assertEqual(expected, actual)
|