From 76d5d8493f853da5e371e761e5c31515883d2ed9 Mon Sep 17 00:00:00 2001
From: Miek Gieben <miek@miek.nl>
Date: Mon, 15 Oct 2012 08:12:49 +0200
Subject: [PATCH] Add zone test from Alex Polvi

---
 zone_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/zone_test.go b/zone_test.go
index 7f15a6a5..abd247e4 100644
--- a/zone_test.go
+++ b/zone_test.go
@@ -21,6 +21,26 @@ func TestRadixName(t *testing.T) {
 }
 
 func TestInsert(t *testing.T) {
+	z := NewZone("miek.nl.")
+	mx, _ := NewRR("foo.miek.nl. MX 10 mx.miek.nl.")
+	z.Insert(mx)
+	zd, exact := z.Find("foo.miek.nl.")
+	if exact != true {
+		t.Fail() // insert broken?
+	}
 }
+
 func TestRemove(t *testing.T) {
+	z := NewZone("miek.nl.")
+	mx, _ := NewRR("foo.miek.nl. MX 10 mx.miek.nl.")
+	z.Insert(mx)
+	zd, exact := z.Find("foo.miek.nl.")
+	if exact != true {
+		t.Fail() // insert broken?
+	}
+	z.Remove(mx)
+	zd, exact = z.Find("foo.miek.nl.")
+	if exact != false {
+		t.Errorf("zd(%s) exact(%s) still exists", zd, exact) // it should no longer be in the zone
+	}
 }