Starting on test coverage

README.md

@@ -2,9 +2,9 @@
 
 [Apache Airflow](https://airflow.apache.org) utilities for running long-running or always-on jobs with [supervisord](http://supervisord.org)
 
-[![Build Status](https://github.com/timkpaine/airflow-supervisor/actions/workflows/build.yml/badge.svg)](https://github.com/timkpaine/airflow-supervisor/actions?query=workflow%3A%22Build+Status%22)
-[![codecov](https://codecov.io/gh/timkpaine/airflow-supervisor/branch/main/graph/badge.svg)](https://codecov.io/gh/timkpaine/airflow-supervisor)
-[![License](https://img.shields.io/github/license/timkpaine/airflow-supervisor)](https://github.com/timkpaine/airflow-supervisor)
+[![Build Status](https://github.com/airflow-laminar/airflow-supervisor/actions/workflows/build.yml/badge.svg)](https://github.com/airflow-laminar/airflow-supervisor/actions?query=workflow%3A%22Build+Status%22)
+[![codecov](https://codecov.io/gh/airflow-laminar/airflow-supervisor/branch/main/graph/badge.svg)](https://codecov.io/gh/airflow-laminar/airflow-supervisor)
+[![License](https://img.shields.io/github/license/airflow-laminar/airflow-supervisor)](https://github.com/airflow-laminar/airflow-supervisor)
 [![PyPI](https://img.shields.io/pypi/v/airflow-supervisor.svg)](https://pypi.python.org/pypi/airflow-supervisor)
 
 ## Overview

airflow_supervisor/airflow/local.py

@@ -25,12 +25,12 @@
 
 
 class Supervisor(object):
-    _dag: "DAG"
+    _dag: DAG
     _cfg: SupervisorAirflowConfiguration
-    _kill_dag: "DAG"
+    _kill_dag: DAG
     _xmlrpc_client: SupervisorRemoteXMLRPCClient
 
-    def __init__(self, dag: "DAG", cfg: SupervisorAirflowConfiguration, **kwargs):
+    def __init__(self, dag: DAG, cfg: SupervisorAirflowConfiguration, **kwargs):
         # store config
         self._cfg = cfg
 

airflow_supervisor/tests/test_dag.py

@@ -0,0 +1,24 @@
+from airflow.models.dag import DAG
+
+from airflow_supervisor import Supervisor, SupervisorAirflowConfiguration
+
+
+class TestDag:
+    def test_instantiation(self, supervisor_airflow_configuration: SupervisorAirflowConfiguration):
+        dag = DAG(dag_id="test_dag", default_args={}, schedule=None, params={})
+        s = Supervisor(dag=dag, cfg=supervisor_airflow_configuration)
+        assert len(dag.tasks) == 17
+        assert dag.catchup is False
+        assert dag.concurrency == 1
+        assert dag.max_active_tasks == 1
+        assert dag.max_active_runs == 1
+
+        assert s.configure_supervisor in dag.tasks
+        assert s.configure_supervisor in dag.tasks
+        assert s.start_supervisor in dag.tasks
+        assert s.start_programs in dag.tasks
+        assert s.check_programs in dag.tasks
+        assert s.restart_programs in dag.tasks
+        assert s.stop_programs in dag.tasks
+        assert s.stop_supervisor in dag.tasks
+        assert s.unconfigure_supervisor in dag.tasks