[AIRFLOW-4857] Add templated fields to SlackWebhookOperator (apache#5490

) (cherry picked from commit 8b7f93d)
kagai · Jun 27, 2019 · 3355ad9 · 3355ad9
1 parent 0660820
commit 3355ad9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/airflow/contrib/operators/slack_webhook_operator.py b/airflow/contrib/operators/slack_webhook_operator.py
@@ -54,6 +54,9 @@ class SlackWebhookOperator(SimpleHttpOperator):
     :type proxy: str
     """
 
+    template_fields = ['webhook_token', 'message', 'attachments', 'channel',
+                       'username', 'proxy', ]
+
     @apply_defaults
     def __init__(self,
                  http_conn_id=None,

diff --git a/tests/contrib/operators/test_slack_webhook_operator.py b/tests/contrib/operators/test_slack_webhook_operator.py
@@ -67,6 +67,17 @@ def test_execute(self):
         self.assertEqual(self._config['link_names'], operator.link_names)
         self.assertEqual(self._config['proxy'], operator.proxy)
 
+    def test_assert_templated_fields(self):
+        operator = SlackWebhookOperator(
+            task_id='slack_webhook_job',
+            dag=self.dag,
+            **self._config
+        )
+
+        template_fields = ['webhook_token', 'message', 'attachments', 'channel', 'username', 'proxy']
+
+        self.assertEqual(operator.template_fields, template_fields)
+
 
 if __name__ == '__main__':
     unittest.main()