From a832ab696a67f18ae61744b7c4863690d4cd61d0 Mon Sep 17 00:00:00 2001
From: Miek Gieben <miek@miek.nl>
Date: Fri, 25 Mar 2016 20:30:38 +0000
Subject: [PATCH] bugfix: actually *use* the CH class

---
 middleware/chaos/chaos.go      | 2 +-
 middleware/chaos/chaos_test.go | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/middleware/chaos/chaos.go b/middleware/chaos/chaos.go
index a7a63c1a1..0064e4f94 100644
--- a/middleware/chaos/chaos.go
+++ b/middleware/chaos/chaos.go
@@ -17,7 +17,7 @@ type Chaos struct {
 
 func (c Chaos) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
 	state := middleware.State{W: w, Req: r}
-	if state.QClass() != dns.ClassINET || state.QType() != dns.TypeTXT {
+	if state.QClass() != dns.ClassCHAOS || state.QType() != dns.TypeTXT {
 		return c.Next.ServeDNS(ctx, w, r)
 	}
 	m := new(dns.Msg)
diff --git a/middleware/chaos/chaos_test.go b/middleware/chaos/chaos_test.go
index 4d01a456e..e8cdef5f1 100644
--- a/middleware/chaos/chaos_test.go
+++ b/middleware/chaos/chaos_test.go
@@ -54,6 +54,7 @@ func TestChaos(t *testing.T) {
 			test.qtype = dns.TypeTXT
 		}
 		req.SetQuestion(dns.Fqdn(test.qname), test.qtype)
+		req.Question[0].Qclass = dns.ClassCHAOS
 		em.Next = test.next
 
 		rec := middleware.NewResponseRecorder(&middleware.TestResponseWriter{})