123 lines
4.3 KiB
Go
123 lines
4.3 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package utils_test
|
|
|
|
import (
|
|
"math/big"
|
|
"math/rand"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
"github.com/ethereum/go-ethereum/statediff"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
|
|
"github.com/vulcanize/vulcanizedb/libraries/shared/test_data"
|
|
"github.com/vulcanize/vulcanizedb/pkg/eth/fakes"
|
|
)
|
|
|
|
var _ = Describe("Storage row parsing", func() {
|
|
Describe("FromParityCsvRow", func() {
|
|
It("converts an array of strings to a row struct", func() {
|
|
contract := "0x123"
|
|
blockHash := "0x456"
|
|
blockHeight := "789"
|
|
storageKey := "0x987"
|
|
storageValue := "0x654"
|
|
data := []string{contract, blockHash, blockHeight, storageKey, storageValue}
|
|
|
|
result, err := utils.FromParityCsvRow(data)
|
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
expectedKeccakOfContractAddress := utils.HexToKeccak256Hash(contract)
|
|
Expect(result.HashedAddress).To(Equal(expectedKeccakOfContractAddress))
|
|
Expect(result.BlockHash).To(Equal(common.HexToHash(blockHash)))
|
|
Expect(result.BlockHeight).To(Equal(789))
|
|
Expect(result.StorageKey).To(Equal(common.HexToHash(storageKey)))
|
|
Expect(result.StorageValue).To(Equal(common.HexToHash(storageValue)))
|
|
})
|
|
|
|
It("returns an error if row is missing data", func() {
|
|
_, err := utils.FromParityCsvRow([]string{"0x123"})
|
|
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err).To(MatchError(utils.ErrRowMalformed{Length: 1}))
|
|
})
|
|
|
|
It("returns error if block height malformed", func() {
|
|
_, err := utils.FromParityCsvRow([]string{"", "", "", "", ""})
|
|
|
|
Expect(err).To(HaveOccurred())
|
|
})
|
|
})
|
|
|
|
Describe("FromGethStateDiff", func() {
|
|
var (
|
|
accountDiff = statediff.AccountDiff{Key: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0}}
|
|
stateDiff = &statediff.StateDiff{
|
|
BlockNumber: big.NewInt(rand.Int63()),
|
|
BlockHash: fakes.FakeHash,
|
|
}
|
|
)
|
|
|
|
It("adds relevant fields to diff", func() {
|
|
storageValueBytes := []byte{3}
|
|
storageValueRlp, encodeErr := rlp.EncodeToBytes(storageValueBytes)
|
|
Expect(encodeErr).NotTo(HaveOccurred())
|
|
|
|
storageDiff := statediff.StorageDiff{
|
|
Key: []byte{0, 9, 8, 7, 6, 5, 4, 3, 2, 1},
|
|
Value: storageValueRlp,
|
|
}
|
|
|
|
result, err := utils.FromGethStateDiff(accountDiff, stateDiff, storageDiff)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
expectedAddress := common.BytesToHash(accountDiff.Key)
|
|
Expect(result.HashedAddress).To(Equal(expectedAddress))
|
|
Expect(result.BlockHash).To(Equal(fakes.FakeHash))
|
|
expectedBlockHeight := int(stateDiff.BlockNumber.Int64())
|
|
Expect(result.BlockHeight).To(Equal(expectedBlockHeight))
|
|
expectedStorageKey := common.BytesToHash(storageDiff.Key)
|
|
Expect(result.StorageKey).To(Equal(expectedStorageKey))
|
|
expectedStorageValue := common.BytesToHash(storageValueBytes)
|
|
Expect(result.StorageValue).To(Equal(expectedStorageValue))
|
|
})
|
|
|
|
It("handles decoding large storage values from their RLP", func() {
|
|
storageValueBytes := []byte{1, 2, 3, 4, 5, 0, 9, 8, 7, 6}
|
|
storageValueRlp, encodeErr := rlp.EncodeToBytes(storageValueBytes)
|
|
Expect(encodeErr).NotTo(HaveOccurred())
|
|
|
|
storageDiff := statediff.StorageDiff{
|
|
Key: []byte{0, 9, 8, 7, 6, 5, 4, 3, 2, 1},
|
|
Value: storageValueRlp,
|
|
}
|
|
|
|
result, err := utils.FromGethStateDiff(accountDiff, stateDiff, storageDiff)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(result.StorageValue).To(Equal(common.BytesToHash(storageValueBytes)))
|
|
})
|
|
|
|
It("returns an err if decoding the storage value Rlp fails", func() {
|
|
_, err := utils.FromGethStateDiff(accountDiff, stateDiff, test_data.StorageWithBadValue)
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err).To(MatchError("rlp: input contains more than one value"))
|
|
})
|
|
})
|
|
})
|