Setting up necessary dependencies for tests

run_unit_tests.sh

@@ -1,5 +1,21 @@
 export AIRFLOW_HOME=${AIRFLOW_HOME:=~/airflow}
 export AIRFLOW_CONFIG=$AIRFLOW_HOME/unittests.cfg
-nosetests --with-doctest --with-coverage --cover-erase --cover-html --cover-package=airflow -v --cover-html-dir=airflow/www/static/coverage --logging-level=DEBUG
+
+# Generate the `airflow` executable if needed
+which airflow > /dev/null || python setup.py develop
+
+# initialize the test db
+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
+
+nosetests --with-doctest \
+          --with-coverage \
+          --cover-erase \
+          --cover-html \
+          --cover-package=airflow \
+          --cover-html-dir=airflow/www/static/coverage \
+          -v \
+          --logging-level=DEBUG
 # To run individual tests:
 # nosetests tests.core:CoreTest.test_scheduler_job