istanbul-tools/tests/load/load_test.go

// 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 <http://www.gnu.org/licenses/>.

package load

import (
	"math/big"
	"sync"
	"testing"
	"time"

	"github.com/ethereum/go-ethereum/common"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"

	istcommon "github.com/getamis/istanbul-tools/common"
	"github.com/getamis/istanbul-tools/container"
	"github.com/getamis/istanbul-tools/k8s"
	"github.com/getamis/istanbul-tools/metrics"
	"github.com/getamis/istanbul-tools/tests"
)

var _ = Describe("TPS-01: Large amount of transactions", func() {
	tests.CaseTable("with number of validators",
		func(numberOfValidators int) {
			tests.CaseTable("with gas limit",
				func(gaslimit int) {
					tests.CaseTable("with txpool size",
						func(txpoolSize int) {
							runTests(numberOfValidators, gaslimit, txpoolSize)
						},

						tests.Case("10240", 10240),
					)
				},

				tests.Case("21000*3000", 21000*3000),
			)

		},

		tests.Case("4 validators", 4),
	)
})

func runTests(numberOfValidators int, gaslimit int, txpoolSize int) {
	Describe("", func() {
		var (
			blockchain     container.Blockchain
			sendEtherAddrs map[common.Address]common.Address

			duration        = 10 * time.Minute
			accountsPerGeth = 30
		)

		BeforeEach(func() {
			blockchain = k8s.NewBlockchain(
				numberOfValidators,
				accountsPerGeth,
				uint64(gaslimit),
				k8s.ImageRepository("quay.io/amis/geth"),
				k8s.ImageTag("istanbul_develop"),
				k8s.Mine(),
				k8s.TxPoolSize(txpoolSize),
			)
			blockchain = metrics.NewMetricChain(blockchain)
			Expect(blockchain).NotTo(BeNil())
			Expect(blockchain.Start(true)).To(BeNil())

			sendEtherAddrs = make(map[common.Address]common.Address)
			num := len(blockchain.Validators())
			for i, v := range blockchain.Validators() {
				sendEtherAddrs[v.Address()] = blockchain.Validators()[(i+1)%num].Address()
			}
		})

		AfterEach(func() {
			Expect(blockchain).NotTo(BeNil())
			var blocksCnt int = 5
			metricsExport, ok := blockchain.(metrics.Exporter)
			if ok {
				blockSize := gaslimit / int(istcommon.DefaultGasLimit)
				blocksCnt = int(metricsExport.SentTxCount()-metricsExport.ExcutedTxCount())/blockSize + 1
			}
			tests.WaitFor(blockchain.Validators(), func(geth container.Ethereum, wg *sync.WaitGroup) {
				Expect(geth.WaitForBlocks(blocksCnt)).To(BeNil())
				wg.Done()
			})
			Expect(blockchain.Stop(true)).To(BeNil())
			blockchain.Finalize()
		})

		It("", func() {
			By("Wait for p2p connection", func() {
				tests.WaitFor(blockchain.Validators(), func(geth container.Ethereum, wg *sync.WaitGroup) {
					Expect(geth.WaitForPeersConnected(numberOfValidators - 1)).To(BeNil())
					wg.Done()
				})
			})

			By("Send transactions", func() {
				tests.WaitFor(blockchain.Validators(), func(geth container.Ethereum, wg *sync.WaitGroup) {
					transactor, ok := geth.(k8s.Transactor)
					Expect(ok).To(BeTrue())

					client := geth.NewClient()
					Expect(client).NotTo(BeNil())

					Expect(transactor.SendTransactions(
						client,
						new(big.Int).Exp(big.NewInt(10), big.NewInt(3), nil),
						duration)).To(BeNil())

					wg.Done()
				})
			})
		})
	})
}

func TestIstanbulLoadTesting(t *testing.T) {
	RegisterFailHandler(Fail)
	RunSpecs(t, "Istanbul Load Test Suite")
}