mirror of
https://github.com/luxfi/database.git
synced 2026-10-30 15:01:18 +00:00
zeekay
b933866a35
- Rename badgerdb package to zapdb - Update dependency from dgraph-io/badger/v4 to luxfi/zapdb/v4 v4.9.1 - Update all imports and references - zapdb maintains version parity with underlying BadgerDB (v4.9.x) - Prepares for ZAP protocol integration enhancements
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
//go:build test
|
|
|
|
package zapdb
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/luxfi/metric"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/luxfi/database"
|
|
"github.com/luxfi/database/dbtest"
|
|
)
|
|
|
|
func newDB(t testing.TB) database.Database {
|
|
folder := t.TempDir()
|
|
db, err := New(folder, nil, "test", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
return db
|
|
}
|
|
|
|
func TestInterface(t *testing.T) {
|
|
for name, test := range dbtest.Tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
db := newDB(t)
|
|
test(t, db)
|
|
_ = db.Close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func FuzzKeyValue(f *testing.F) {
|
|
db := newDB(f)
|
|
defer db.Close()
|
|
|
|
dbtest.FuzzKeyValue(f, db)
|
|
}
|
|
|
|
func FuzzNewIteratorWithPrefix(f *testing.F) {
|
|
db := newDB(f)
|
|
defer db.Close()
|
|
|
|
dbtest.FuzzNewIteratorWithPrefix(f, db)
|
|
}
|
|
|
|
func FuzzNewIteratorWithStartAndPrefix(f *testing.F) {
|
|
db := newDB(f)
|
|
defer db.Close()
|
|
|
|
dbtest.FuzzNewIteratorWithStartAndPrefix(f, db)
|
|
}
|
|
|
|
func BenchmarkInterface(b *testing.B) {
|
|
for _, size := range dbtest.BenchmarkSizes {
|
|
keys, values := dbtest.SetupBenchmark(b, size[0], size[1], size[2])
|
|
for name, bench := range dbtest.Benchmarks {
|
|
b.Run(fmt.Sprintf("badgerdb_%d_pairs_%d_keys_%d_values_%s", size[0], size[1], size[2], name), func(b *testing.B) {
|
|
db := newDB(b)
|
|
|
|
bench(b, db, keys, values)
|
|
|
|
// The database may have been closed by the test, so we don't care if it
|
|
// errors here.
|
|
_ = db.Close()
|
|
})
|
|
}
|
|
}
|
|
}
|