60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package mdb
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func (i Index[T]) AssertEqual(t *testing.T, tx1, tx2 *Snapshot) {
|
|
t.Helper()
|
|
|
|
state1 := i.getState(tx1)
|
|
state2 := i.getState(tx2)
|
|
|
|
if state1.BTree.Len() != state2.BTree.Len() {
|
|
t.Fatalf("(%s) Unequal lengths: %d != %d",
|
|
i.name,
|
|
state1.BTree.Len(),
|
|
state2.BTree.Len())
|
|
}
|
|
|
|
errStr := ""
|
|
i.Ascend(tx1, func(item1 *T) bool {
|
|
item2, ok := i.Get(tx2, item1)
|
|
if !ok {
|
|
errStr = fmt.Sprintf("Indices don't match. %v not found.", item1)
|
|
return false
|
|
}
|
|
if !reflect.DeepEqual(item1, item2) {
|
|
errStr = fmt.Sprintf("%v != %v", item1, item2)
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
|
|
if errStr != "" {
|
|
t.Fatal(errStr)
|
|
}
|
|
}
|
|
|
|
func (c *Collection[T]) AssertEqual(t *testing.T, tx1, tx2 *Snapshot) {
|
|
t.Helper()
|
|
c.ByID.AssertEqual(t, tx1, tx2)
|
|
|
|
for _, idx := range c.indices {
|
|
idx.AssertEqual(t, tx1, tx2)
|
|
}
|
|
}
|
|
|
|
func (db *Database) AssertEqual(t *testing.T, db2 *Database) {
|
|
tx1 := db.Snapshot()
|
|
tx2 := db.Snapshot()
|
|
for _, c := range db.collections {
|
|
cc := c.(interface {
|
|
AssertEqual(t *testing.T, tx1, tx2 *Snapshot)
|
|
})
|
|
cc.AssertEqual(t, tx1, tx2)
|
|
}
|
|
}
|