blob: 26ee64e3a4a95e05b01345599c248acd18477395 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
package file
import (
"bytes"
"fmt"
"testing"
"github.com/coredns/coredns/plugin/file/tree"
"github.com/coredns/coredns/plugin/test"
"github.com/miekg/dns"
)
/*
Create a zone with:
apex
/
a MX
a A
Test that: we create the proper tree and that delete
deletes the correct elements
*/
var tz = NewZone("example.org.", "db.example.org.")
type treebuf struct {
*bytes.Buffer
}
func (t *treebuf) printFunc(e *tree.Elem, rrs map[uint16][]dns.RR) error {
fmt.Fprintf(t.Buffer, "%v\n", rrs) // should be fixed order in new go versions.
return nil
}
func TestZoneInsertAndDelete(t *testing.T) {
tz.Insert(test.SOA("example.org. IN SOA 1 2 3 4 5"))
if x := tz.Apex.SOA.Header().Name; x != "example.org." {
t.Errorf("Failed to insert SOA, expected %s, git %s", "example.org.", x)
}
// Insert two RRs and then remove one.
tz.Insert(test.A("a.example.org. IN A 127.0.0.1"))
tz.Insert(test.MX("a.example.org. IN MX 10 mx.example.org."))
tz.Delete(test.MX("a.example.org. IN MX 10 mx.example.org."))
tb := treebuf{new(bytes.Buffer)}
tz.Walk(tb.printFunc)
if tb.String() != "map[1:[a.example.org.\t3600\tIN\tA\t127.0.0.1]]\n" {
t.Errorf("Expected 1 A record in tree, got %s", tb.String())
}
tz.Delete(test.A("a.example.org. IN A 127.0.0.1"))
tb.Reset()
tz.Walk(tb.printFunc)
if tb.String() != "" {
t.Errorf("Expected no record in tree, got %s", tb.String())
}
}
|
