diff --git a/internal/llm/prompt/prompt_test.go b/internal/llm/prompt/prompt_test.go index 77fe86a827749e0f7f0ef285e100c043b908bdea..2087ca149a372209e8cd8c8cdb56aaf8cbc4d68e 100644 --- a/internal/llm/prompt/prompt_test.go +++ b/internal/llm/prompt/prompt_test.go @@ -89,20 +89,19 @@ func TestProcessContextPaths(t *testing.T) { } // Test with tilde expansion (if we can create a file in home directory) - home, err := os.UserHomeDir() + tmpDir := t.TempDir() + t.Setenv("HOME", tmpDir) + homeTestFile := filepath.Join(tmpDir, "crush_test_file.txt") + err := os.WriteFile(homeTestFile, []byte(testContent), 0o644) if err == nil { - homeTestFile := filepath.Join(home, "crush_test_file.txt") - err = os.WriteFile(homeTestFile, []byte(testContent), 0o644) - if err == nil { - defer os.Remove(homeTestFile) // Clean up + defer os.Remove(homeTestFile) // Clean up - tildeFile := "~/crush_test_file.txt" - result = processContextPaths("", []string{tildeFile}) - expected = "# From:" + homeTestFile + "\n" + testContent + tildeFile := "~/crush_test_file.txt" + result = processContextPaths("", []string{tildeFile}) + expected = "# From:" + homeTestFile + "\n" + testContent - if result != expected { - t.Errorf("processContextPaths with tilde expansion failed.\nGot: %q\nWant: %q", result, expected) - } + if result != expected { + t.Errorf("processContextPaths with tilde expansion failed.\nGot: %q\nWant: %q", result, expected) } } }