def test_read(self):
result = JsonFormat().read(TEST_JSON_ELEMENT)
self.assertEqual(result, {'someObject': {'data': 'some data'}})
def test_read_invalid(self):
with self.assertRaises(BadFormatError) as context:
JsonFormat().read('notJson')
self.assertTrue(
'Failed to read content as JSON: ' in str(context.exception))
def test_convert_element(self):
element = {'someObject': {'data': 'some data'}}
output = JsonFormat().convert_element(element)
self.assertEqual(output, TEST_JSON_ELEMENT)
def test_convert_list(self):
test_list = ['abc', 123, {'someObject': {'data': 'some data'}}]
output = JsonFormat().convert_list(test_list)
self.assertEqual(output, TEST_JSON_LIST)
def default_output_format_handler():
return output_format_handler()\
.add_choice(YAML_VALUE, YamlFormat(), is_default=True)\
.add_choice(JSON_VALUE, JsonFormat())
def common_output_format_handler(table: Table):
return output_format_handler()\
.add_choice(TABLE_VALUE, TableFormat(table=table), is_default=True)\
.add_choice(YAML_VALUE, YamlFormat())\
.add_choice(JSON_VALUE, JsonFormat())
def default_file_inputs_handler(**kwargs):
return file_inputs_handler(**kwargs)\
.add_format(YAML_VALUE, YamlFormat())\
.add_format(JSON_VALUE, JsonFormat())