test_dag.py

import pytest
from csst_dag import gen_level1_dag_message


def test_gen_level1_dag_message_msc_sls():
    msg = gen_level1_dag_message(obsid="11009101682009", chipid="01", return_dict=True)
    assert msg["dag_id"] == "csst-msc-l1-sls"


def test_gen_level1_dag_message_msc_mbi():
    msg = gen_level1_dag_message(obsid="11009101682009", chipid="09", return_dict=True)
    assert msg["dag_id"] == "csst-msc-l1-mbi"


def test_gen_level1_dag_message_msc_invalid_chipid():
    with pytest.raises(ValueError):
        msg = gen_level1_dag_message(
            obsid="11009101682009", chipid="00", return_dict=True
        )


def test_gen_level1_dag_message_msc_none():
    msg = gen_level1_dag_message(obsid="12009101682009", chipid="00", return_dict=True)
    assert msg is None