mirror of
https://github.com/hanzoai/agents.git
synced 2026-10-30 18:48:36 +00:00
zeekayandGitHub
5b5cf9e9db
Transforms hanzoai/agents from a marketplace config repo into a full agent orchestration platform by merging the AgentField infrastructure: - Go control plane (K8s-like agent orchestration, service mesh, DAG workflows) - Python SDK (FastAPI-based agent framework, LiteLLM, memory, DID/VC) - TypeScript SDK (@hanzo/agents - Vercel AI SDK, MCP, workflows) - Go SDK (idiomatic agent builder + client) - React 19 embedded dashboard (XyFlow DAGs, Recharts, Radix UI) - Docker/K8s/Helm deployment configs - Examples: hello world, deep research, RAG, simulation, benchmarks - Functional test suite across Python/Go/TypeScript All references rebranded: AgentField → Hanzo Agents - Module: github.com/hanzoai/agents/control-plane - Python: hanzo_agents - TypeScript: @hanzo/agents - Go: github.com/hanzoai/agents/sdk/go - Env vars: HANZO_AGENTS_* - Config: hanzo-agents.yaml Previous marketplace configs moved to config/ directory.
110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
//go:build integration
|
|
// +build integration
|
|
|
|
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/agents/control-plane/pkg/types"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
// Add CleanupWorkflow method to existing MockStorageProvider
|
|
func (m *MockStorageProvider) CleanupWorkflow(ctx context.Context, workflowID string, dryRun bool) (*types.WorkflowCleanupResult, error) {
|
|
args := m.Called(ctx, workflowID, dryRun)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*types.WorkflowCleanupResult), args.Error(1)
|
|
}
|
|
|
|
func TestCleanupWorkflowHandler_RequiresConfirmation(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
mockStorage := new(MockStorageProvider)
|
|
router := gin.New()
|
|
router.DELETE("/workflows/:workflow_id/cleanup", CleanupWorkflowHandler(mockStorage))
|
|
|
|
req, _ := http.NewRequest("DELETE", "/workflows/test-workflow-123/cleanup", nil)
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusBadRequest, w.Code)
|
|
assert.Contains(t, w.Body.String(), "confirmation_required")
|
|
}
|
|
|
|
func TestCleanupWorkflowHandler_DryRun(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
mockStorage := new(MockStorageProvider)
|
|
expectedResult := &types.WorkflowCleanupResult{
|
|
WorkflowID: "test-workflow-123",
|
|
DryRun: true,
|
|
DeletedRecords: map[string]int{
|
|
"execution_vcs": 5,
|
|
"workflow_vcs": 2,
|
|
"workflow_executions": 10,
|
|
"workflows": 1,
|
|
"total": 18,
|
|
},
|
|
FreedSpaceBytes: 0,
|
|
Success: true,
|
|
DurationMS: 150,
|
|
}
|
|
|
|
mockStorage.On("CleanupWorkflow", mock.Anything, "test-workflow-123", true).Return(expectedResult, nil)
|
|
|
|
router := gin.New()
|
|
router.DELETE("/workflows/:workflow_id/cleanup", CleanupWorkflowHandler(mockStorage))
|
|
|
|
req, _ := http.NewRequest("DELETE", "/workflows/test-workflow-123/cleanup?dry_run=true", nil)
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
assert.Contains(t, w.Body.String(), "test-workflow-123")
|
|
assert.Contains(t, w.Body.String(), "\"dry_run\":true")
|
|
|
|
mockStorage.AssertExpectations(t)
|
|
}
|
|
|
|
func TestCleanupWorkflowHandler_UIRoute(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
mockStorage := new(MockStorageProvider)
|
|
expectedResult := &types.WorkflowCleanupResult{
|
|
WorkflowID: "ui-workflow-456",
|
|
DryRun: false,
|
|
DeletedRecords: map[string]int{
|
|
"workflow_executions": 3,
|
|
"workflows": 1,
|
|
"total": 4,
|
|
},
|
|
FreedSpaceBytes: 2048,
|
|
Success: true,
|
|
DurationMS: 275,
|
|
}
|
|
|
|
mockStorage.On("CleanupWorkflow", mock.Anything, "ui-workflow-456", false).Return(expectedResult, nil)
|
|
|
|
router := gin.New()
|
|
router.DELETE("/workflows/:workflowId/cleanup", CleanupWorkflowHandler(mockStorage))
|
|
|
|
req, _ := http.NewRequest("DELETE", "/workflows/ui-workflow-456/cleanup?confirm=true", nil)
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
assert.Contains(t, w.Body.String(), "ui-workflow-456")
|
|
assert.Contains(t, w.Body.String(), "\"dry_run\":false")
|
|
|
|
mockStorage.AssertExpectations(t)
|
|
}
|