60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
|
package core
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/core/storage"
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDecodeEncode(t *testing.T) {
|
||
|
|
unspent := &UnspentCoinState{
|
||
|
|
states: []CoinState{
|
||
|
|
CoinStateConfirmed,
|
||
|
|
CoinStateSpent,
|
||
|
|
CoinStateSpent,
|
||
|
|
CoinStateSpent,
|
||
|
|
CoinStateConfirmed,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
assert.Nil(t, unspent.EncodeBinary(buf))
|
||
|
|
unspentDecode := &UnspentCoinState{}
|
||
|
|
assert.Nil(t, unspentDecode.DecodeBinary(buf))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommit(t *testing.T) {
|
||
|
|
var (
|
||
|
|
store = storage.NewMemoryStore()
|
||
|
|
batch = store.Batch()
|
||
|
|
unspentCoins = make(UnspentCoins)
|
||
|
|
)
|
||
|
|
|
||
|
|
txA := util.RandomUint256()
|
||
|
|
txB := util.RandomUint256()
|
||
|
|
txC := util.RandomUint256()
|
||
|
|
|
||
|
|
unspentCoins[txA] = &UnspentCoinState{
|
||
|
|
states: []CoinState{CoinStateConfirmed},
|
||
|
|
}
|
||
|
|
unspentCoins[txB] = &UnspentCoinState{
|
||
|
|
states: []CoinState{
|
||
|
|
CoinStateConfirmed,
|
||
|
|
CoinStateConfirmed,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
unspentCoins[txC] = &UnspentCoinState{
|
||
|
|
states: []CoinState{
|
||
|
|
CoinStateConfirmed,
|
||
|
|
CoinStateConfirmed,
|
||
|
|
CoinStateConfirmed,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Nil(t, unspentCoins.commit(batch))
|
||
|
|
assert.Nil(t, store.PutBatch(batch))
|
||
|
|
}
|