Function _create_run

tests/unit/zen_stores/test_run_statistics.py:88–119

(
    sql_store: SqlZenStore,
    *,
    project_id: UUID,
    name: str,
    status: ExecutionStatus = ExecutionStatus.COMPLETED,
    start_time: Optional[datetime] = None,
    end_time: Optional[datetime] = None,
    pipeline_id: Optional[UUID] = None,
    snapshot_id: Optional[UUID] = None,
    user_id: Optional[UUID] = None,
    index: int = 1,
)

def _create_run(
    sql_store: SqlZenStore,
    *,
    project_id: UUID,
    name: str,
    status: ExecutionStatus = ExecutionStatus.COMPLETED,
    start_time: Optional[datetime] = None,
    end_time: Optional[datetime] = None,
    pipeline_id: Optional[UUID] = None,
    snapshot_id: Optional[UUID] = None,
    user_id: Optional[UUID] = None,
    index: int = 1,
) -> UUID:
    run = PipelineRunSchema(
        id=uuid4(),
        project_id=project_id,
        name=name,
        orchestrator_run_id=None,
        start_time=start_time,
        end_time=end_time,
        status=status.value,
        index=index,
        in_progress=False,
        enable_heartbeat=False,
        pipeline_id=pipeline_id,
        snapshot_id=snapshot_id,
        user_id=user_id,
    )
    with Session(sql_store.engine, expire_on_commit=False) as session:
        session.add(run)
        session.commit()
    return run.id


def _create_run_metadata(