def test_print_json(self, JD):
from kamaki.cli.utils import print_json, INDENT_TAB
out = StringIO()
print_json(u'some data', out)
JD.assert_called_once_with(u'some data', indent=INDENT_TAB)
self.assertEqual(out.getvalue(), u'(dumps output)\n')
def print_json(self, *args, **kwargs):
kwargs.setdefault('out', self)
return print_json(*args, **kwargs)
def _print(self, output, print_method=print_items, **print_method_kwargs):
if self['json_output'] or self['output_format']:
print_json(output, out=self._out)
else:
print_method_kwargs.setdefault('out', self._out)
print_method(output, **print_method_kwargs)
def print_json(self, *args, **kwargs):
kwargs.setdefault('out', self)
return print_json(*args, **kwargs)
def _optional_output(self, r):
if self['json_output']:
print_json(r, out=self._out)
elif self['with_output']:
print_items([r] if isinstance(r, dict) else r, out=self._out)
