Refactor and more unit tests.

1 files changed, 54 insertions, 0 deletions

diff --git a/app/cache_test.go b/app/cache_test.go
new file mode 100644
index 0000000..a44d2e1
--- /dev/null
+++ b/app/cache_test.go
@@ -0,0 +1,54 @@
+// +build all dep
+
+package main
+
+import (
+	"testing"
+)
+
+func TestCache(t *testing.T) {
+	h := []Host{}
+
+	err := CacheHosts("_test", h)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	resp, err := GetCache("_test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if resp != "[]" {
+		t.Fatal("Expected '[]' as result.")
+	}
+
+	if err = DelCache("_test"); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func BenchmarkCacheHosts(b *testing.B) {
+	b.StopTimer()
+	h := []Host{}
+	b.StartTimer()
+
+	for i := 0; i < b.N; i++ {
+		CacheHosts("_test", h)
+	}
+}
+
+func BenchmarkGetCache(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		GetCache("_test")
+	}
+}
+
+func BenchmarkDelCache(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		DelCache("_test")
+		b.StopTimer()
+		h := []Host{}
+		CacheHosts("_test", h)
+		b.StartTimer()
+	}
+}