neo-go/pkg/core/unspent_coint_state_test.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 TestDecodeEncodeUnspentCoinState(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 TestCommitUnspentCoins(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))
}