122 lines
3.0 KiB
Go
122 lines
3.0 KiB
Go
package codex
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func TestSkillDirs_UsesProjectAgentAndCodexHomes(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
home := filepath.Join(tmp, "home")
|
|
codexHome := filepath.Join(tmp, "codex-home")
|
|
repo := filepath.Join(tmp, "repo")
|
|
workDir := filepath.Join(repo, "nested", "pkg")
|
|
|
|
setTestHome(t, home)
|
|
t.Setenv("CODEX_HOME", "")
|
|
|
|
for _, dir := range []string{
|
|
filepath.Join(repo, "nested", "pkg"),
|
|
filepath.Join(repo, "nested"),
|
|
repo,
|
|
codexHome,
|
|
} {
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatalf("mkdir %s: %v", dir, err)
|
|
}
|
|
}
|
|
if err := os.WriteFile(filepath.Join(repo, ".git"), []byte("gitdir: fake\n"), 0o644); err != nil {
|
|
t.Fatalf("write .git: %v", err)
|
|
}
|
|
|
|
a := &Agent{workDir: workDir, codexHome: codexHome}
|
|
got := a.SkillDirs()
|
|
want := []string{
|
|
filepath.Join(workDir, ".agents", "skills"),
|
|
filepath.Join(workDir, ".codex", "skills"),
|
|
filepath.Join(repo, "nested", ".agents", "skills"),
|
|
filepath.Join(repo, "nested", ".codex", "skills"),
|
|
filepath.Join(repo, ".agents", "skills"),
|
|
filepath.Join(repo, ".codex", "skills"),
|
|
filepath.Join(codexHome, "skills"),
|
|
filepath.Join(home, ".agents", "skills"),
|
|
}
|
|
if len(got) != len(want) {
|
|
t.Fatalf("len(SkillDirs()) = %d, want %d\n got=%v", len(got), len(want), got)
|
|
}
|
|
for i := range want {
|
|
if got[i] != want[i] {
|
|
t.Fatalf("SkillDirs()[%d] = %q, want %q\nfull=%v", i, got[i], want[i], got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSkillDirs_FallsBackToEnvCodexHome(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
home := filepath.Join(tmp, "home")
|
|
codexHome := filepath.Join(tmp, "profile-home")
|
|
workDir := filepath.Join(tmp, "workspace")
|
|
if err := os.MkdirAll(workDir, 0o755); err != nil {
|
|
t.Fatalf("mkdir workdir: %v", err)
|
|
}
|
|
|
|
setTestHome(t, home)
|
|
t.Setenv("CODEX_HOME", codexHome)
|
|
|
|
a := &Agent{workDir: workDir}
|
|
got := a.SkillDirs()
|
|
found := false
|
|
for _, dir := range got {
|
|
if dir == filepath.Join(codexHome, "skills") {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatalf("SkillDirs() missing CODEX_HOME skills dir: %v", got)
|
|
}
|
|
}
|
|
|
|
func setTestHome(t *testing.T, home string) {
|
|
t.Helper()
|
|
t.Setenv("HOME", home)
|
|
if runtime.GOOS == "windows" {
|
|
t.Setenv("USERPROFILE", home)
|
|
t.Setenv("HOMEDRIVE", "")
|
|
t.Setenv("HOMEPATH", "")
|
|
}
|
|
}
|
|
|
|
// TestSkillDirs_RaceFreeAgainstSetWorkDir pins the bug where SkillDirs
|
|
// read a.workDir and a.codexHome without holding a.mu, while
|
|
// SetWorkDir writes a.workDir under the lock. Run with -race to detect
|
|
// the data race; with the production fix the test stays clean.
|
|
func TestSkillDirs_RaceFreeAgainstSetWorkDir(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
a := &Agent{workDir: tmp, codexHome: filepath.Join(tmp, "codex")}
|
|
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 30; i++ {
|
|
wg.Add(1)
|
|
go func(i int) {
|
|
defer wg.Done()
|
|
if i%2 == 0 {
|
|
a.SetWorkDir(filepath.Join(tmp, "a"))
|
|
} else {
|
|
a.SetWorkDir(filepath.Join(tmp, "b"))
|
|
}
|
|
}(i)
|
|
}
|
|
for i := 0; i < 30; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
_ = a.SkillDirs()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|