def test_remove_color_no_tty(self, mock_isatty): output_producer = OutputProducer(cli_ctx=self.mock_ctx) mock_isatty.return_value = False formatter = output_producer.get_formatter('jsonc') self.assertEqual(formatter, format_json) formatter = output_producer.get_formatter('yamlc') self.assertEqual(formatter, format_yaml) mock_isatty.return_value = True formatter = output_producer.get_formatter('jsonc') self.assertEqual(formatter, format_json_color) formatter = output_producer.get_formatter('yamlc') self.assertEqual(formatter, format_yaml_color)
def test_remove_color_no_tty(self): output_producer = OutputProducer(cli_ctx=self.mock_ctx) self.mock_ctx.enable_color = False formatter = output_producer.get_formatter('jsonc') self.assertEqual(formatter, format_json) formatter = output_producer.get_formatter('yamlc') self.assertEqual(formatter, format_yaml) self.mock_ctx.enable_color = True formatter = output_producer.get_formatter('jsonc') self.assertEqual(formatter, format_json_color) formatter = output_producer.get_formatter('yamlc') self.assertEqual(formatter, format_yaml_color)