arnaucube
arnaucube/hermez-nodePublic Self-hosted repository
hermez-node/eth/main_test.go
eth/main_test.goRaw
  1package eth
  2
  3import (
  4	"fmt"
  5	"io/ioutil"
  6	"math/big"
  7	"os"
  8	"strconv"
  9	"testing"
 10
 11	"github.com/ethereum/go-ethereum/accounts"
 12	"github.com/ethereum/go-ethereum/accounts/keystore"
 13	ethCommon "github.com/ethereum/go-ethereum/common"
 14	"github.com/ethereum/go-ethereum/crypto"
 15	"github.com/ethereum/go-ethereum/ethclient"
 16	"github.com/joho/godotenv"
 17)
 18
 19var errEnvVar = fmt.Errorf("Some environment variable is missing")
 20
 21var (
 22	ethClientDialURL = "http://localhost:8545"
 23	password         = "pass"
 24	deadline, _      = new(big.Int).SetString("ffffffffffffffffffffffffffffffff", 16)
 25)
 26
 27// Smart Contract Addresses
 28var (
 29	genesisBlock             int64
 30	auctionAddressConst      ethCommon.Address
 31	auctionTestAddressConst  ethCommon.Address
 32	tokenHEZAddressConst     ethCommon.Address
 33	hermezRollupAddressConst ethCommon.Address
 34	wdelayerAddressConst     ethCommon.Address
 35	wdelayerTestAddressConst ethCommon.Address
 36	tokenHEZ                 TokenConfig
 37
 38	donationAddressStr          = "0x6c365935CA8710200C7595F0a72EB6023A7706Cd"
 39	donationAddressConst        = ethCommon.HexToAddress(donationAddressStr)
 40	bootCoordinatorAddressStr   = "0xc783df8a850f42e7f7e57013759c285caa701eb6"
 41	bootCoordinatorAddressConst = ethCommon.HexToAddress(bootCoordinatorAddressStr)
 42	safetyAddressStr            = "0xE5904695748fe4A84b40b3fc79De2277660BD1D3"
 43	safetyAddressConst          = ethCommon.HexToAddress(safetyAddressStr)
 44)
 45
 46// Ethereum Accounts
 47var (
 48	hermezGovernanceDAOAddressSK    = "2a8aede924268f84156a00761de73998dac7bf703408754b776ff3f873bcec60"
 49	hermezGovernanceDAOAddressStr   = "0x84Fae3d3Cba24A97817b2a18c2421d462dbBCe9f"
 50	hermezGovernanceDAOAddressConst = ethCommon.HexToAddress(hermezGovernanceDAOAddressStr)
 51
 52	whiteHackGroupAddressSK    = "8b24fd94f1ce869d81a34b95351e7f97b2cd88a891d5c00abc33d0ec9501902e"
 53	whiteHackGroupAddressStr   = "0xfa3BdC8709226Da0dA13A4d904c8b66f16c3c8BA"
 54	whiteHackGroupAddressConst = ethCommon.HexToAddress(whiteHackGroupAddressStr)
 55
 56	hermezKeeperAddressSK    = "7f307c41137d1ed409f0a7b028f6c7596f12734b1d289b58099b99d60a96efff"
 57	hermezKeeperAddressStr   = "0xFbC51a9582D031f2ceaaD3959256596C5D3a5468"
 58	hermezKeeperAddressConst = ethCommon.HexToAddress(hermezKeeperAddressStr)
 59
 60	governanceAddressSK    = "d49743deccbccc5dc7baa8e69e5be03298da8688a15dd202e20f15d5e0e9a9fb"
 61	governanceAddressStr   = "0xead9c93b79ae7c1591b1fb5323bd777e86e150d4"
 62	governanceAddressConst = ethCommon.HexToAddress(governanceAddressStr)
 63
 64	auxAddressSK    = "28d1bfbbafe9d1d4f5a11c3c16ab6bf9084de48d99fbac4058bdfa3c80b29089"
 65	auxAddressStr   = "0x3d91185a02774C70287F6c74Dd26d13DFB58ff16"
 66	auxAddressConst = ethCommon.HexToAddress(auxAddressStr)
 67
 68	aux2AddressSK = "28d1bfbbafe9d1d4f5a11c3c16ab6bf9084de48d99fbac4058bdfa3c80b29087"
 69	// aux2AddressStr = "0x532792b73c0c6e7565912e7039c59986f7e1dd1f"
 70	// aux2AddressConst = ethCommon.HexToAddress(aux2AddressStr)
 71
 72	hermezRollupTestSK           = "28d1bfbbafe9d1d4f5a11c3c16ab6bf9084de48d99fbac4058bdfa3c80b29088"
 73	hermezRollupTestAddressStr   = "0xEa960515F8b4C237730F028cBAcF0a28E7F45dE0"
 74	hermezRollupAddressTestConst = ethCommon.HexToAddress(hermezRollupTestAddressStr)
 75)
 76
 77var (
 78	accountGov           *accounts.Account
 79	accountKep           *accounts.Account
 80	accountWhite         *accounts.Account
 81	accountGovDAO        *accounts.Account
 82	accountAux           *accounts.Account
 83	accountAux2          *accounts.Account
 84	accountHermez        *accounts.Account
 85	ks                   *keystore.KeyStore
 86	ethClient            *ethclient.Client
 87	ethereumClientWhite  *EthereumClient
 88	ethereumClientKep    *EthereumClient
 89	ethereumClientGovDAO *EthereumClient
 90	ethereumClientAux    *EthereumClient
 91	ethereumClientAux2   *EthereumClient
 92	ethereumClientHermez *EthereumClient
 93)
 94
 95func addKey(ks *keystore.KeyStore, skHex string) *accounts.Account {
 96	key, err := crypto.HexToECDSA(skHex)
 97	if err != nil {
 98		panic(err)
 99	}
100	account, err := ks.ImportECDSA(key, password)
101	if err != nil {
102		panic(err)
103	}
104	err = ks.Unlock(account, password)
105	if err != nil {
106		panic(err)
107	}
108	return &account
109}
110
111func getEnvVariables() {
112	err := godotenv.Load()
113	if err != nil {
114		fmt.Println("Variables loaded from environment")
115	} else {
116		fmt.Println("Variables loaded from .env file")
117	}
118	var auctionAddressStr = os.Getenv("AUCTION")
119	var auctionTestAddressStr = os.Getenv("AUCTION_TEST")
120	var tokenHEZAddressStr = os.Getenv("TOKENHEZ")
121	var hermezRollupAddressStr = os.Getenv("HERMEZ")
122	var wdelayerAddressStr = os.Getenv("WDELAYER")
123	var wdelayerTestAddressStr = os.Getenv("WDELAYER_TEST")
124	genesisBlockEnv := os.Getenv("GENESIS_BLOCK")
125	genesisBlock, err = strconv.ParseInt(genesisBlockEnv, 10, 64)
126	if err != nil {
127		panic(errEnvVar)
128	}
129	if auctionAddressStr == "" || auctionTestAddressStr == "" || tokenHEZAddressStr == "" || hermezRollupAddressStr == "" || wdelayerAddressStr == "" || wdelayerTestAddressStr == "" || genesisBlockEnv == "" {
130		panic(errEnvVar)
131	}
132
133	auctionAddressConst = ethCommon.HexToAddress(auctionAddressStr)
134	auctionTestAddressConst = ethCommon.HexToAddress(auctionTestAddressStr)
135	tokenHEZAddressConst = ethCommon.HexToAddress(tokenHEZAddressStr)
136	hermezRollupAddressConst = ethCommon.HexToAddress(hermezRollupAddressStr)
137	wdelayerAddressConst = ethCommon.HexToAddress(wdelayerAddressStr)
138	wdelayerTestAddressConst = ethCommon.HexToAddress(wdelayerTestAddressStr)
139	tokenHEZ = TokenConfig{
140		Address: tokenHEZAddressConst,
141		Name:    "Hermez Network Token",
142	}
143}
144
145func TestMain(m *testing.M) {
146	exitVal := 0
147
148	if os.Getenv("INTEGRATION") != "" {
149		getEnvVariables()
150		dir, err := ioutil.TempDir("", "tmpks")
151		if err != nil {
152			panic(err)
153		}
154		defer func() {
155			if err := os.RemoveAll(dir); err != nil {
156				panic(err)
157			}
158		}()
159		ks = keystore.NewKeyStore(dir, keystore.LightScryptN, keystore.LightScryptP)
160
161		// Load ethereum accounts from private keys
162		accountGov = addKey(ks, governanceAddressSK)
163		accountKep = addKey(ks, hermezKeeperAddressSK)
164		accountWhite = addKey(ks, whiteHackGroupAddressSK)
165		accountGovDAO = addKey(ks, hermezGovernanceDAOAddressSK)
166		accountAux = addKey(ks, auxAddressSK)
167		accountAux2 = addKey(ks, aux2AddressSK)
168		accountHermez = addKey(ks, hermezRollupTestSK)
169
170		ethClient, err = ethclient.Dial(ethClientDialURL)
171		if err != nil {
172			panic(err)
173		}
174
175		// Controllable Governance Address
176		ethereumClientGov := NewEthereumClient(ethClient, accountGov, ks, nil)
177		auctionClient, err = NewAuctionClient(ethereumClientGov, auctionAddressConst, tokenHEZ)
178		if err != nil {
179			panic(err)
180		}
181		auctionClientTest, err = NewAuctionClient(ethereumClientGov, auctionTestAddressConst, tokenHEZ)
182		if err != nil {
183			panic(err)
184		}
185		rollupClient, err = NewRollupClient(ethereumClientGov, hermezRollupAddressConst, tokenHEZ)
186		if err != nil {
187			panic(err)
188		}
189		wdelayerClient, err = NewWDelayerClient(ethereumClientGov, wdelayerAddressConst)
190		if err != nil {
191			panic(err)
192		}
193		wdelayerClientTest, err = NewWDelayerClient(ethereumClientGov, wdelayerTestAddressConst)
194		if err != nil {
195			panic(err)
196		}
197
198		ethereumClientKep = NewEthereumClient(ethClient, accountKep, ks, nil)
199		ethereumClientWhite = NewEthereumClient(ethClient, accountWhite, ks, nil)
200		ethereumClientGovDAO = NewEthereumClient(ethClient, accountGovDAO, ks, nil)
201		ethereumClientAux = NewEthereumClient(ethClient, accountAux, ks, nil)
202		ethereumClientAux2 = NewEthereumClient(ethClient, accountAux2, ks, nil)
203		ethereumClientHermez = NewEthereumClient(ethClient, accountHermez, ks, nil)
204
205		exitVal = m.Run()
206	}
207	os.Exit(exitVal)
208}