diff options
Diffstat (limited to 'middleware/erratic/erratic_test.go')
-rw-r--r-- | middleware/erratic/erratic_test.go | 79 |
1 files changed, 0 insertions, 79 deletions
diff --git a/middleware/erratic/erratic_test.go b/middleware/erratic/erratic_test.go deleted file mode 100644 index 4b54e0c12..000000000 --- a/middleware/erratic/erratic_test.go +++ /dev/null @@ -1,79 +0,0 @@ -package erratic - -import ( - "testing" - - "github.com/coredns/coredns/middleware/pkg/dnsrecorder" - "github.com/coredns/coredns/middleware/test" - - "github.com/miekg/dns" - "golang.org/x/net/context" -) - -func TestErraticDrop(t *testing.T) { - e := &Erratic{drop: 2} // 50% drops - - tests := []struct { - expectedCode int - expectedErr error - drop bool - }{ - {expectedCode: dns.RcodeSuccess, expectedErr: nil, drop: true}, - {expectedCode: dns.RcodeSuccess, expectedErr: nil, drop: false}, - } - - ctx := context.TODO() - - for i, tc := range tests { - req := new(dns.Msg) - req.SetQuestion("example.org.", dns.TypeA) - - rec := dnsrecorder.New(&test.ResponseWriter{}) - code, err := e.ServeDNS(ctx, rec, req) - - if err != tc.expectedErr { - t.Errorf("Test %d: Expected error %q, but got %q", i, tc.expectedErr, err) - } - if code != int(tc.expectedCode) { - t.Errorf("Test %d: Expected status code %d, but got %d", i, tc.expectedCode, code) - } - - if tc.drop && rec.Msg != nil { - t.Errorf("Test %d: Expected dropped message, but got %q", i, rec.Msg.Question[0].Name) - } - } -} - -func TestErraticTruncate(t *testing.T) { - e := &Erratic{truncate: 2} // 50% drops - - tests := []struct { - expectedCode int - expectedErr error - truncate bool - }{ - {expectedCode: dns.RcodeSuccess, expectedErr: nil, truncate: true}, - {expectedCode: dns.RcodeSuccess, expectedErr: nil, truncate: false}, - } - - ctx := context.TODO() - - for i, tc := range tests { - req := new(dns.Msg) - req.SetQuestion("example.org.", dns.TypeA) - - rec := dnsrecorder.New(&test.ResponseWriter{}) - code, err := e.ServeDNS(ctx, rec, req) - - if err != tc.expectedErr { - t.Errorf("Test %d: Expected error %q, but got %q", i, tc.expectedErr, err) - } - if code != int(tc.expectedCode) { - t.Errorf("Test %d: Expected status code %d, but got %d", i, tc.expectedCode, code) - } - - if tc.truncate && !rec.Msg.Truncated { - t.Errorf("Test %d: Expected truncated message, but got %q", i, rec.Msg.Question[0].Name) - } - } -} |