neo-go/pkg/core/storage_item_test.go

package core

import (
	"testing"

	"github.com/CityOfZion/neo-go/pkg/core/storage"
	"github.com/CityOfZion/neo-go/pkg/util"
	"github.com/stretchr/testify/assert"
)

func TestPutGetDeleteStorageItem(t *testing.T) {
	s := storage.NewMemoryStore()
	si := &StorageItem{
		Value: []byte("smth"),
	}
	key := []byte("key")
	cHash, err := util.Uint160DecodeBytesBE([]byte("abcdefghijklmnopqrst"))
	assert.Nil(t, err)
	assert.NoError(t, putStorageItemIntoStore(s, cHash, key, si))
	siRead := getStorageItemFromStore(s, cHash, key)
	assert.NotNil(t, siRead)
	assert.Equal(t, si, siRead)
	assert.NoError(t, deleteStorageItemInStore(s, cHash, key))
	siRead2 := getStorageItemFromStore(s, cHash, key)
	assert.Nil(t, siRead2)
}