81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func setMainTestHome(t *testing.T) string {
|
|
t.Helper()
|
|
home := t.TempDir()
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("USERPROFILE", home)
|
|
return home
|
|
}
|
|
|
|
func TestGuardScriptPathRejectsUnsafeAgentID(t *testing.T) {
|
|
setMainTestHome(t)
|
|
_, err := guardScriptPathForAgent("../escape")
|
|
if err == nil || !strings.Contains(err.Error(), "invalid agent ID") {
|
|
t.Fatalf("guardScriptPathForAgent() error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestFindAgentIDByNameReturnsValidatedDirectoryIdentity(t *testing.T) {
|
|
home := setMainTestHome(t)
|
|
runtimeDirectory := filepath.Join(home, ".elydora", "agent-1")
|
|
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
|
|
t.Fatalf("MkdirAll() error = %v", err)
|
|
}
|
|
config := []byte(`{"agent_name":"opencode","agent_id":"agent-1"}`)
|
|
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
|
|
agentID, err := findAgentIDByName("opencode")
|
|
if err != nil {
|
|
t.Fatalf("findAgentIDByName() error = %v", err)
|
|
}
|
|
if agentID != "agent-1" {
|
|
t.Fatalf("agent ID = %q, want agent-1", agentID)
|
|
}
|
|
}
|
|
|
|
func TestFindAgentIDByNameRejectsConfigDirectoryMismatch(t *testing.T) {
|
|
home := setMainTestHome(t)
|
|
runtimeDirectory := filepath.Join(home, ".elydora", "stored-directory")
|
|
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
|
|
t.Fatalf("MkdirAll() error = %v", err)
|
|
}
|
|
config := []byte(`{"agent_name":"opencode","agent_id":"different-agent"}`)
|
|
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
|
|
_, err := findAgentIDByName("opencode")
|
|
if err == nil || !strings.Contains(err.Error(), "crosses its runtime directory") {
|
|
t.Fatalf("findAgentIDByName() error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestFindAgentIDByNameRejectsAmbiguousRuntimes(t *testing.T) {
|
|
home := setMainTestHome(t)
|
|
for _, agentID := range []string{"agent-1", "agent-2"} {
|
|
runtimeDirectory := filepath.Join(home, ".elydora", agentID)
|
|
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
|
|
t.Fatalf("MkdirAll() error = %v", err)
|
|
}
|
|
config := []byte(`{"agent_name":"opencode","agent_id":"` + agentID + `"}`)
|
|
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
}
|
|
|
|
_, err := findAgentIDByName("opencode")
|
|
if err == nil || !strings.Contains(err.Error(), "multiple") {
|
|
t.Fatalf("findAgentIDByName() error = %v", err)
|
|
}
|
|
}
|