git.iko.soy/crush
on main help sign in
Update internal/llm/prompt/prompt_test.go
b7935b4Tai Groot
internal/llm/prompt/prompt_test.go+10−11
compare with main
unifiedsplit
@@ -89,20 +89,19 @@ func TestProcessContextPaths(t *testing.T) {
89 }
89 }
90
90
91 // Test with tilde expansion (if we can create a file in home directory)
91 // Test with tilde expansion (if we can create a file in home directory)
92− home, err := os.UserHomeDir()
92+ tmpDir := t.TempDir()
93+ t.Setenv("HOME", tmpDir)
94+ homeTestFile := filepath.Join(tmpDir, "crush_test_file.txt")
95+ err := os.WriteFile(homeTestFile, []byte(testContent), 0o644)
93 if err == nil {
96 if err == nil {
94− homeTestFile := filepath.Join(home, "crush_test_file.txt")
95− err = os.WriteFile(homeTestFile, []byte(testContent), 0o644)
96− if err == nil {
97− defer os.Remove(homeTestFile) // Clean up
97+ defer os.Remove(homeTestFile) // Clean up
98
98
99− tildeFile := "~/crush_test_file.txt"
99+ tildeFile := "~/crush_test_file.txt"
100− result = processContextPaths("", []string{tildeFile})
100+ result = processContextPaths("", []string{tildeFile})
101− expected = "# From:" + homeTestFile + "\n" + testContent
101+ expected = "# From:" + homeTestFile + "\n" + testContent
102
102
103− if result != expected {
103+ if result != expected {
104− t.Errorf("processContextPaths with tilde expansion failed.\nGot: %q\nWant: %q", result, expected)
104+ t.Errorf("processContextPaths with tilde expansion failed.\nGot: %q\nWant: %q", result, expected)
105− }
106 }
105 }
107 }
106 }
108}
107}