Example for testing SLA in Airflow

my_dag_to_test_sla.py

from datetime import datetime, timedelta
from airflow.models import DAG
from airflow.operators.bash_operator import BashOperator

dag = DAG(
    dag_id="my_dag_to_test_sla",
    catchup=False,
    start_date=datetime(2021, 8, 28, 0, 0, 0),
    # If schedule_interval=None, SLA won't be evaluated for this DAG!
    schedule_interval="*/5 * * * *",
)

task_1 = BashOperator(
    task_id='task_1',
    bash_command='sleep 150',
    dag=dag,
)

task_2 = BashOperator(
    dag=dag,
    task_id='task_2',
    bash_command='echo 1',
    # If the task finishes (or starts) after 120 seconds,
    #  it will register an SLA miss
    sla=timedelta(seconds=120)
)

# The task `task_2` will start after 150 seconds
#  so Airflow will register an SLA miss
task_1 >> task_2