package state

import (
	"os"
	"testing"
)

func TestKVStore(t *testing.T) {
	kv := KVStore{
		DbFileName: "test.db",
		BucketName: "Test",
	}
	kv.Init()
	defer kv.Close()

	err := kv.Set([]byte("foo"), []byte("bar"))
	if err != nil {
		t.Error("Counter not incremented")
	}

	value := kv.Get([]byte("foo"))
	if string(value) != "bar" {
		t.Errorf("Expected value at foo to be bar, go %v", value)
	}

	kv.Delete([]byte("foo"))
	value = kv.Get([]byte("foo"))
	if string(value) != "" {
		t.Errorf("Expected value at foo to be empty, got %v", value)
	}

	os.Remove("test.db")
}

package state

import (
	"os"
	"testing"
)

func TestKVStore(t *testing.T) {
	kv := KVStore{
		DbFileName: "test.db",
		BucketName: "Test",
	}
	kv.Init()
	defer kv.Close()

	err := kv.Set([]byte("foo"), []byte("bar"))
	if err != nil {
		t.Error("Counter not incremented")
	}

	value := kv.Get([]byte("foo"))
	if string(value) != "bar" {
		t.Errorf("Expected value at foo to be bar, go %v", value)
	}

	os.Remove("test.db")
}

--- 
+++ 
@@ -23,5 +23,11 @@
 		t.Errorf("Expected value at foo to be bar, go %v", value)
 	}
 
+	kv.Delete([]byte("foo"))
+	value = kv.Get([]byte("foo"))
+	if string(value) != "" {
+		t.Errorf("Expected value at foo to be empty, got %v", value)
+	}
+
 	os.Remove("test.db")
 }
Add test to ensure we actually delete a key
