From b7935b4ef8a834da5ec4514e96aeab65a66f537f Mon Sep 17 00:00:00 2001 From: Tai Groot Date: Thu, 10 Jul 2025 18:21:37 -0700 Subject: [PATCH] Update internal/llm/prompt/prompt_test.go Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --- internal/llm/prompt/prompt_test.go | 21 ++++++++++----------- 1 file changed, 10 insertions(+), 11 deletions(-) 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) } } }