Skip to content

Commit c807c8e

Browse files
committed
fix: check error returns in agent test file operations
1 parent 5890fc2 commit c807c8e

2 files changed

Lines changed: 28 additions & 10 deletions

File tree

agents/agents_test.go

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -90,19 +90,31 @@ func TestListAll_FromDir(t *testing.T) {
9090
dir := t.TempDir()
9191

9292
// Write some agent files
93-
os.WriteFile(filepath.Join(dir, "worker.md"), []byte("---\nname: worker\n---\nDo work."), 0o644)
94-
os.WriteFile(filepath.Join(dir, "reviewer.md"), []byte("---\nname: reviewer\n---\nReview code."), 0o644)
95-
os.WriteFile(filepath.Join(dir, "not-md.txt"), []byte("ignored"), 0o644)
93+
if err := os.WriteFile(filepath.Join(dir, "worker.md"), []byte("---\nname: worker\n---\nDo work."), 0o644); err != nil {
94+
t.Fatal(err)
95+
}
96+
if err := os.WriteFile(filepath.Join(dir, "reviewer.md"), []byte("---\nname: reviewer\n---\nReview code."), 0o644); err != nil {
97+
t.Fatal(err)
98+
}
99+
if err := os.WriteFile(filepath.Join(dir, "not-md.txt"), []byte("ignored"), 0o644); err != nil {
100+
t.Fatal(err)
101+
}
96102

97103
// Override the agent dirs for testing
98104
origHome := os.Getenv("HOME")
99-
os.Setenv("HOME", dir)
100-
defer os.Setenv("HOME", origHome)
105+
if err := os.Setenv("HOME", dir); err != nil {
106+
t.Fatal(err)
107+
}
108+
defer os.Setenv("HOME", origHome) //nolint:errcheck
101109

102110
// Create ~/.hawk/agents
103111
agentDir := filepath.Join(dir, ".hawk", "agents")
104-
os.MkdirAll(agentDir, 0o755)
105-
os.WriteFile(filepath.Join(agentDir, "test.md"), []byte("---\nname: test\n---\nTest prompt."), 0o644)
112+
if err := os.MkdirAll(agentDir, 0o755); err != nil {
113+
t.Fatal(err)
114+
}
115+
if err := os.WriteFile(filepath.Join(agentDir, "test.md"), []byte("---\nname: test\n---\nTest prompt."), 0o644); err != nil {
116+
t.Fatal(err)
117+
}
106118

107119
agents, err := ListAll()
108120
if err != nil {

agents/persona_test.go

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -520,9 +520,15 @@ temperature: 0.7
520520
---
521521
You are persona two.
522522
`
523-
os.WriteFile(filepath.Join(dir, "persona-one.md"), []byte(file1), 0o644)
524-
os.WriteFile(filepath.Join(dir, "persona-two.md"), []byte(file2), 0o644)
525-
os.WriteFile(filepath.Join(dir, "not-a-persona.txt"), []byte("ignored"), 0o644)
523+
if err := os.WriteFile(filepath.Join(dir, "persona-one.md"), []byte(file1), 0o644); err != nil {
524+
t.Fatal(err)
525+
}
526+
if err := os.WriteFile(filepath.Join(dir, "persona-two.md"), []byte(file2), 0o644); err != nil {
527+
t.Fatal(err)
528+
}
529+
if err := os.WriteFile(filepath.Join(dir, "not-a-persona.txt"), []byte("ignored"), 0o644); err != nil {
530+
t.Fatal(err)
531+
}
526532

527533
r := NewPersonaRegistry(dir)
528534
if err := r.LoadAll(); err != nil {

0 commit comments

Comments
 (0)