yao/agent/robot/api/robot_test.go
Max c6ca68876d Refactor API Structure and Update TODO for MVP Integration
- Updated the API documentation in TODO.md to reflect the completion of the Go API implementation and end-to-end tests for the main flow (P0 → P1 → P2 → P3 → P4).
- Clarified that Process handlers and JSAPI are optional features, moving them to Phase 12, while emphasizing that the Go API is sufficient for MVP integration.
- Removed obsolete API files (`api.go`, `jsapi.go`, `process.go`) as they were stubs and not implemented in Phase 10, streamlining the codebase.
- Revised the structure of the TODO.md to better outline the goals and tasks for Phases 11 and 12, ensuring clear tracking of progress and dependencies.
2026-01-19 09:17:33 +08:00

101 lines
2.6 KiB
Go

package api_test
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/yaoapp/yao/agent/robot/api"
"github.com/yaoapp/yao/agent/robot/types"
"github.com/yaoapp/yao/agent/testutils"
)
// TestGetRobotValidation tests parameter validation for GetRobot
func TestGetRobotValidation(t *testing.T) {
if testing.Short() {
t.Skip("Skipping integration test")
}
testutils.Prepare(t)
defer testutils.Clean(t)
t.Run("returns error for empty member_id", func(t *testing.T) {
ctx := types.NewContext(nil, nil)
robot, err := api.GetRobot(ctx, "")
assert.Error(t, err)
assert.Nil(t, robot)
assert.Contains(t, err.Error(), "member_id is required")
})
t.Run("returns error for non-existent robot", func(t *testing.T) {
ctx := types.NewContext(nil, nil)
robot, err := api.GetRobot(ctx, "non_existent_member_id_xyz")
assert.Error(t, err)
assert.Nil(t, robot)
})
}
// TestListRobotsValidation tests parameter validation for ListRobots
func TestListRobotsValidation(t *testing.T) {
if testing.Short() {
t.Skip("Skipping integration test")
}
testutils.Prepare(t)
defer testutils.Clean(t)
ctx := types.NewContext(nil, nil)
t.Run("applies default pagination when query is nil", func(t *testing.T) {
result, err := api.ListRobots(ctx, nil)
assert.NoError(t, err)
assert.NotNil(t, result)
assert.Equal(t, 1, result.Page)
assert.Equal(t, 20, result.PageSize)
})
t.Run("applies default pagination when values are zero", func(t *testing.T) {
result, err := api.ListRobots(ctx, &api.ListQuery{
Page: 0,
PageSize: 0,
})
assert.NoError(t, err)
assert.NotNil(t, result)
assert.Equal(t, 1, result.Page)
assert.Equal(t, 20, result.PageSize)
})
t.Run("caps pagesize at 100", func(t *testing.T) {
result, err := api.ListRobots(ctx, &api.ListQuery{
Page: 1,
PageSize: 500,
})
assert.NoError(t, err)
assert.NotNil(t, result)
assert.Equal(t, 100, result.PageSize)
})
}
// TestGetRobotStatusValidation tests parameter validation for GetRobotStatus
func TestGetRobotStatusValidation(t *testing.T) {
if testing.Short() {
t.Skip("Skipping integration test")
}
testutils.Prepare(t)
defer testutils.Clean(t)
t.Run("returns error for empty member_id", func(t *testing.T) {
ctx := types.NewContext(nil, nil)
status, err := api.GetRobotStatus(ctx, "")
assert.Error(t, err)
assert.Nil(t, status)
assert.Contains(t, err.Error(), "member_id is required")
})
t.Run("returns error for non-existent robot", func(t *testing.T) {
ctx := types.NewContext(nil, nil)
status, err := api.GetRobotStatus(ctx, "non_existent_member_id_xyz")
assert.Error(t, err)
assert.Nil(t, status)
})
}