orchestration_print_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"os"
6	"strings"
7	"testing"
8)
9
10func TestRunOrchestratedPrintEmitsOnlyFinalAnswer(t *testing.T) {
11	home := t.TempDir()
12	t.Setenv("ZUT_HOME", home)
13	server := orchestratedModeTestServer("final answer")
14	defer server.Close()
15
16	output, runErr := captureTestStdout(t, func() error {
17		return runOrchestratedPrintMode(context.Background(), Args{
18			Mode:        ModePrint,
19			Orchestrate: true,
20			Provider:    "openai",
21			Model:       "gpt-5",
22			BaseURL:     server.URL,
23			APIKey:      "test-key",
24			Prompt:      "say hello",
25		}, "test")
26	})
27	if runErr != nil {
28		t.Fatalf("runOrchestratedPrintMode error: %v", runErr)
29	}
30	if got := output; got != "final answer\n" {
31		t.Fatalf("stdout = %q, want final answer only", got)
32	}
33}
34
35func TestRunOrchestratedPrintPropagatesConfigLoadError(t *testing.T) {
36	home := t.TempDir()
37	t.Setenv("ZUT_HOME", home)
38	if err := os.WriteFile(ConfigPath(), []byte("{"), 0o600); err != nil {
39		t.Fatal(err)
40	}
41
42	err := runOrchestratedPrintMode(context.Background(), Args{
43		Mode:        ModePrint,
44		Orchestrate: true,
45		Provider:    "ollama",
46		Model:       "llama3",
47		Prompt:      "test",
48	}, "test")
49	if err == nil || !strings.Contains(err.Error(), "load config for orchestrated print") {
50		t.Fatalf("runOrchestratedPrintMode error = %v, want config load error", err)
51	}
52}