From e04828b9976748e594e880918e8109488eb2cb55 Mon Sep 17 00:00:00 2001
From: matt durham <mattdurham@ppog.org>
Date: Mon, 30 Oct 2023 15:59:23 -0400
Subject: [PATCH] tests

---
 service/kv/kv_bench_test.go | 95 +++++++++++++++++++++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100644 service/kv/kv_bench_test.go

diff --git a/service/kv/kv_bench_test.go b/service/kv/kv_bench_test.go
new file mode 100644
index 000000000000..498f9eed3ab3
--- /dev/null
+++ b/service/kv/kv_bench_test.go
@@ -0,0 +1,95 @@
+package kv
+
+import (
+	"github.com/stretchr/testify/require"
+	"math/rand"
+	"path/filepath"
+	"strconv"
+	"testing"
+)
+
+func BenchmarkSequentialPut(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := []byte("data")
+	for i := 0; i < b.N; i++ {
+		db.Put("test", strconv.Itoa(i), data)
+	}
+}
+
+func BenchmarkRandomPut(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := []byte("data")
+	for i := 0; i < b.N; i++ {
+		val := rand.Int()
+		db.Put("test", strconv.Itoa(val), data)
+	}
+}
+
+func BenchmarkSequentialPutLargePayload(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := make([]byte, 1024*1024)
+	for i := 0; i < b.N; i++ {
+		db.Put("test", strconv.Itoa(i), data)
+	}
+}
+
+func BenchmarkRandomPutLargePayload(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := make([]byte, 1024*1024)
+	for i := 0; i < b.N; i++ {
+		val := rand.Int()
+		db.Put("test", strconv.Itoa(val), data)
+	}
+}
+
+func BenchmarkBatchPut(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := []byte("data")
+	kvs := make(map[string][]byte)
+	for i := 0; i < 10_000; i++ {
+		kvs[strconv.Itoa(i)] = data
+	}
+	for i := 0; i < b.N; i++ {
+		db.PutRange("test", kvs)
+	}
+}
+
+func BenchmarkGet(b *testing.B) {
+	dir := b.TempDir()
+	db, err := NewKVDB(filepath.Join(dir, "bolt.db"))
+
+	require.NoError(b, err)
+	defer db.Close()
+	data := []byte("data")
+	kvs := make(map[string][]byte)
+	for i := 0; i < 10_000; i++ {
+		kvs[strconv.Itoa(i)] = data
+	}
+	db.PutRange("test", kvs)
+	for i := 0; i < b.N; i++ {
+		r := rand.Int31n(10_000)
+		val, found, _ := db.Get("test", strconv.Itoa(int(r)))
+		require.True(b, found)
+		require.True(b, string(val) == "data")
+	}
+}