aboutsummaryrefslogtreecommitdiff
path: root/plugin/rewrite/name_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'plugin/rewrite/name_test.go')
-rw-r--r--plugin/rewrite/name_test.go28
1 files changed, 28 insertions, 0 deletions
diff --git a/plugin/rewrite/name_test.go b/plugin/rewrite/name_test.go
index fe2a2eb06..f2245f448 100644
--- a/plugin/rewrite/name_test.go
+++ b/plugin/rewrite/name_test.go
@@ -32,6 +32,34 @@ func TestRewriteIllegalName(t *testing.T) {
}
}
+func TestRewriteNamePrefix(t *testing.T) {
+ r, err := newNameRule("stop", "prefix", "test", "not-a-test")
+ if err != nil {
+ t.Fatalf("Expected no error, got %s", err)
+ }
+
+ rw := Rewrite{
+ Next: plugin.HandlerFunc(msgPrinter),
+ Rules: []Rule{r},
+ noRevert: true,
+ }
+
+ ctx := context.TODO()
+ m := new(dns.Msg)
+ m.SetQuestion("test.example.org.", dns.TypeA)
+
+ rec := dnstest.NewRecorder(&test.ResponseWriter{})
+ _, err = rw.ServeDNS(ctx, rec, m)
+ if err != nil {
+ t.Fatalf("Expected no error, got %s", err)
+ }
+ expected := "not-a-test.example.org."
+ actual := rec.Msg.Question[0].Name
+ if actual != expected {
+ t.Fatalf("Expected rewrite to %v, got %v", expected, actual)
+ }
+}
+
func TestNewNameRule(t *testing.T) {
tests := []struct {
next string