140 lines
4.8 KiB
Go
140 lines
4.8 KiB
Go
package watched_contracts_test
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
|
"github.com/8thlight/vulcanizedb/pkg/fakes"
|
|
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
|
"github.com/8thlight/vulcanizedb/pkg/watched_contracts"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func NewCurrentContractSummary(blockchain core.Blockchain, repository repositories.Repository, contractHash string) (watched_contracts.ContractSummary, error) {
|
|
return watched_contracts.NewSummary(blockchain, repository, contractHash, nil)
|
|
}
|
|
|
|
var _ = Describe("The watched contract summary", func() {
|
|
|
|
Context("when the given contract is not being watched", func() {
|
|
It("returns an error", func() {
|
|
repository := repositories.NewInMemory()
|
|
blockchain := fakes.NewBlockchain()
|
|
|
|
contractSummary, err := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary).To(Equal(watched_contracts.ContractSummary{}))
|
|
Expect(err).NotTo(BeNil())
|
|
})
|
|
})
|
|
|
|
Context("when the given contract is being watched", func() {
|
|
It("returns the summary", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
blockchain := fakes.NewBlockchain()
|
|
|
|
contractSummary, err := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary).NotTo(Equal(watched_contracts.ContractSummary{}))
|
|
Expect(err).To(BeNil())
|
|
})
|
|
|
|
It("includes the contract hash in the summary", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
blockchain := fakes.NewBlockchain()
|
|
|
|
contractSummary, _ := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary.ContractHash).To(Equal("0x123"))
|
|
})
|
|
|
|
It("sets the number of transactions", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
block := core.Block{
|
|
Transactions: []core.Transaction{
|
|
{To: "0x123"},
|
|
{To: "0x123"},
|
|
},
|
|
}
|
|
repository.CreateBlock(block)
|
|
blockchain := fakes.NewBlockchain()
|
|
|
|
contractSummary, _ := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary.NumberOfTransactions).To(Equal(2))
|
|
})
|
|
|
|
It("sets the last transaction", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
block := core.Block{
|
|
Transactions: []core.Transaction{
|
|
{Hash: "TRANSACTION2", To: "0x123"},
|
|
{Hash: "TRANSACTION1", To: "0x123"},
|
|
},
|
|
}
|
|
repository.CreateBlock(block)
|
|
blockchain := fakes.NewBlockchain()
|
|
|
|
contractSummary, _ := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary.LastTransaction.Hash).To(Equal("TRANSACTION2"))
|
|
})
|
|
|
|
It("gets contract state attribute for the contract from the blockchain", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
blockchain := fakes.NewBlockchain()
|
|
blockchain.SetContractStateAttribute("0x123", nil, "foo", "bar")
|
|
|
|
contractSummary, _ := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
attribute := contractSummary.GetStateAttribute("foo")
|
|
|
|
Expect(attribute).To(Equal("bar"))
|
|
})
|
|
|
|
It("gets contract state attribute for the contract from the blockchain at specific block height", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
blockchain := fakes.NewBlockchain()
|
|
blockNumber := big.NewInt(1000)
|
|
blockchain.SetContractStateAttribute("0x123", nil, "foo", "bar")
|
|
blockchain.SetContractStateAttribute("0x123", blockNumber, "foo", "baz")
|
|
|
|
contractSummary, _ := watched_contracts.NewSummary(blockchain, repository, "0x123", blockNumber)
|
|
attribute := contractSummary.GetStateAttribute("foo")
|
|
|
|
Expect(attribute).To(Equal("baz"))
|
|
})
|
|
|
|
It("gets attributes for the contract from the blockchain", func() {
|
|
repository := repositories.NewInMemory()
|
|
watchedContract := repositories.WatchedContract{Hash: "0x123"}
|
|
repository.CreateWatchedContract(watchedContract)
|
|
blockchain := fakes.NewBlockchain()
|
|
blockchain.SetContractStateAttribute("0x123", nil, "foo", "bar")
|
|
blockchain.SetContractStateAttribute("0x123", nil, "baz", "bar")
|
|
|
|
contractSummary, _ := NewCurrentContractSummary(blockchain, repository, "0x123")
|
|
|
|
Expect(contractSummary.Attributes).To(Equal(
|
|
core.ContractAttributes{
|
|
{Name: "baz", Type: "string"},
|
|
{Name: "foo", Type: "string"},
|
|
},
|
|
))
|
|
})
|
|
})
|
|
|
|
})
|