|
@ -4,6 +4,7 @@ import ( |
|
|
"testing" |
|
|
"testing" |
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert" |
|
|
"github.com/stretchr/testify/assert" |
|
|
|
|
|
"github.com/stretchr/testify/require" |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
func TestMemDbIterator(t *testing.T) { |
|
|
func TestMemDbIterator(t *testing.T) { |
|
@ -26,3 +27,22 @@ func TestMemDbIterator(t *testing.T) { |
|
|
} |
|
|
} |
|
|
assert.Equal(t, i, len(db.db), "iterator didnt cover whole db") |
|
|
assert.Equal(t, i, len(db.db), "iterator didnt cover whole db") |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func TestMemDBClose(t *testing.T) { |
|
|
|
|
|
db := NewMemDB() |
|
|
|
|
|
copyDB := func(orig map[string][]byte) map[string][]byte { |
|
|
|
|
|
copy := make(map[string][]byte) |
|
|
|
|
|
for k, v := range orig { |
|
|
|
|
|
copy[k] = v |
|
|
|
|
|
} |
|
|
|
|
|
return copy |
|
|
|
|
|
} |
|
|
|
|
|
k, v := []byte("foo"), []byte("bar") |
|
|
|
|
|
db.Set(k, v) |
|
|
|
|
|
require.Equal(t, db.Get(k), v, "expecting a successful get") |
|
|
|
|
|
copyBefore := copyDB(db.db) |
|
|
|
|
|
db.Close() |
|
|
|
|
|
require.Equal(t, db.Get(k), v, "Close is a noop, expecting a successful get") |
|
|
|
|
|
copyAfter := copyDB(db.db) |
|
|
|
|
|
require.Equal(t, copyBefore, copyAfter, "Close is a noop and shouldn't modify any internal data") |
|
|
|
|
|
} |