From a7878f1d0d8eaebf15f87bc2df15f7a1088cce7f Mon Sep 17 00:00:00 2001
From: Jae Kwon <jae@tendermint.com>
Date: Sun, 17 Jan 2016 13:13:19 -0800
Subject: [PATCH] Add basic test for level_db

---
 README.md        |  1 +
 level_db_test.go | 83 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 84 insertions(+)
 create mode 100644 README.md
 create mode 100644 level_db_test.go

diff --git a/README.md b/README.md
new file mode 100644
index 000000000..ca5ab33f9
--- /dev/null
+++ b/README.md
@@ -0,0 +1 @@
+TODO: syndtr/goleveldb should be replaced with actual LevelDB instance
diff --git a/level_db_test.go b/level_db_test.go
new file mode 100644
index 000000000..2a8e9ac1e
--- /dev/null
+++ b/level_db_test.go
@@ -0,0 +1,83 @@
+package db
+
+import (
+	"bytes"
+	"encoding/binary"
+	"fmt"
+	"testing"
+
+	. "github.com/tendermint/go-common"
+)
+
+func BenchmarkRandomReadsWrites(b *testing.B) {
+	b.StopTimer()
+
+	numItems := int64(1000000)
+	internal := map[int64]int64{}
+	for i := 0; i < int(numItems); i++ {
+		internal[int64(i)] = int64(0)
+	}
+	db, err := NewLevelDB(Fmt("test_%x", RandStr(12)))
+	if err != nil {
+		b.Fatal(err.Error())
+		return
+	}
+
+	fmt.Println("ok, starting")
+	b.StartTimer()
+
+	for i := 0; i < b.N; i++ {
+		// Write something
+		{
+			idx := (int64(RandInt()) % numItems)
+			internal[idx] += 1
+			val := internal[idx]
+			idxBytes := int642Bytes(int64(idx))
+			valBytes := int642Bytes(int64(val))
+			//fmt.Printf("Set %X -> %X\n", idxBytes, valBytes)
+			db.Set(
+				idxBytes,
+				valBytes,
+			)
+		}
+		// Read something
+		{
+			idx := (int64(RandInt()) % numItems)
+			val := internal[idx]
+			idxBytes := int642Bytes(int64(idx))
+			valBytes := db.Get(idxBytes)
+			//fmt.Printf("Get %X -> %X\n", idxBytes, valBytes)
+			if val == 0 {
+				if !bytes.Equal(valBytes, nil) {
+					b.Errorf("Expected %X for %v, got %X",
+						nil, idx, valBytes)
+					break
+				}
+			} else {
+				if len(valBytes) != 8 {
+					b.Errorf("Expected length 8 for %v, got %X",
+						idx, valBytes)
+					break
+				}
+				valGot := bytes2Int64(valBytes)
+				if val != valGot {
+					b.Errorf("Expected %v for %v, got %v",
+						val, idx, valGot)
+					break
+				}
+			}
+		}
+	}
+
+	db.Close()
+}
+
+func int642Bytes(i int64) []byte {
+	buf := make([]byte, 8)
+	binary.BigEndian.PutUint64(buf, uint64(i))
+	return buf
+}
+
+func bytes2Int64(buf []byte) int64 {
+	return int64(binary.BigEndian.Uint64(buf))
+}