1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
package executor
import (
"context"
"io"
"log/slog"
"strings"
"testing"
"github.com/thepeterstone/claudomator/internal/storage"
"github.com/thepeterstone/claudomator/internal/task"
)
func TestGeminiRunner_BuildArgs_BasicTask(t *testing.T) {
r := &GeminiRunner{}
tk := &task.Task{
Agent: task.AgentConfig{
Type: "gemini",
Instructions: "fix the bug",
Model: "gemini-2.0-flash",
SkipPlanning: true,
},
}
args := r.buildArgs(tk, &storage.Execution{ID: "test-exec"}, "/tmp/q.json")
// Gemini CLI: instructions is the first positional arg
if len(args) < 1 || args[0] != "fix the bug" {
t.Errorf("expected instructions as first arg, got: %v", args)
}
argMap := make(map[string]bool)
for _, a := range args {
argMap[a] = true
}
for _, want := range []string{"--output-format", "stream-json", "--model", "gemini-2.0-flash"} {
if !argMap[want] {
t.Errorf("missing arg %q in %v", want, args)
}
}
}
func TestGeminiRunner_BuildArgs_PreamblePrepended(t *testing.T) {
r := &GeminiRunner{}
tk := &task.Task{
Agent: task.AgentConfig{
Type: "gemini",
Instructions: "fix the bug",
SkipPlanning: false,
},
}
args := r.buildArgs(tk, &storage.Execution{ID: "test-exec"}, "/tmp/q.json")
if len(args) < 1 {
t.Fatalf("expected at least 1 arg, got: %v", args)
}
if !strings.HasPrefix(args[0], planningPreamble) {
t.Errorf("instructions should start with planning preamble")
}
if !strings.HasSuffix(args[0], "fix the bug") {
t.Errorf("instructions should end with original instructions")
}
}
func TestGeminiRunner_Run_InaccessibleWorkingDir_ReturnsError(t *testing.T) {
r := &GeminiRunner{
BinaryPath: "true", // would succeed if it ran
Logger: slog.New(slog.NewTextHandler(io.Discard, nil)),
LogDir: t.TempDir(),
}
tk := &task.Task{
Agent: task.AgentConfig{
Type: "gemini",
WorkingDir: "/nonexistent/path/does/not/exist",
SkipPlanning: true,
},
}
exec := &storage.Execution{ID: "test-exec"}
err := r.Run(context.Background(), tk, exec)
if err == nil {
t.Fatal("expected error for inaccessible working_dir, got nil")
}
if !strings.Contains(err.Error(), "working_dir") {
t.Errorf("expected 'working_dir' in error, got: %v", err)
}
}
func TestGeminiRunner_BinaryPath_Default(t *testing.T) {
r := &GeminiRunner{}
if r.binaryPath() != "gemini" {
t.Errorf("want 'gemini', got %q", r.binaryPath())
}
}
func TestGeminiRunner_BinaryPath_Custom(t *testing.T) {
r := &GeminiRunner{BinaryPath: "/usr/local/bin/gemini"}
if r.binaryPath() != "/usr/local/bin/gemini" {
t.Errorf("want custom path, got %q", r.binaryPath())
}
}
|
