Add new composer quickstart code (#2249)

* Add new quickstart code

* Modify bash command

composer/workflows/quickstart.py

@@ -0,0 +1,43 @@
+# Copyright 2019 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# [START composer_quickstart]
+import datetime
+
+import airflow
+from airflow.operators import bash_operator
+
+YESTERDAY = datetime.datetime.now() - datetime.timedelta(days=1)
+
+default_args = {
+    'owner': 'Composer Example',
+    'depends_on_past': False,
+    'email': [''],
+    'email_on_failure': False,
+    'email_on_retry': False,
+    'retries': 1,
+    'retry_delay': datetime.timedelta(minutes=5),
+    'start_date': YESTERDAY,
+}
+
+with airflow.DAG(
+        'composer_sample_dag',
+        'catchup=False',
+        default_args=default_args,
+        schedule_interval=datetime.timedelta(days=1)) as dag:
+
+    # Print the dag_run id from the Airflow logs
+    print_dag_run_conf = bash_operator.BashOperator(
+        task_id='print_dag_run_conf', bash_command='echo {{ dag_run.id }}')
+# [END composer_quickstart]

composer/workflows/quickstart_test.py

@@ -0,0 +1,31 @@
+# Copyright 2019 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from airflow import models
+
+from . import unit_testing
+
+
+def test_dag_import():
+    """Test that the DAG file can be successfully imported.
+
+    This tests that the DAG can be parsed, but does not run it in an Airflow
+    environment. This is a recommended sanity check by the official Airflow
+    docs: https://airflow.incubator.apache.org/tutorial.html#testing
+    """
+    models.Variable.set('gcs_bucket', 'example_bucket')
+    models.Variable.set('gcp_project', 'example-project')
+    models.Variable.set('gce_zone', 'us-central1-f')
+    from . import quickstart as module
+    unit_testing.assert_has_valid_dag(module)