Move nonwriter to mw/pkg/nonwriter (#948)

Make it its own package as shared between autopath and federation.

Fixes #933
This commit is contained in:
Miek Gieben
2017-08-19 17:28:42 +01:00
committed by GitHub
parent 219a899772
commit 87eb0d39a4
6 changed files with 46 additions and 46 deletions

View File

@@ -0,0 +1,23 @@
// Package nonwriter implements a dns.ResponseWriter that never writes, but captures the dns.Msg being written.
package nonwriter
import (
"github.com/miekg/dns"
)
// Writer is a type of ResponseWriter that captures the message, but never writes to the client.
type Writer struct {
dns.ResponseWriter
Msg *dns.Msg
}
// New makes and returns a new NonWriter.
func New(w dns.ResponseWriter) *Writer { return &Writer{ResponseWriter: w} }
// WriteMsg records the message, but doesn't write it itself.
func (w *Writer) WriteMsg(res *dns.Msg) error {
w.Msg = res
return nil
}
func (w *Writer) Write(buf []byte) (int, error) { return len(buf), nil }

View File

@@ -0,0 +1,19 @@
package nonwriter
import (
"testing"
"github.com/miekg/dns"
)
func TestNonWriter(t *testing.T) {
nw := New(nil)
m := new(dns.Msg)
m.SetQuestion("example.org.", dns.TypeA)
if err := nw.WriteMsg(m); err != nil {
t.Errorf("Got error when writing to nonwriter: %s", err)
}
if x := nw.Msg.Question[0].Name; x != "example.org." {
t.Errorf("Expacted 'example.org.' got %q:", x)
}
}