Tweaks

mtustin-handy · Oct 21, 2015 · be39b35 · be39b35
1 parent 1ca1776
commit be39b35
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/airflow/utils.py b/airflow/utils.py
@@ -119,8 +119,8 @@ def initdb():
                 host='localhost', login='root', password='',
                 schema='airflow'))
         session.commit()
-    conn = session.query(C).filter(C.conn_id == 'local_mysql').first()
 
+    conn = session.query(C).filter(C.conn_id == 'local_mysql').first()
     if not conn:
         session.add(
             models.Connection(

diff --git a/run_unit_tests.sh b/run_unit_tests.sh
@@ -8,8 +8,11 @@ which airflow > /dev/null || python setup.py develop
 #AIRFLOW_DB=$AIRFLOW_HOME/unittests.db
 #ls -s $AIRFLOW_DB > /dev/null 2>&1 || airflow initdb # if it's missing
 #ls -s $AIRFLOW_DB | egrep '^0 ' > /dev/null && airflow initdb # if it's blank
+
+echo "Initializing the DB"
 airflow initdb
 
+echo "Starting the unit tests"
 nosetests --with-doctest \
           --with-coverage \
           --cover-erase \