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)