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)
	}
}