"""Integration tests for lambda function."""
from lambdacommon import util

from tests.consts.sql import ASSERT_JOB_INSERT
from tests.consts import config


def test_job_priorities_written_to_db(direct_delivery_db, invoke_lambda):
    """Assert jobs created after lambda run."""
    with util.dd_connection(config.DD_MYSQL_CONN_INFO) as conn:
        with conn.cursor() as cursor:
            cursor.execute(ASSERT_JOB_INSERT)
            results = cursor.fetchall()

    assert len(results) == 5