1 files changed, 58 insertions, 0 deletions
diff --git a/internal/recordsdir/recordsdir_test.go b/internal/recordsdir/recordsdir_test.go
new file mode 100644
index 0000000..bee8d67
--- /dev/null
+++ b/internal/recordsdir/recordsdir_test.go
@@ -0,0 +1,58 @@
+package recordsdir
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestHostFromFileName(t *testing.T) {
+	tests := []struct {
+		file string
+		want string
+	}{
+		{"earth.records", "earth"},
+		{"myhost.example.com.records", "myhost"},
+		{"single.records", "single"},
+	}
+	for _, tt := range tests {
+		t.Run(tt.file, func(t *testing.T) {
+			if got := HostFromFileName(tt.file); got != tt.want {
+				t.Fatalf("HostFromFileName(%q) = %q, want %q", tt.file, got, tt.want)
+			}
+		})
+	}
+}
+
+func TestListNonEmptyFiles(t *testing.T) {
+	dir := t.TempDir()
+	if err := os.WriteFile(filepath.Join(dir, "skip.txt"), []byte("x"), 0o644); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(dir, "empty.records"), nil, 0o644); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(dir, "h1.records"), []byte("line\n"), 0o644); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.Mkdir(filepath.Join(dir, "nested.records"), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	entries, err := ListNonEmptyFiles(dir)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(entries) != 1 {
+		t.Fatalf("len(entries) = %d, want 1: %#v", len(entries), entries)
+	}
+	if entries[0].Host != "h1" || filepath.Base(entries[0].Path) != "h1.records" {
+		t.Fatalf("unexpected entry: %#v", entries[0])
+	}
+}
+
+func TestListNonEmptyFiles_ReadError(t *testing.T) {
+	_, err := ListNonEmptyFiles(filepath.Join(t.TempDir(), "nonexistent-subdir-nope"))
+	if err == nil {
+		t.Fatal("expected error")
+	}
+}