mirror of
https://github.com/coredns/coredns.git
synced 2025-09-14 02:01:05 +02:00
c2780f42c4
Exporting Zone.File to avoid getters and setters Updating getter and setter for Zone.File to be less racy Renaming GetFile to File in zone plugin
102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package auto
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"regexp"
|
|
"testing"
|
|
)
|
|
|
|
func TestWatcher(t *testing.T) {
|
|
tempdir, err := createFiles()
|
|
if err != nil {
|
|
if tempdir != "" {
|
|
os.RemoveAll(tempdir)
|
|
}
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tempdir)
|
|
|
|
ldr := loader{
|
|
directory: tempdir,
|
|
re: regexp.MustCompile(`db\.(.*)`),
|
|
template: `${1}`,
|
|
}
|
|
|
|
a := Auto{
|
|
loader: ldr,
|
|
Zones: &Zones{},
|
|
}
|
|
|
|
a.Walk()
|
|
|
|
// example.org and example.com should exist
|
|
if x := len(a.Zones.Z["example.org."].All()); x != 4 {
|
|
t.Fatalf("Expected 4 RRs, got %d", x)
|
|
}
|
|
if x := len(a.Zones.Z["example.com."].All()); x != 4 {
|
|
t.Fatalf("Expected 4 RRs, got %d", x)
|
|
}
|
|
|
|
// Now remove one file, rescan and see if it's gone.
|
|
if err := os.Remove(path.Join(tempdir, "db.example.com")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
a.Walk()
|
|
|
|
if _, ok := a.Zones.Z["example.com."]; ok {
|
|
t.Errorf("Expected %q to be gone.", "example.com.")
|
|
}
|
|
if _, ok := a.Zones.Z["example.org."]; !ok {
|
|
t.Errorf("Expected %q to still be there.", "example.org.")
|
|
}
|
|
|
|
}
|
|
|
|
func TestSymlinks(t *testing.T) {
|
|
tempdir, err := createFiles()
|
|
if err != nil {
|
|
if tempdir != "" {
|
|
os.RemoveAll(tempdir)
|
|
}
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tempdir)
|
|
|
|
ldr := loader{
|
|
directory: tempdir,
|
|
re: regexp.MustCompile(`db\.(.*)`),
|
|
template: `${1}`,
|
|
}
|
|
|
|
a := Auto{
|
|
loader: ldr,
|
|
Zones: &Zones{},
|
|
}
|
|
|
|
a.Walk()
|
|
|
|
// Now create a duplicate file in a subdirectory and repoint the symlink
|
|
if err := os.Remove(path.Join(tempdir, "db.example.com")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dataDir := path.Join(tempdir, "..data")
|
|
if err = os.Mkdir(dataDir, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
newFile := path.Join(dataDir, "db.example.com")
|
|
if err = os.Symlink(path.Join(tempdir, "db.example.org"), newFile); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
a.Walk()
|
|
|
|
if storedZone, ok := a.Zones.Z["example.com."]; ok {
|
|
storedFile := storedZone.File()
|
|
if storedFile != newFile {
|
|
t.Errorf("Expected %q to reflect new path %q", storedFile, newFile)
|
|
}
|
|
}
|
|
}
|