29 lines
499 B
Go
29 lines
499 B
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMemDbIterator(t *testing.T) {
|
||
|
|
db := NewMemDB()
|
||
|
|
keys := make([][]byte, 100)
|
||
|
|
for i := 0; i < 100; i++ {
|
||
|
|
keys[i] = []byte{byte(i)}
|
||
|
|
}
|
||
|
|
|
||
|
|
value := []byte{5}
|
||
|
|
for _, k := range keys {
|
||
|
|
db.Set(k, value)
|
||
|
|
}
|
||
|
|
|
||
|
|
iter := db.Iterator()
|
||
|
|
i := 0
|
||
|
|
for iter.Next() {
|
||
|
|
assert.Equal(t, db.Get(iter.Key()), iter.Value(), "values dont match for key")
|
||
|
|
i += 1
|
||
|
|
}
|
||
|
|
assert.Equal(t, i, len(db.db), "iterator didnt cover whole db")
|
||
|
|
}
|