69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
|
package state
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/tendermint/basecoin/types"
|
||
|
|
eyes "github.com/tendermint/merkleeyes/client"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestState(t *testing.T) {
|
||
|
|
|
||
|
|
s := NewState(types.NewMemKVStore())
|
||
|
|
|
||
|
|
s.SetChainID("testchain")
|
||
|
|
assert.True(t, s.GetChainID() == "testchain", "ChainID is improperly stored")
|
||
|
|
|
||
|
|
setRecords := func(kv types.KVStore) {
|
||
|
|
kv.Set([]byte("foo"), []byte("snake"))
|
||
|
|
kv.Set([]byte("bar"), []byte("mouse"))
|
||
|
|
}
|
||
|
|
|
||
|
|
setRecords(s)
|
||
|
|
assert.True(t, bytes.Equal(s.Get([]byte("foo")), []byte("snake")), "state doesn't retrieve after Set")
|
||
|
|
assert.True(t, bytes.Equal(s.Get([]byte("bar")), []byte("mouse")), "state doesn't retrieve after Set")
|
||
|
|
|
||
|
|
// Test account retrieve
|
||
|
|
dumAddr := []byte("dummyAddress")
|
||
|
|
|
||
|
|
acc := &types.Account{
|
||
|
|
PubKey: nil,
|
||
|
|
Sequence: 1,
|
||
|
|
Balance: nil,
|
||
|
|
}
|
||
|
|
|
||
|
|
s.SetAccount(dumAddr, acc)
|
||
|
|
assert.True(t, s.GetAccount(dumAddr).Sequence == 1, "GetAccount not retrieving")
|
||
|
|
|
||
|
|
//Test CacheWrap with local mem store
|
||
|
|
store := types.NewMemKVStore()
|
||
|
|
s = NewState(store)
|
||
|
|
cache := s.CacheWrap()
|
||
|
|
setRecords(cache)
|
||
|
|
assert.True(t, !bytes.Equal(store.Get([]byte("foo")), []byte("snake")), "store retrieving before Commit")
|
||
|
|
assert.True(t, !bytes.Equal(store.Get([]byte("bar")), []byte("mouse")), "store retrieving before Commit")
|
||
|
|
cache.CacheSync()
|
||
|
|
assert.True(t, bytes.Equal(store.Get([]byte("foo")), []byte("snake")), "store doesn't retrieve after Commit")
|
||
|
|
assert.True(t, bytes.Equal(store.Get([]byte("bar")), []byte("mouse")), "store doesn't retrieve after Commit")
|
||
|
|
|
||
|
|
//Test Commit on state with non-merkle store
|
||
|
|
assert.True(t, !s.Commit().IsOK(), "Commit shouldn't work with non-merkle store")
|
||
|
|
|
||
|
|
//Test CacheWrap with merkleeyes client store
|
||
|
|
eyesCli := eyes.NewLocalClient("", 0)
|
||
|
|
s = NewState(eyesCli)
|
||
|
|
|
||
|
|
cache = s.CacheWrap()
|
||
|
|
setRecords(cache)
|
||
|
|
assert.True(t, !bytes.Equal(eyesCli.Get([]byte("foo")), []byte("snake")), "store retrieving before Commit")
|
||
|
|
assert.True(t, !bytes.Equal(eyesCli.Get([]byte("bar")), []byte("mouse")), "store retrieving before Commit")
|
||
|
|
cache.CacheSync()
|
||
|
|
assert.True(t, s.Commit().IsOK(), "Bad Commit")
|
||
|
|
assert.True(t, bytes.Equal(eyesCli.Get([]byte("foo")), []byte("snake")), "store doesn't retrieve after Commit")
|
||
|
|
assert.True(t, bytes.Equal(eyesCli.Get([]byte("bar")), []byte("mouse")), "store doesn't retrieve after Commit")
|
||
|
|
|
||
|
|
}
|