return (
            session.query(TI)
            .filter(
                TI.dag_id == dag.dag_id,
                TI.task_id == task.task_id,
                TI.run_id == dagrun.run_id,
            )
            .one()
        )
    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.SUCCESS
    get_ti_from_db(task_5).state = State.UPSTREAM_FAILED
    get_ti_from_db(task_6).state = State.FAILED
    get_ti_from_db(task_7).state = State.SKIPPED
    session.flush()
    altered = dag.set_task_group_state(
        group_id=section_1.group_id,
        run_id=run_id,
        execution_date=execution_date,
        state=State.SUCCESS,
        session=session,
    )
    
    assert get_ti_from_db(task_1).state == State.SUCCESS
    
    assert get_ti_from_db(task_2).state == State.SUCCESS