Initial commit: SRDB - High-performance LSM-Tree database
- Core engine with MemTable, SST, WAL - B+Tree indexing for SST files - Leveled compaction strategy - Multi-table database management - Schema validation and secondary indexes - Query builder with complex conditions - Web UI with HTMX for data visualization - Command-line tools for diagnostics
This commit is contained in:
121
memtable/memtable_test.go
Normal file
121
memtable/memtable_test.go
Normal file
@@ -0,0 +1,121 @@
|
||||
package memtable
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestMemTable(t *testing.T) {
|
||||
mt := New()
|
||||
|
||||
// 1. 插入数据
|
||||
for i := int64(1); i <= 100; i++ {
|
||||
mt.Put(i, []byte("value_"+string(rune(i))))
|
||||
}
|
||||
|
||||
if mt.Count() != 100 {
|
||||
t.Errorf("Expected 100 entries, got %d", mt.Count())
|
||||
}
|
||||
|
||||
t.Logf("Inserted 100 entries, size: %d bytes", mt.Size())
|
||||
|
||||
// 2. 查询数据
|
||||
for i := int64(1); i <= 100; i++ {
|
||||
value, exists := mt.Get(i)
|
||||
if !exists {
|
||||
t.Errorf("Key %d not found", i)
|
||||
}
|
||||
if value == nil {
|
||||
t.Errorf("Key %d: value is nil", i)
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 查询不存在的 key
|
||||
_, exists := mt.Get(101)
|
||||
if exists {
|
||||
t.Error("Key 101 should not exist")
|
||||
}
|
||||
|
||||
t.Log("All tests passed!")
|
||||
}
|
||||
|
||||
func TestMemTableIterator(t *testing.T) {
|
||||
mt := New()
|
||||
|
||||
// 插入数据 (乱序)
|
||||
keys := []int64{5, 2, 8, 1, 9, 3, 7, 4, 6, 10}
|
||||
for _, key := range keys {
|
||||
mt.Put(key, []byte("value"))
|
||||
}
|
||||
|
||||
// 迭代器应该按顺序返回
|
||||
iter := mt.NewIterator()
|
||||
var result []int64
|
||||
for iter.Next() {
|
||||
result = append(result, iter.Key())
|
||||
}
|
||||
|
||||
// 验证顺序
|
||||
for i := 0; i < len(result)-1; i++ {
|
||||
if result[i] >= result[i+1] {
|
||||
t.Errorf("Keys not in order: %v", result)
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if len(result) != 10 {
|
||||
t.Errorf("Expected 10 keys, got %d", len(result))
|
||||
}
|
||||
|
||||
t.Logf("Iterator returned keys in order: %v", result)
|
||||
}
|
||||
|
||||
func TestMemTableClear(t *testing.T) {
|
||||
mt := New()
|
||||
|
||||
// 插入数据
|
||||
for i := int64(1); i <= 10; i++ {
|
||||
mt.Put(i, []byte("value"))
|
||||
}
|
||||
|
||||
if mt.Count() != 10 {
|
||||
t.Errorf("Expected 10 entries, got %d", mt.Count())
|
||||
}
|
||||
|
||||
// 清空
|
||||
mt.Clear()
|
||||
|
||||
if mt.Count() != 0 {
|
||||
t.Errorf("Expected 0 entries after clear, got %d", mt.Count())
|
||||
}
|
||||
|
||||
if mt.Size() != 0 {
|
||||
t.Errorf("Expected size 0 after clear, got %d", mt.Size())
|
||||
}
|
||||
|
||||
t.Log("Clear test passed!")
|
||||
}
|
||||
|
||||
func BenchmarkMemTablePut(b *testing.B) {
|
||||
mt := New()
|
||||
value := make([]byte, 100)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
mt.Put(int64(i), value)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkMemTableGet(b *testing.B) {
|
||||
mt := New()
|
||||
value := make([]byte, 100)
|
||||
|
||||
// 预先插入数据
|
||||
for i := int64(0); i < 10000; i++ {
|
||||
mt.Put(i, value)
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
mt.Get(int64(i % 10000))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user