"""Tests that dags load successfully."""
import unittest

from airflow.models import DagBag


class TestDagIntegrity(unittest.TestCase):
    """Dag Integrity Test Class."""

    LOAD_SECOND_THRESHOLD = 2

    def setUp(self):
        """Set up."""
        self.dagbag = DagBag()

    def test_import_dags(self):
        """Test loading available dags."""
        self.assertFalse(
            len(self.dagbag.import_errors),
            'DAG import failures. Errors: {}'.format(self.dagbag.import_errors))


suite = unittest.TestLoader().loadTestsFromTestCase(TestDagIntegrity)
unittest.TextTestRunner(verbosity=2).run(suite)