def test_init_sets_format_func_to_csv_function_when_csv_format_option_is_passed(
self, mock_to_csv):
output_format = output_formats_module.OutputFormat.CSV
formatter = output_formats_module.OutputFormatter(output_format)
for _ in formatter.get_formatted_output("TEST"):
pass
mock_to_csv.assert_called_once_with("TEST")
def test_init_sets_format_func_to_formatted_json_function_when_json_format_option_is_passed(
self, mock_to_json):
output_format = output_formats_module.OutputFormat.RAW
formatter = output_formats_module.OutputFormatter(output_format)
for _ in formatter.get_formatted_output([{"TEST": "FOOBAR"}]):
pass
mock_to_json.assert_called_once_with({"TEST": "FOOBAR"})
def test_init_sets_format_func_to_json_function_when_raw_json_format_option_is_passed(
self, mock_to_formatted_json):
output_format = output_formats_module.OutputFormat.JSON
formatter = output_formats_module.OutputFormatter(output_format)
for _ in formatter.get_formatted_output(["TEST"]):
pass
mock_to_formatted_json.assert_called_once_with("TEST")
def test_init_sets_format_func_to_table_function_when_table_format_option_is_passed(
self, mock_to_table):
output_format = output_formats_module.OutputFormat.TABLE
formatter = output_formats_module.OutputFormatter(output_format)
for _ in formatter.get_formatted_output("TEST"):
pass
mock_to_table.assert_called_once_with("TEST", None)
def test_init_sets_format_func_to_table_function_when_no_format_option_is_passed(
self, mock_to_table):
formatter = output_formats_module.OutputFormatter(None)
for _ in formatter.get_formatted_output("TEST"):
pass
mock_to_table.assert_called_once_with("TEST",
None,
include_header=True)