gecko/database/test_database.go

552 lines
17 KiB
Go

// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package database
import (
"bytes"
"testing"
)
var (
// Tests is a list of all database tests
Tests = []func(t *testing.T, db Database){
TestSimpleKeyValue,
TestSimpleKeyValueClosed,
TestBatchPut,
TestBatchDelete,
TestBatchReset,
TestBatchReplay,
TestIterator,
TestIteratorStart,
TestIteratorPrefix,
TestIteratorStartPrefix,
TestIteratorClosed,
TestStatNoPanic,
TestCompactNoPanic,
}
)
// TestSimpleKeyValue ...
func TestSimpleKeyValue(t *testing.T, db Database) {
key := []byte("hello")
value := []byte("world")
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if has {
t.Fatalf("db.Has unexpectedly returned true on key %s", key)
} else if v, err := db.Get(key); err != ErrNotFound {
t.Fatalf("Expected %s on db.Get for missing key %s. Returned 0x%x", ErrNotFound, key, v)
} else if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
if err := db.Put(key, value); err != nil {
t.Fatalf("Unexpected error on db.Put: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if !has {
t.Fatalf("db.Has unexpectedly returned false on key %s", key)
} else if v, err := db.Get(key); err != nil {
t.Fatalf("Unexpected error on db.Get: %s", err)
} else if !bytes.Equal(value, v) {
t.Fatalf("db.Get: Returned: 0x%x ; Expected: 0x%x", v, value)
}
if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if has {
t.Fatalf("db.Has unexpectedly returned true on key %s", key)
} else if v, err := db.Get(key); err != ErrNotFound {
t.Fatalf("Expected %s on db.Get for missing key %s. Returned 0x%x", ErrNotFound, key, v)
} else if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
}
// TestSimpleKeyValueClosed ...
func TestSimpleKeyValueClosed(t *testing.T, db Database) {
key := []byte("hello")
value := []byte("world")
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if has {
t.Fatalf("db.Has unexpectedly returned true on key %s", key)
} else if v, err := db.Get(key); err != ErrNotFound {
t.Fatalf("Expected %s on db.Get for missing key %s. Returned 0x%x", ErrNotFound, key, v)
} else if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
if err := db.Put(key, value); err != nil {
t.Fatalf("Unexpected error on db.Put: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if !has {
t.Fatalf("db.Has unexpectedly returned false on key %s", key)
} else if v, err := db.Get(key); err != nil {
t.Fatalf("Unexpected error on db.Get: %s", err)
} else if !bytes.Equal(value, v) {
t.Fatalf("db.Get: Returned: 0x%x ; Expected: 0x%x", v, value)
}
if err := db.Close(); err != nil {
t.Fatalf("Unexpected error on db.Close: %s", err)
}
if _, err := db.Has(key); err != ErrClosed {
t.Fatalf("Expected %s on db.Has after close", ErrClosed)
} else if _, err := db.Get(key); err != ErrClosed {
t.Fatalf("Expected %s on db.Get after close", ErrClosed)
} else if err := db.Put(key, value); err != ErrClosed {
t.Fatalf("Expected %s on db.Put after close", ErrClosed)
} else if err := db.Delete(key); err != ErrClosed {
t.Fatalf("Expected %s on db.Delete after close", ErrClosed)
} else if err := db.Close(); err != ErrClosed {
t.Fatalf("Expected %s on db.Close after close", ErrClosed)
}
}
// TestBatchPut ...
func TestBatchPut(t *testing.T, db Database) {
key := []byte("hello")
value := []byte("world")
batch := db.NewBatch()
if batch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Put(key, value); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if size := batch.ValueSize(); size <= 0 {
t.Fatalf("batch.ValueSize: Returned: %d ; Expected: > 0", size)
}
if err := batch.Write(); err != nil {
t.Fatalf("Unexpected error on batch.Write: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if !has {
t.Fatalf("db.Has unexpectedly returned false on key %s", key)
} else if v, err := db.Get(key); err != nil {
t.Fatalf("Unexpected error on db.Get: %s", err)
} else if !bytes.Equal(value, v) {
t.Fatalf("db.Get: Returned: 0x%x ; Expected: 0x%x", v, value)
}
if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
batch = db.NewBatch()
if batch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Put(key, value); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
db.Close()
if err := batch.Write(); err != ErrClosed {
t.Fatalf("Expected %s on batch.Write", ErrClosed)
}
}
// TestBatchDelete ...
func TestBatchDelete(t *testing.T, db Database) {
key := []byte("hello")
value := []byte("world")
if err := db.Put(key, value); err != nil {
t.Fatalf("Unexpected error on db.Put: %s", err)
}
batch := db.NewBatch()
if batch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Delete(key); err != nil {
t.Fatalf("Unexpected error on batch.Delete: %s", err)
}
if err := batch.Write(); err != nil {
t.Fatalf("Unexpected error on batch.Write: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if has {
t.Fatalf("db.Has unexpectedly returned true on key %s", key)
} else if v, err := db.Get(key); err != ErrNotFound {
t.Fatalf("Expected %s on db.Get for missing key %s. Returned 0x%x", ErrNotFound, key, v)
} else if err := db.Delete(key); err != nil {
t.Fatalf("Unexpected error on db.Delete: %s", err)
}
}
// TestBatchReset ...
func TestBatchReset(t *testing.T, db Database) {
key := []byte("hello")
value := []byte("world")
if err := db.Put(key, value); err != nil {
t.Fatalf("Unexpected error on db.Put: %s", err)
}
batch := db.NewBatch()
if batch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Delete(key); err != nil {
t.Fatalf("Unexpected error on batch.Delete: %s", err)
}
batch.Reset()
if err := batch.Write(); err != nil {
t.Fatalf("Unexpected error on batch.Write: %s", err)
}
if has, err := db.Has(key); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if !has {
t.Fatalf("db.Has unexpectedly returned false on key %s", key)
} else if v, err := db.Get(key); err != nil {
t.Fatalf("Unexpected error on db.Get: %s", err)
} else if !bytes.Equal(value, v) {
t.Fatalf("db.Get: Returned: 0x%x ; Expected: 0x%x", v, value)
}
}
// TestBatchReplay ...
func TestBatchReplay(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("hello2")
value2 := []byte("world2")
batch := db.NewBatch()
if batch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := batch.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
secondBatch := db.NewBatch()
if secondBatch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := batch.Replay(secondBatch); err != nil {
t.Fatalf("Unexpected error on batch.Replay: %s", err)
}
if err := secondBatch.Write(); err != nil {
t.Fatalf("Unexpected error on batch.Write: %s", err)
}
if has, err := db.Has(key1); err != nil {
t.Fatalf("Unexpected error on db.Has: %s", err)
} else if !has {
t.Fatalf("db.Has unexpectedly returned false on key %s", key1)
} else if v, err := db.Get(key1); err != nil {
t.Fatalf("Unexpected error on db.Get: %s", err)
} else if !bytes.Equal(value1, v) {
t.Fatalf("db.Get: Returned: 0x%x ; Expected: 0x%x", v, value1)
}
thirdBatch := db.NewBatch()
if thirdBatch == nil {
t.Fatalf("db.NewBatch returned nil")
}
if err := thirdBatch.Delete(key1); err != nil {
t.Fatalf("Unexpected error on batch.Delete: %s", err)
} else if err := thirdBatch.Delete(key2); err != nil {
t.Fatalf("Unexpected error on batch.Delete: %s", err)
}
if err := db.Close(); err != nil {
t.Fatalf("Unexpected error on db.Close: %s", err)
}
if err := batch.Replay(db); err != ErrClosed {
t.Fatalf("Expected %s on batch.Replay", ErrClosed)
} else if err := thirdBatch.Replay(db); err != ErrClosed {
t.Fatalf("Expected %s on batch.Replay", ErrClosed)
}
}
// TestIterator ...
func TestIterator(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("hello2")
value2 := []byte("world2")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
iterator := db.NewIterator()
if iterator == nil {
t.Fatalf("db.NewIterator returned nil")
}
defer iterator.Release()
if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key1) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key1)
} else if value := iterator.Value(); !bytes.Equal(value, value1) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value1)
} else if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key2) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key2)
} else if value := iterator.Value(); !bytes.Equal(value, value2) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value2)
} else if iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", true, false)
} else if key := iterator.Key(); key != nil {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: nil", key)
} else if value := iterator.Value(); value != nil {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: nil", value)
} else if err := iterator.Error(); err != nil {
t.Fatalf("iterator.Error Returned: %s ; Expected: nil", err)
}
}
// TestIteratorStart ...
func TestIteratorStart(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("hello2")
value2 := []byte("world2")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
iterator := db.NewIteratorWithStart(key2)
if iterator == nil {
t.Fatalf("db.NewIteratorWithStart returned nil")
}
defer iterator.Release()
if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key2) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key2)
} else if value := iterator.Value(); !bytes.Equal(value, value2) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value2)
} else if iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", true, false)
} else if key := iterator.Key(); key != nil {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: nil", key)
} else if value := iterator.Value(); value != nil {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: nil", value)
} else if err := iterator.Error(); err != nil {
t.Fatalf("iterator.Error Returned: %s ; Expected: nil", err)
}
}
// TestIteratorPrefix ...
func TestIteratorPrefix(t *testing.T, db Database) {
key1 := []byte("hello")
value1 := []byte("world1")
key2 := []byte("goodbye")
value2 := []byte("world2")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
iterator := db.NewIteratorWithPrefix([]byte("h"))
if iterator == nil {
t.Fatalf("db.NewIteratorWithPrefix returned nil")
}
defer iterator.Release()
if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key1) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key1)
} else if value := iterator.Value(); !bytes.Equal(value, value1) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value1)
} else if iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", true, false)
} else if key := iterator.Key(); key != nil {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: nil", key)
} else if value := iterator.Value(); value != nil {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: nil", value)
} else if err := iterator.Error(); err != nil {
t.Fatalf("iterator.Error Returned: %s ; Expected: nil", err)
}
}
// TestIteratorStartPrefix ...
func TestIteratorStartPrefix(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("z")
value2 := []byte("world2")
key3 := []byte("hello3")
value3 := []byte("world3")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key3, value3); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
iterator := db.NewIteratorWithStartAndPrefix(key1, []byte("h"))
if iterator == nil {
t.Fatalf("db.NewIteratorWithStartAndPrefix returned nil")
}
defer iterator.Release()
if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key1) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key1)
} else if value := iterator.Value(); !bytes.Equal(value, value1) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value1)
} else if !iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", false, true)
} else if key := iterator.Key(); !bytes.Equal(key, key3) {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: 0x%x", key, key3)
} else if value := iterator.Value(); !bytes.Equal(value, value3) {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: 0x%x", value, value3)
} else if iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", true, false)
} else if key := iterator.Key(); key != nil {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: nil", key)
} else if value := iterator.Value(); value != nil {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: nil", value)
} else if err := iterator.Error(); err != nil {
t.Fatalf("iterator.Error Returned: %s ; Expected: nil", err)
}
}
// TestIteratorClosed ...
func TestIteratorClosed(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
if err := db.Close(); err != nil {
t.Fatalf("Unexpected error on db.Close: %s", err)
}
iterator := db.NewIterator()
if iterator == nil {
t.Fatalf("db.NewIterator returned nil")
}
defer iterator.Release()
if iterator.Next() {
t.Fatalf("iterator.Next Returned: %v ; Expected: %v", true, false)
} else if key := iterator.Key(); key != nil {
t.Fatalf("iterator.Key Returned: 0x%x ; Expected: nil", key)
} else if value := iterator.Value(); value != nil {
t.Fatalf("iterator.Value Returned: 0x%x ; Expected: nil", value)
} else if err := iterator.Error(); err != ErrClosed {
t.Fatalf("Expected %s on iterator.Error", ErrClosed)
}
}
// TestStatNoPanic ...
func TestStatNoPanic(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("z")
value2 := []byte("world2")
key3 := []byte("hello3")
value3 := []byte("world3")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key3, value3); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
db.Stat("")
if err := db.Close(); err != nil {
t.Fatalf("Unexpected error on db.Close: %s", err)
}
db.Stat("")
}
// TestCompactNoPanic ...
func TestCompactNoPanic(t *testing.T, db Database) {
key1 := []byte("hello1")
value1 := []byte("world1")
key2 := []byte("z")
value2 := []byte("world2")
key3 := []byte("hello3")
value3 := []byte("world3")
if err := db.Put(key1, value1); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key2, value2); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
} else if err := db.Put(key3, value3); err != nil {
t.Fatalf("Unexpected error on batch.Put: %s", err)
}
db.Compact(nil, nil)
if err := db.Close(); err != nil {
t.Fatalf("Unexpected error on db.Close: %s", err)
}
db.Compact(nil, nil)
}