Add back in template fields for new operator (#61)

* Add back in template fields for new operator

* Update version number so PR can be a release candidate as well.

* Add template fields unit test

* Add query_to_validate as a template field and .sql files as template extensions to allow sql files to be templated and passed into query_to_validate

CHANGELOG.md

@@ -3,6 +3,9 @@
 ## Upcoming 
 * (please add here)
 
+## 0.2.1
+* [BUGFIX] Add back in removed template fields
+
 ## 0.2.0
 * [FEATURE] Enable the use of a default Checkpoint when a Checkpoint is not supplied, with the option to use the OpenLineage Validation Action
 * [FEATURE] Add support to use Airflow connection information instead of separate Great Expectation Datasources.

great_expectations_provider/operators/great_expectations.py

@@ -115,7 +115,15 @@ class GreatExpectationsOperator(BaseOperator):
 
     ui_color = "#AFEEEE"
     ui_fgcolor = "#000000"
-
+    template_fields = (
+        "run_name",
+        "conn_id",
+        "data_context_root_dir",
+        "checkpoint_name",
+        "checkpoint_kwargs",
+        "query_to_validate"
+    )
+    template_ext = (".sql")
     operator_extra_links = (GreatExpectationsDataDocsLink(),)
 
     def __init__(

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="airflow-provider-great-expectations",
-    version="0.2.0",
+    version="0.2.1",
     author="Great Expectations",
     description="An Apache Airflow provider for Great Expectations",
     entry_points="""

tests/operators/test_great_expectations.py

@@ -260,6 +260,29 @@ def configured_sql_operator(in_memory_data_context_config):
     return operator
 
 
+def test_great_expectations_operator__assert_template_fields_exist():
+    operator = GreatExpectationsOperator(
+        task_id="task_id",
+        data_context_root_dir=ge_root_dir,
+        checkpoint_name="taxi.pass.chk",
+    )
+    assert "run_name" in operator.template_fields
+    assert "conn_id" in operator.template_fields
+    assert "data_context_root_dir" in operator.template_fields
+    assert "checkpoint_name" in operator.template_fields
+    assert "checkpoint_kwargs" in operator.template_fields
+    assert "query_to_validate" in operator.template_fields
+
+
+def test_great_expectations_operator__assert_template_ext_exist():
+    operator = GreatExpectationsOperator(
+        task_id="task_id",
+        data_context_root_dir=ge_root_dir,
+        checkpoint_name="taxi.pass.chk",
+    )
+    assert ".sql" in operator.template_ext
+
+
 def test_great_expectations_operator__context_root_dir_and_checkpoint_name_pass():
     operator = GreatExpectationsOperator(
         task_id="task_id",