tests: clean up output

Some document improvements and add a few more tests.

1 files changed, 29 insertions, 0 deletions

diff --git a/middleware/pkg/rcode/rcode_test.go b/middleware/pkg/rcode/rcode_test.go
new file mode 100644
index 000000000..bfca32f1d
--- /dev/null
+++ b/middleware/pkg/rcode/rcode_test.go
@@ -0,0 +1,29 @@
+package rcode
+
+import (
+	"testing"
+
+	"github.com/miekg/dns"
+)
+
+func TestToString(t *testing.T) {
+	tests := []struct {
+		in       int
+		expected string
+	}{
+		{
+			dns.RcodeSuccess,
+			"NOERROR",
+		},
+		{
+			28,
+			"RCODE28",
+		},
+	}
+	for i, test := range tests {
+		got := ToString(test.in)
+		if got != test.expected {
+			t.Errorf("Test %d, expected %s, got %s", i, test.expected, got)
+		}
+	}
+}