137 lines
4.5 KiB
Go
137 lines
4.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)
|
|
}
|
|
}
|
|
|
|
func TestExplicitUninstallRejectsAgentOwnershipMismatch(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":"codex","agent_id":"agent-1"}`)
|
|
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
|
|
_, _, _, err := resolveAgentRuntimeForUninstall("opencode", "agent-1")
|
|
if err == nil || !strings.Contains(err.Error(), "belongs to") {
|
|
t.Fatalf("resolveAgentRuntimeForUninstall() error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestFindAgentIDByNameRejectsLinkedRuntime(t *testing.T) {
|
|
home := setMainTestHome(t)
|
|
root := filepath.Join(home, ".elydora")
|
|
if err := os.MkdirAll(root, 0700); err != nil {
|
|
t.Fatalf("MkdirAll() error = %v", err)
|
|
}
|
|
target := t.TempDir()
|
|
link := filepath.Join(root, "agent-1")
|
|
if err := os.Symlink(target, link); err != nil {
|
|
t.Skipf("directory symbolic links unavailable: %v", err)
|
|
}
|
|
|
|
_, err := findAgentIDByName("opencode")
|
|
if err == nil || !strings.Contains(err.Error(), "physical directory") {
|
|
t.Fatalf("findAgentIDByName() error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestFindAgentIDByNameRejectsLinkedConfig(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)
|
|
}
|
|
target := filepath.Join(t.TempDir(), "config.json")
|
|
config := []byte(`{"agent_name":"opencode","agent_id":"agent-1"}`)
|
|
if err := os.WriteFile(target, config, 0600); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
if err := os.Symlink(target, filepath.Join(runtimeDirectory, "config.json")); err != nil {
|
|
t.Skipf("file symbolic links unavailable: %v", err)
|
|
}
|
|
|
|
_, err := findAgentIDByName("opencode")
|
|
if err == nil || !strings.Contains(err.Error(), "physical file") {
|
|
t.Fatalf("findAgentIDByName() error = %v", err)
|
|
}
|
|
}
|