aboutsummaryrefslogtreecommitdiff
path: root/plugin/file/delete_test.go
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())
	}
}