104 lines
3.2 KiB
Go
104 lines
3.2 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 storage_test
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/common"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
|
|
"github.com/vulcanize/vulcanizedb/libraries/shared/factories/storage"
|
|
"github.com/vulcanize/vulcanizedb/libraries/shared/mocks"
|
|
"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
|
|
"github.com/vulcanize/vulcanizedb/pkg/fakes"
|
|
)
|
|
|
|
var _ = Describe("Storage transformer", func() {
|
|
var (
|
|
mappings *mocks.MockMappings
|
|
repository *mocks.MockStorageRepository
|
|
t storage.Transformer
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
mappings = &mocks.MockMappings{}
|
|
repository = &mocks.MockStorageRepository{}
|
|
t = storage.Transformer{
|
|
Address: common.Address{},
|
|
Mappings: mappings,
|
|
Repository: repository,
|
|
}
|
|
})
|
|
|
|
It("returns the contract address being watched", func() {
|
|
fakeAddress := common.HexToAddress("0x12345")
|
|
t.Address = fakeAddress
|
|
|
|
Expect(t.ContractAddress()).To(Equal(fakeAddress))
|
|
})
|
|
|
|
It("looks up metadata for storage key", func() {
|
|
t.Execute(utils.StorageDiffRow{})
|
|
|
|
Expect(mappings.LookupCalled).To(BeTrue())
|
|
})
|
|
|
|
It("returns error if lookup fails", func() {
|
|
mappings.LookupErr = fakes.FakeError
|
|
|
|
err := t.Execute(utils.StorageDiffRow{})
|
|
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err).To(MatchError(fakes.FakeError))
|
|
})
|
|
|
|
It("creates storage row with decoded data", func() {
|
|
fakeMetadata := utils.StorageValueMetadata{Type: utils.Address}
|
|
mappings.Metadata = fakeMetadata
|
|
rawValue := common.HexToAddress("0x12345")
|
|
fakeBlockNumber := 123
|
|
fakeBlockHash := "0x67890"
|
|
fakeRow := utils.StorageDiffRow{
|
|
Contract: common.Address{},
|
|
BlockHash: common.HexToHash(fakeBlockHash),
|
|
BlockHeight: fakeBlockNumber,
|
|
StorageKey: common.Hash{},
|
|
StorageValue: rawValue.Hash(),
|
|
}
|
|
|
|
err := t.Execute(fakeRow)
|
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(repository.PassedBlockNumber).To(Equal(fakeBlockNumber))
|
|
Expect(repository.PassedBlockHash).To(Equal(common.HexToHash(fakeBlockHash).Hex()))
|
|
Expect(repository.PassedMetadata).To(Equal(fakeMetadata))
|
|
Expect(repository.PassedValue.(string)).To(Equal(rawValue.Hex()))
|
|
})
|
|
|
|
It("returns error if creating row fails", func() {
|
|
rawValue := common.HexToAddress("0x12345")
|
|
fakeMetadata := utils.StorageValueMetadata{Type: utils.Address}
|
|
mappings.Metadata = fakeMetadata
|
|
repository.CreateErr = fakes.FakeError
|
|
|
|
err := t.Execute(utils.StorageDiffRow{StorageValue: rawValue.Hash()})
|
|
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err).To(MatchError(fakes.FakeError))
|
|
})
|
|
})
|