96 lines
3.3 KiB
Python
96 lines
3.3 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.
|
||
|
|
||
|
import argparse
|
||
|
import io
|
||
|
from unittest import mock
|
||
|
|
||
|
from cliff.formatters import shell
|
||
|
from cliff.tests import base
|
||
|
from cliff.tests import test_columns
|
||
|
|
||
|
|
||
|
class TestShellFormatter(base.TestBase):
|
||
|
|
||
|
def test(self):
|
||
|
sf = shell.ShellFormatter()
|
||
|
c = ('a', 'b', 'c', 'd')
|
||
|
d = ('A', 'B', 'C', '"escape me"')
|
||
|
expected = 'a="A"\nb="B"\nd="\\"escape me\\""\n'
|
||
|
output = io.StringIO()
|
||
|
args = mock.Mock()
|
||
|
args.variables = ['a', 'b', 'd']
|
||
|
args.prefix = ''
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
actual = output.getvalue()
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_args(self):
|
||
|
sf = shell.ShellFormatter()
|
||
|
c = ('a', 'b', 'c', 'd')
|
||
|
d = ('A', 'B', 'C', '"escape me"')
|
||
|
expected = 'Xd="\\"escape me\\""\n'
|
||
|
output = io.StringIO()
|
||
|
# Parse arguments as if passed on the command-line
|
||
|
parser = argparse.ArgumentParser(description='Testing...')
|
||
|
sf.add_argument_group(parser)
|
||
|
parsed_args = parser.parse_args(['--variable', 'd', '--prefix', 'X'])
|
||
|
sf.emit_one(c, d, output, parsed_args)
|
||
|
actual = output.getvalue()
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_formattable_column(self):
|
||
|
sf = shell.ShellFormatter()
|
||
|
c = ('a', 'b', 'c')
|
||
|
d = ('A', 'B', test_columns.FauxColumn(['the', 'value']))
|
||
|
expected = '\n'.join([
|
||
|
'a="A"',
|
||
|
'b="B"',
|
||
|
'c="[\'the\', \'value\']"\n',
|
||
|
])
|
||
|
output = io.StringIO()
|
||
|
args = mock.Mock()
|
||
|
args.variables = ['a', 'b', 'c']
|
||
|
args.prefix = ''
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
actual = output.getvalue()
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_non_string_values(self):
|
||
|
sf = shell.ShellFormatter()
|
||
|
c = ('a', 'b', 'c', 'd', 'e')
|
||
|
d = (True, False, 100, '"esc"', str('"esc"'))
|
||
|
expected = ('a="True"\nb="False"\nc="100"\n'
|
||
|
'd="\\"esc\\""\ne="\\"esc\\""\n')
|
||
|
output = io.StringIO()
|
||
|
args = mock.Mock()
|
||
|
args.variables = ['a', 'b', 'c', 'd', 'e']
|
||
|
args.prefix = ''
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
actual = output.getvalue()
|
||
|
self.assertEqual(expected, actual)
|
||
|
|
||
|
def test_non_bash_friendly_values(self):
|
||
|
sf = shell.ShellFormatter()
|
||
|
c = ('a', 'foo-bar', 'provider:network_type')
|
||
|
d = (True, 'baz', 'vxlan')
|
||
|
expected = 'a="True"\nfoo_bar="baz"\nprovider_network_type="vxlan"\n'
|
||
|
output = io.StringIO()
|
||
|
args = mock.Mock()
|
||
|
args.variables = ['a', 'foo-bar', 'provider:network_type']
|
||
|
args.prefix = ''
|
||
|
sf.emit_one(c, d, output, args)
|
||
|
actual = output.getvalue()
|
||
|
self.assertEqual(expected, actual)
|