package behavior

import (
	"os"
	"path/filepath"
	"testing"
)

// Regression: WalkYAMLDir only matched .yaml, so a .yml trigger file loaded
// and fired at runtime while every validation pass ignored it.
func TestWalkYAMLDirIncludesYml(t *testing.T) {
	dir := t.TempDir()
	if err := os.WriteFile(filepath.Join(dir, "a.yaml"), []byte("x: 1"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "b.yml"), []byte("x: 2"), 0644); err != nil {
		t.Fatal(err)
	}
	ids := map[string]bool{}
	if err := WalkYAMLDir(dir, func(path, id string, data []byte) error {
		ids[id] = true
		return nil
	}); err != nil {
		t.Fatal(err)
	}
	if !ids["a"] || !ids["b"] {
		t.Fatalf("WalkYAMLDir ids = %v, want both a and b", ids)
	}
}

func TestBuildPathIndexIncludesYml(t *testing.T) {
	dir := t.TempDir()
	if err := os.WriteFile(filepath.Join(dir, "a.yaml"), []byte("x: 1"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "b.yml"), []byte("x: 2"), 0644); err != nil {
		t.Fatal(err)
	}
	idx := BuildPathIndex(dir)
	if idx["a"] == "" || idx["b"] == "" {
		t.Fatalf("BuildPathIndex = %v, want both a and b", idx)
	}
}

func TestBuildRoomIndexIncludesYml(t *testing.T) {
	dir := t.TempDir()
	if err := os.WriteFile(filepath.Join(dir, "1001.yaml"), []byte("name: a"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "1002.yml"), []byte("name: b"), 0644); err != nil {
		t.Fatal(err)
	}
	idx := BuildRoomIndex(dir)
	if idx[1001] == "" || idx[1002] == "" {
		t.Fatalf("BuildRoomIndex = %v, want both 1001 and 1002", idx)
	}
}

func TestLoadDropTableRejectsPathEscape(t *testing.T) {
	dataDir := t.TempDir()
	if _, err := LoadDropTable(dataDir, "../../../etc/passwd"); err == nil {
		t.Fatal("accepted a path-escaping drop table id")
	}
}