def test_mixture(self):
"""Test to ensure that both dict and OrderedDict can be parsed
by ordered_dump."""
ordered_dict = OrderedDict()
ordered_dict['a'] = {}
ordered_dict['b'] = OrderedDict()
for item in ordered_dict.values():
for i in reversed('abcdefg'):
item[i] = ord(i) - ord('a')
try:
parser.ordered_dump(ordered_dict,
Dumper=yaml.SafeDumper,
default_flow_style=False)
except Exception:
self.fail("No exceptions should be raised here.")
def test_mixture(self):
"""Test to ensure that both dict and OrderedDict can be parsed
by ordered_dump."""
ordered_dict = OrderedDict()
ordered_dict['a'] = {}
ordered_dict['b'] = OrderedDict()
for item in ordered_dict.values():
for i in reversed('abcdefg'):
item[i] = ord(i) - ord('a')
try:
parser.ordered_dump(ordered_dict,
Dumper=yaml.SafeDumper,
default_flow_style=False)
except Exception:
self.fail("No exceptions should be raised here.")
def _format_yaml(self, payload):
"""Convert the payload into a YAML string with proper
indentation and return it.
"""
return ordered_dump(payload,
Dumper=yaml.SafeDumper,
default_flow_style=False)
def test_output_order(self):
"""Test that ordered_dump perserves the order of OrderedDict."""
ordered_dict = OrderedDict()
for i in reversed('abcdefg'):
ordered_dict[i] = ord(i) - ord('a')
self.assertEqual(
parser.ordered_dump(ordered_dict,
Dumper=yaml.SafeDumper,
default_flow_style=False), self.CORRECT_OUTPUT)
def test_output_order(self):
"""Test that ordered_dump perserves the order of OrderedDict."""
ordered_dict = OrderedDict()
for i in reversed('abcdefg'):
ordered_dict[i] = ord(i) - ord('a')
self.assertEqual(parser.ordered_dump(ordered_dict,
Dumper=yaml.SafeDumper,
default_flow_style=False),
self.CORRECT_OUTPUT)
def receive(self, all=False, asset_input=None):
exported_objects = self.export_assets(all, asset_input)
stdout = click.get_text_stream('stdout')
if settings.format == 'human' or settings.format == 'json':
import json
stdout.write(json.dumps(exported_objects, indent=2))
elif settings.format == 'yaml':
import yaml
stdout.write(parser.ordered_dump(exported_objects, Dumper=yaml.SafeDumper, default_flow_style=False))
else:
raise TowerCLIError("Format {} is unsupported".format(settings.format))
stdout.write("\n")
def receive(self, all=False, asset_input=None):
exported_objects = self.export_assets(all, asset_input)
stdout = click.get_text_stream('stdout')
if settings.format == 'human' or settings.format == 'json':
import json
stdout.write(json.dumps(exported_objects, indent=2))
elif settings.format == 'yaml':
import yaml
stdout.write(parser.ordered_dump(exported_objects, Dumper=yaml.SafeDumper, default_flow_style=False))
else:
raise TowerCLIError("Format {} is unsupported".format(settings.format))
stdout.write("\n")
def _format_yaml(self, payload):
"""Convert the payload into a YAML string with proper
indentation and return it.
"""
return ordered_dump(payload, Dumper=yaml.SafeDumper,
default_flow_style=False)
