get_ti_from_db(task_1).state = State.FAILED
    get_ti_from_db(task_2).state = State.SUCCESS
    get_ti_from_db(task_3).state = State.UPSTREAM_FAILED
    get_ti_from_db(task_4).state = State.FAILED
    get_ti_from_db(task_5).state = State.SKIPPED
    session.flush()
    altered = dag.set_task_instance_state(
        task_id=task_1.task_id,
        run_id=run_id,
        execution_date=execution_date,
        state=State.SUCCESS,
        session=session,
    )
    
    ti1 = get_ti_from_db(task_1)
    assert ti1.state == State.SUCCESS
    
    assert isinstance(inspect(ti1).attrs.dag_run.loaded_value, DagRun)
    
    assert get_ti_from_db(task_2).state == State.SUCCESS
    
    assert get_ti_from_db(task_3).state == State.NONE
    assert get_ti_from_db(task_4).state == State.NONE
    
    assert get_ti_from_db(task_5).state == State.SKIPPED
    dagrun.refresh_from_db(session=session)
    
    assert dagrun.get_state() == State.QUEUED
    assert {t.key for t in altered} == {("test_set_task_instance_state", "task_1", dagrun.run_id, 1, -1)}
def test_set_task_instance_state_mapped(dag_maker, session):