plugin_test.go

package warnlist

import (
	"bytes"
	"context"
	"testing"

	"github.com/coredns/coredns/plugin/pkg/dnstest"
	"github.com/coredns/coredns/plugin/test"

	"github.com/miekg/dns"
)

func TestWarnlist(t *testing.T) {

	// Create a minimal warnlist for this test.
	wl := NewWarnlist()
	wl.Add("example.org.")
	wl.Add("totally.cool")
	wl.Close()

	// Create a new Warnlist Plugin. Use the test.ErrorHandler as the next plugin.
	m := WarnlistPlugin{Next: test.ErrorHandler(), warnlist: wl}

	// Setup a new output buffer that is *not* standard output, so we can check if
	// example is really being printed.
	b := &bytes.Buffer{}
	out = b

	ctx := context.TODO()
	r := new(dns.Msg)
	r.SetQuestion("example.org.", dns.TypeA)
	// Create a new Recorder that captures the result, this isn't actually used in this test
	// as it just serves as something that implements the dns.ResponseWriter interface.
	rec := dnstest.NewRecorder(&test.ResponseWriter{})

	// Call our plugin directly, and check the result.
	_, err := m.ServeDNS(ctx, rec, r)
	if err != nil {
		t.Fatalf("Error serving DNS: %v", err)
	}
	// if a := b.String(); a != "host 10.240.0.1 requested warnlisted domain: example.org.\n" { // TODO: Check log output instead of response
	// 	t.Errorf("Failed to print '%s', got %s", "example", a)
	// }
}