Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions composer/config/running_dags.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
ephemeral_dataproc_spark_dag
wordcount_dag
5 changes: 2 additions & 3 deletions composer/dags/wordcount_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

import datetime
import os
from airflow import models
from airflow import models, settings
from airflow.contrib.operators.bigquery_operator import BigQueryOperator
from airflow.contrib.operators.dataflow_operator import DataFlowJavaOperator
from airflow.contrib.operators.gcs_download_operator import \
Expand All @@ -40,8 +40,7 @@

# Dynamic prefix gives us flexibility for running airflow in a ci container or
# on composer.
SQL_PREFIX = os.path.join(os.environ.get('AIRFLOW_HOME', '/home/airflow'),
'gcs', 'data', 'sql')
SQL_PREFIX = os.path.join(settings.DAGS_FOLDER, 'sql')

SHAKESPEARE_SQL = os.path.join(SQL_PREFIX, 'shakespeare_top_25.sql')

Expand Down
3 changes: 3 additions & 0 deletions helpers/run_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,3 +25,6 @@ fi

echo "running dataflow java tests..."
find ./dataflow/java/ -name pom.xml -execdir mvn test \;

echo "running airflow tests"
(cd composer && ./cloudbuild/bin/run_tests.sh ../bigquery/sql ./config/AirflowVariables.json ./plugins)