38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package split
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
golog "log"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/pkg/dnstest"
|
|
"github.com/coredns/coredns/plugin/test"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestExample(t *testing.T) {
|
|
// Create a new Split Plugin. Use the test.ErrorHandler as the next plugin.
|
|
x := Split{Next: test.ErrorHandler()}
|
|
|
|
// Setup a new output buffer that is *not* standard output, so we can check if
|
|
// example is really being printed.
|
|
b := &bytes.Buffer{}
|
|
golog.SetOutput(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.
|
|
x.ServeDNS(ctx, rec, r)
|
|
if a := b.String(); !strings.Contains(a, "[INFO] plugin/split: example") {
|
|
t.Errorf("Failed to print '%s', got %s", "[INFO] plugin/split: example", a)
|
|
}
|
|
}
|