diff --git a/airflow/utils/log/es_task_handler.py b/airflow/utils/log/es_task_handler.py index d54cb1231f39b..47f970f6041f5 100644 --- a/airflow/utils/log/es_task_handler.py +++ b/airflow/utils/log/es_task_handler.py @@ -208,7 +208,6 @@ def set_context(self, ti): if self.json_format: self.formatter = JSONFormatter( - self.formatter._fmt, # pylint: disable=protected-access json_fields=self.json_fields, extras={ 'dag_id': str(ti.dag_id), diff --git a/tests/utils/log/test_es_task_handler.py b/tests/utils/log/test_es_task_handler.py index 408e7cf245cfb..b4e8fac344704 100644 --- a/tests/utils/log/test_es_task_handler.py +++ b/tests/utils/log/test_es_task_handler.py @@ -254,9 +254,8 @@ def test_set_context(self): self.assertTrue(self.es_task_handler.mark_end_on_close) def test_set_context_w_json_format_and_write_stdout(self): - self.es_task_handler.formatter = mock.MagicMock() - self.es_task_handler.formatter._fmt = mock.MagicMock() - self.es_task_handler.formatter._fmt.find = mock.MagicMock(return_value=1) + formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') + self.es_task_handler.formatter = formatter self.es_task_handler.write_stdout = True self.es_task_handler.json_format = True self.es_task_handler.set_context(self.ti)