Function setupTestServer

server/server_test.go:21–79

(t *testing.T)

)

func setupTestServer(t *testing.T) (*Server, func()) {
	// Create test store
	st, err := store.NewJSONStore(t.TempDir())
	require.NoError(t, err)

	// Create dependencies
	toolRegistry := tools.NewRegistry()
	sandboxFactory := sandbox.NewFactory()
	providerFactory := NewMockProviderFactory()
	templateRegistry := agent.NewTemplateRegistry()

	// Register test templates
	chatTemplate := &types.AgentTemplateDefinition{
		ID:           "chat",
		Version:      "1.0.0",
		SystemPrompt: "You are a helpful assistant.",
		Model:        "test-model",
		Tools:        []string{},
	}
	templateRegistry.Register(chatTemplate)

	defaultModel := &types.ModelConfig{
		Provider: "mock",
		Model:    "test-model",
	}
	routes := []router.StaticRouteEntry{
		{Task: "chat", Priority: router.PriorityQuality, Model: defaultModel},
	}
	rt := router.NewStaticRouter(defaultModel, routes)

	agentDeps := &agent.Dependencies{
		Store:            st,
		ToolRegistry:     toolRegistry,
		SandboxFactory:   sandboxFactory,
		ProviderFactory:  providerFactory,
		TemplateRegistry: templateRegistry,
		Router:           rt,
	}

	deps := &Dependencies{
		Store:     st,
		AgentDeps: agentDeps,
	}

	// Create server with test config
	config := DefaultConfig()
	config.Auth.APIKey.Enabled = false // Disable auth for tests

	srv, err := New(config, deps)
	require.NoError(t, err)

	cleanup := func() {
		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		_ = srv.Stop(ctx)
	}

	return srv, cleanup