Avik Datta 4 年 前
コミット
1c84b936ad
1 ファイル変更25 行追加0 行削除
  1. 25 0
      test/dag_check.py

+ 25 - 0
test/dag_check.py

@@ -0,0 +1,25 @@
+import unittest
+from airflow.models import DagBag,Variable
+
+class TestAllDagIntegrity(unittest.TestCase):
+
+  LOAD_SECOND_THRESHOLD = 2
+
+  def setUp(self):
+    self.dagbag = DagBag('dags')
+
+
+  def test_dag_import(self):
+    self.assertFalse(
+      len(self.dagbag.import_errors),
+      'DAG import errors: {}'.format(
+        self.dagbag.import_errors))
+
+if __name__=='__main__':
+  checks = \
+    unittest.\
+      TestLoader().\
+        loadTestsFromTestCase(TestAllDagIntegrity)
+  unittest.\
+    TextTestRunner(verbosity=1).\
+      run(checks)