Function get_testsets

api/oss/src/routers/testset_router.py:345–397

Get all testsets. Returns: - A list of testset objects. Raises: - `HTTPException` with status code 404 if no testsets are found.

(
    request: Request,
)

@router.get("/", operation_id="get_testsets")
async def get_testsets(
    request: Request,
) -> List[TestSetOutputResponse]:
    """
    Get all testsets.

    Returns:
    - A list of testset objects.

    Raises:
    - `HTTPException` with status code 404 if no testsets are found.
    """

    try:
        if is_ee():
            has_permission = await check_action_access(
                user_uid=request.state.user_id,
                project_id=request.state.project_id,
                permission=Permission.VIEW_TESTSET,
            )
            if not has_permission:
                error_msg = (
                    "You do not have permission to perform this action. "
                    + "Please contact your organization admin."
                )
                log.error(error_msg)

                return JSONResponse(
                    status_code=403,
                    content={"detail": error_msg},
                )

        testsets = await db_manager.fetch_testsets_by_project_id(
            project_id=request.state.project_id,
        )

        return [
            TestSetOutputResponse(
                _id=str(testset.id),  # type: ignore
                name=testset.name,
                created_at=str(testset.created_at),
                updated_at=str(testset.updated_at),
            )
            for testset in testsets
        ]

    except Exception as e:
        log.exception(f"An error occurred: {str(e)}")

        raise HTTPException(
            status_code=500,
            detail=str(e),
        )


@router.get("/{testset_id}/", operation_id="get_single_testset")