From 36a7eb65fe977da8f197fd0cc32eeb51df712f90 Mon Sep 17 00:00:00 2001 From: Miya Chen Date: Mon, 11 Sep 2017 18:00:57 +0800 Subject: [PATCH] tests/quorum: add private transaction test --- container/blockchain.go | 2 +- .../functional/private_transaction_test.go | 234 ++++++++++++++++++ 2 files changed, 235 insertions(+), 1 deletion(-) create mode 100644 tests/quorum/functional/private_transaction_test.go diff --git a/container/blockchain.go b/container/blockchain.go index da7a4faf..7c42d778 100644 --- a/container/blockchain.go +++ b/container/blockchain.go @@ -204,7 +204,7 @@ func NewDefaultQuorumBlockchain(network *DockerNetwork, ctn ConstellationNetwork return NewQuorumBlockchain(network, ctn, ImageRepository("quay.io/amis/quorum"), - ImageTag("latest"), + ImageTag("feature_istanbul"), DataDir("/data"), WebSocket(), WebSocketAddress("0.0.0.0"), diff --git a/tests/quorum/functional/private_transaction_test.go b/tests/quorum/functional/private_transaction_test.go new file mode 100644 index 00000000..4c532382 --- /dev/null +++ b/tests/quorum/functional/private_transaction_test.go @@ -0,0 +1,234 @@ +// Copyright 2017 AMIS Technologies +// This file is part of the go-ethereum library. +// +// The go-ethereum library is free software: you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// The go-ethereum library 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 Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the go-ethereum library. If not, see . + +package functional + +import ( + "context" + "errors" + "fmt" + "math/big" + "time" + + . "github.com/onsi/ginkgo" + . "github.com/onsi/gomega" + + "github.com/ethereum/go-ethereum/common" + "github.com/ethereum/go-ethereum/core/types" + "github.com/getamis/istanbul-tools/client" + "github.com/getamis/istanbul-tools/container" +) + +const ( + testBaseByteCode = "0x6060604052341561000f57600080fd5b604051602080610149833981016040528080519060200190919050505b806000819055505b505b610104806100456000396000f30060606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff1680632a1afcd914605157806360fe47b11460775780636d4ce63c146097575b600080fd5b3415605b57600080fd5b606160bd565b6040518082815260200191505060405180910390f35b3415608157600080fd5b6095600480803590602001909190505060c3565b005b341560a157600080fd5b60a760ce565b6040518082815260200191505060405180910390f35b60005481565b806000819055505b50565b6000805490505b905600a165627a7a72305820278d34fe369cdf9e578c4d5cdbbdffa7e964a8e34060e68788e08e52c20181c10029" +) + +var _ = Describe("QFS-08: Private transaction", func() { + const ( + numberOfValidators = 4 + ) + var ( + constellationNetwork container.ConstellationNetwork + blockchain container.Blockchain + ) + + BeforeEach(func() { + constellationNetwork = container.NewDefaultConstellationNetwork(dockerNetwork, numberOfValidators) + Expect(constellationNetwork.Start()).To(BeNil()) + blockchain = container.NewDefaultQuorumBlockchain(dockerNetwork, constellationNetwork) + Expect(blockchain.Start(true)).To(BeNil()) + }) + + AfterEach(func() { + blockchain.Stop(true) + blockchain.Finalize() + constellationNetwork.Stop() + constellationNetwork.Finalize() + }) + + It("QFS–08–01: Constellation connection", func() { + //Skip for now, constellation network will be ensured in BeforeEach. + }) + + It("QFS–08–02: Sending common contract transaction", func() { + const storedValue = 1 + var txHash common.Hash + + By("Sending tx should be successful", func() { + geth0 := blockchain.Validators()[0] + acc := geth0.Accounts()[0] + client0 := geth0.NewClient() + byteCode := genByteCodeWithValue(storedValue) + hash, err := client0.CreateContract(context.Background(), acc.Address, byteCode, big.NewInt(300000)) + Expect(err).To(BeNil()) + txHash = common.HexToHash(hash) + }) + + By("All geth nodes should can get value of contract storage", func() { + errc := make(chan error, len(blockchain.Validators())) + for _, geth := range blockchain.Validators() { + go func(geth container.Ethereum) { + ethClient := geth.NewClient() + errc <- checkContractValue(ethClient, txHash, storedValue) + }(geth) + } + for i := 0; i < len(blockchain.Validators()); i++ { + err := <-errc + Expect(err).To(BeNil()) + } + }) + }) + + It("QFS–08–03: Sending private contract transaction", func() { + var ( + txHash common.Hash + storedValue = 1 + ) + + By("Sending tx by geth#0 and private for geth#1 should be successful", func() { + geth0 := blockchain.Validators()[0] + acc := geth0.Accounts()[0] + client0 := geth0.NewClient() + ct1 := constellationNetwork.GetConstellation(1) + pubKey1 := ct1.PublicKeys() + byteCode := genByteCodeWithValue(storedValue) + hash, err := client0.CreatePrivateContract(context.Background(), acc.Address, byteCode, big.NewInt(300000), pubKey1) + Expect(err).To(BeNil()) + txHash = common.HexToHash(hash) + }) + + By("Only geth#0 and geth#1 can get value of contract storage", func() { + errc := make(chan error, len(blockchain.Validators())) + for i, geth := range blockchain.Validators() { + var expValue = 0 + if i == 0 || i == 1 { + expValue = storedValue + } + go func(geth container.Ethereum, expValue int) { + ethClient := geth.NewClient() + errc <- checkContractValue(ethClient, txHash, expValue) + }(geth, expValue) + } + for i := 0; i < len(blockchain.Validators()); i++ { + err := <-errc + Expect(err).To(BeNil()) + } + }) + + storedValue = 2 + By("Sending tx by geth#2 and private for geth#3 should be successful", func() { + geth2 := blockchain.Validators()[2] + acc := geth2.Accounts()[0] + client2 := geth2.NewClient() + ct3 := constellationNetwork.GetConstellation(3) + pubKey3 := ct3.PublicKeys() + byteCode := genByteCodeWithValue(storedValue) + hash, err := client2.CreatePrivateContract(context.Background(), acc.Address, byteCode, big.NewInt(300000), pubKey3) + Expect(err).To(BeNil()) + txHash = common.HexToHash(hash) + }) + + By("Only geth#2 and geth#3 can get value of contract storage", func() { + errc := make(chan error, len(blockchain.Validators())) + for i, geth := range blockchain.Validators() { + var expValue = 0 + if i == 2 || i == 3 { + expValue = storedValue + } + go func(geth container.Ethereum, expValue int) { + ethClient := geth.NewClient() + errc <- checkContractValue(ethClient, txHash, expValue) + }(geth, expValue) + } + for i := 0; i < len(blockchain.Validators()); i++ { + err := <-errc + Expect(err).To(BeNil()) + } + }) + + storedValue = 3 + By("Sending common tx after private tx should be successful", func() { + geth0 := blockchain.Validators()[0] + acc := geth0.Accounts()[0] + client0 := geth0.NewClient() + byteCode := genByteCodeWithValue(storedValue) + hash, err := client0.CreateContract(context.Background(), acc.Address, byteCode, big.NewInt(300000)) + Expect(err).To(BeNil()) + txHash = common.HexToHash(hash) + }) + + By("All geth nodes should can get value of contract storage", func() { + errc := make(chan error, len(blockchain.Validators())) + for _, geth := range blockchain.Validators() { + go func(geth container.Ethereum) { + ethClient := geth.NewClient() + errc <- checkContractValue(ethClient, txHash, storedValue) + }(geth) + } + for i := 0; i < len(blockchain.Validators()); i++ { + err := <-errc + Expect(err).To(BeNil()) + } + }) + }) +}) + +func genByteCodeWithValue(v int) string { + return fmt.Sprintf("%s%064x", testBaseByteCode, v) +} + +func getTxReceipt(ethClient *client.Client, txHash common.Hash, timeout time.Duration) (*types.Receipt, error) { + timer := time.After(timeout) + ticker := time.NewTicker(3 * time.Second) + defer ticker.Stop() + for { + select { + case <-timer: + return nil, errors.New("Get TxReceipt timeout.") + case <-ticker.C: + r, err := ethClient.TransactionReceipt(context.Background(), txHash) + if err == nil { + return r, nil + } + } + } +} + +func checkContractValue(ethClient *client.Client, txHash common.Hash, expValue int) error { + receipt, err := getTxReceipt(ethClient, txHash, 10*time.Second) + if err != nil { + return err + } + + emptyAddress := common.Address{} + if receipt.ContractAddress == emptyAddress { + return errors.New("Invalid contract address.") + } + + v, err := ethClient.StorageAt(context.Background(), + receipt.ContractAddress, + common.HexToHash("0x0"), + nil) + if err != nil { + return err + } + + if value := new(big.Int).SetBytes(v).Int64(); value != int64(expValue) { + errMsg := fmt.Sprintf("Wrong value of contract storage, get:%v, want:%v", value, expValue) + return errors.New(errMsg) + } + return nil +}