383 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			383 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package main
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"io/ioutil"
 | |
| 	"os"
 | |
| 	"path/filepath"
 | |
| 	"regexp"
 | |
| 	"runtime"
 | |
| 	"strconv"
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/ethereum/go-ethereum/accounts"
 | |
| 	"github.com/ethereum/go-ethereum/common"
 | |
| 	"github.com/ethereum/go-ethereum/common/compiler"
 | |
| 	"github.com/ethereum/go-ethereum/common/docserver"
 | |
| 	"github.com/ethereum/go-ethereum/common/natspec"
 | |
| 	"github.com/ethereum/go-ethereum/common/resolver"
 | |
| 	"github.com/ethereum/go-ethereum/core"
 | |
| 	"github.com/ethereum/go-ethereum/core/state"
 | |
| 	"github.com/ethereum/go-ethereum/crypto"
 | |
| 	"github.com/ethereum/go-ethereum/eth"
 | |
| )
 | |
| 
 | |
| const (
 | |
| 	testSolcPath = ""
 | |
| 	solcVersion  = "0.9.23"
 | |
| 
 | |
| 	testKey     = "e6fab74a43941f82d89cb7faa408e227cdad3153c4720e540e855c19b15e6674"
 | |
| 	testAddress = "0x8605cdbbdb6d264aa742e77020dcbc58fcdce182"
 | |
| 	testBalance = "10000000000000000000"
 | |
| 	// of empty string
 | |
| 	testHash = "0xc5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470"
 | |
| )
 | |
| 
 | |
| var (
 | |
| 	versionRE   = regexp.MustCompile(strconv.Quote(`"compilerVersion":"` + solcVersion + `"`))
 | |
| 	testNodeKey = crypto.ToECDSA(common.Hex2Bytes("4b50fa71f5c3eeb8fdc452224b2395af2fcc3d125e06c32c82e048c0559db03f"))
 | |
| 	testGenesis = `{"` + testAddress[2:] + `": {"balance": "` + testBalance + `"}}`
 | |
| )
 | |
| 
 | |
| type testjethre struct {
 | |
| 	*jsre
 | |
| 	stateDb     *state.StateDB
 | |
| 	lastConfirm string
 | |
| 	ds          *docserver.DocServer
 | |
| }
 | |
| 
 | |
| func (self *testjethre) UnlockAccount(acc []byte) bool {
 | |
| 	err := self.ethereum.AccountManager().Unlock(common.BytesToAddress(acc), "")
 | |
| 	if err != nil {
 | |
| 		panic("unable to unlock")
 | |
| 	}
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| func (self *testjethre) ConfirmTransaction(tx string) bool {
 | |
| 	if self.ethereum.NatSpec {
 | |
| 		self.lastConfirm = natspec.GetNotice(self.xeth, tx, self.ds)
 | |
| 	}
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| func testJEthRE(t *testing.T) (string, *testjethre, *eth.Ethereum) {
 | |
| 	tmp, err := ioutil.TempDir("", "geth-test")
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	// set up mock genesis with balance on the testAddress
 | |
| 	core.GenesisAccounts = []byte(testGenesis)
 | |
| 
 | |
| 	ks := crypto.NewKeyStorePlain(filepath.Join(tmp, "keystore"))
 | |
| 	am := accounts.NewManager(ks)
 | |
| 	ethereum, err := eth.New(ð.Config{
 | |
| 		NodeKey:        testNodeKey,
 | |
| 		DataDir:        tmp,
 | |
| 		AccountManager: am,
 | |
| 		MaxPeers:       0,
 | |
| 		Name:           "test",
 | |
| 		SolcPath:       testSolcPath,
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		t.Fatal("%v", err)
 | |
| 	}
 | |
| 
 | |
| 	keyb, err := crypto.HexToECDSA(testKey)
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	key := crypto.NewKeyFromECDSA(keyb)
 | |
| 	err = ks.StoreKey(key, "")
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	err = am.Unlock(key.Address, "")
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	assetPath := filepath.Join(os.Getenv("GOPATH"), "src", "github.com", "ethereum", "go-ethereum", "cmd", "mist", "assets", "ext")
 | |
| 	ds, err := docserver.New("/")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("Error creating DocServer: %v", err)
 | |
| 	}
 | |
| 	tf := &testjethre{ds: ds, stateDb: ethereum.ChainManager().State().Copy()}
 | |
| 	repl := newJSRE(ethereum, assetPath, "", "", false, tf)
 | |
| 	tf.jsre = repl
 | |
| 	return tmp, tf, ethereum
 | |
| }
 | |
| 
 | |
| // this line below is needed for transaction to be applied to the state in testing
 | |
| // the heavy lifing is done in XEth.ApplyTestTxs
 | |
| // this is fragile, overwriting xeth will result in
 | |
| // process leaking since xeth loops cannot quit safely
 | |
| // should be replaced by proper mining with testDAG for easy full integration tests
 | |
| // txc, self.xeth = self.xeth.ApplyTestTxs(self.xeth.repl.stateDb, coinbase, txc)
 | |
| 
 | |
| func TestNodeInfo(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Fatalf("error starting ethereum: %v", err)
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 	want := `{"DiscPort":0,"IP":"0.0.0.0","ListenAddr":"","Name":"test","NodeID":"4cb2fc32924e94277bf94b5e4c983beedb2eabd5a0bc941db32202735c6625d020ca14a5963d1738af43b6ac0a711d61b1a06de931a499fe2aa0b1a132a902b5","NodeUrl":"enode://4cb2fc32924e94277bf94b5e4c983beedb2eabd5a0bc941db32202735c6625d020ca14a5963d1738af43b6ac0a711d61b1a06de931a499fe2aa0b1a132a902b5@0.0.0.0:0","TCPPort":0,"Td":"131072"}`
 | |
| 	checkEvalJSON(t, repl, `admin.nodeInfo()`, want)
 | |
| }
 | |
| 
 | |
| func TestAccounts(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Fatalf("error starting ethereum: %v", err)
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `eth.accounts`, `["`+testAddress+`"]`)
 | |
| 	checkEvalJSON(t, repl, `eth.coinbase`, `"`+testAddress+`"`)
 | |
| 
 | |
| 	val, err := repl.re.Run(`admin.newAccount("password")`)
 | |
| 	if err != nil {
 | |
| 		t.Errorf("expected no error, got %v", err)
 | |
| 	}
 | |
| 	addr := val.String()
 | |
| 	if !regexp.MustCompile(`0x[0-9a-f]{40}`).MatchString(addr) {
 | |
| 		t.Errorf("address not hex: %q", addr)
 | |
| 	}
 | |
| 
 | |
| 	// skip until order fixed #824
 | |
| 	// checkEvalJSON(t, repl, `eth.accounts`, `["`+testAddress+`", "`+addr+`"]`)
 | |
| 	// checkEvalJSON(t, repl, `eth.coinbase`, `"`+testAddress+`"`)
 | |
| }
 | |
| 
 | |
| func TestBlockChain(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Fatalf("error starting ethereum: %v", err)
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 	// get current block dump before export/import.
 | |
| 	val, err := repl.re.Run("JSON.stringify(admin.debug.dumpBlock())")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("expected no error, got %v", err)
 | |
| 	}
 | |
| 	beforeExport := val.String()
 | |
| 
 | |
| 	// do the export
 | |
| 	extmp, err := ioutil.TempDir("", "geth-test-export")
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	defer os.RemoveAll(extmp)
 | |
| 	tmpfile := filepath.Join(extmp, "export.chain")
 | |
| 	tmpfileq := strconv.Quote(tmpfile)
 | |
| 
 | |
| 	ethereum.ChainManager().Reset()
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `admin.export(`+tmpfileq+`)`, `true`)
 | |
| 	if _, err := os.Stat(tmpfile); err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	// check import, verify that dumpBlock gives the same result.
 | |
| 	checkEvalJSON(t, repl, `admin.import(`+tmpfileq+`)`, `true`)
 | |
| 	checkEvalJSON(t, repl, `admin.debug.dumpBlock()`, beforeExport)
 | |
| }
 | |
| 
 | |
| func TestMining(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Fatalf("error starting ethereum: %v", err)
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 	checkEvalJSON(t, repl, `eth.mining`, `false`)
 | |
| }
 | |
| 
 | |
| func TestRPC(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Errorf("error starting ethereum: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `admin.startRPC("127.0.0.1", 5004)`, `true`)
 | |
| }
 | |
| 
 | |
| func TestCheckTestAccountBalance(t *testing.T) {
 | |
| 	t.Skip() // i don't think it tests the correct behaviour here. it's actually testing
 | |
| 	// internals which shouldn't be tested. This now fails because of a change in the core
 | |
| 	// and i have no means to fix this, sorry - @obscuren
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Errorf("error starting ethereum: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 
 | |
| 	repl.re.Run(`primary = "` + testAddress + `"`)
 | |
| 	checkEvalJSON(t, repl, `eth.getBalance(primary)`, `"`+testBalance+`"`)
 | |
| }
 | |
| 
 | |
| func TestSignature(t *testing.T) {
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Errorf("error starting ethereum: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 
 | |
| 	val, err := repl.re.Run(`eth.sign("` + testAddress + `", "` + testHash + `")`)
 | |
| 
 | |
| 	// This is a very preliminary test, lacking actual signature verification
 | |
| 	if err != nil {
 | |
| 		t.Errorf("Error running js: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 	output := val.String()
 | |
| 	t.Logf("Output: %v", output)
 | |
| 
 | |
| 	regex := regexp.MustCompile(`^0x[0-9a-f]{130}$`)
 | |
| 	if !regex.MatchString(output) {
 | |
| 		t.Errorf("Signature is not 65 bytes represented in hexadecimal.")
 | |
| 		return
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestContract(t *testing.T) {
 | |
| 	t.Skip()
 | |
| 	tmp, repl, ethereum := testJEthRE(t)
 | |
| 	if err := ethereum.Start(); err != nil {
 | |
| 		t.Errorf("error starting ethereum: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer ethereum.Stop()
 | |
| 	defer os.RemoveAll(tmp)
 | |
| 
 | |
| 	var txc uint64
 | |
| 	coinbase := common.HexToAddress(testAddress)
 | |
| 	resolver.New(repl.xeth).CreateContracts(coinbase)
 | |
| 	// time.Sleep(1000 * time.Millisecond)
 | |
| 
 | |
| 	// checkEvalJSON(t, repl, `eth.getBlock("pending", true).transactions.length`, `2`)
 | |
| 	source := `contract test {\n` +
 | |
| 		"   /// @notice Will multiply `a` by 7." + `\n` +
 | |
| 		`   function multiply(uint a) returns(uint d) {\n` +
 | |
| 		`       return a * 7;\n` +
 | |
| 		`   }\n` +
 | |
| 		`}\n`
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `admin.contractInfo.stop()`, `true`)
 | |
| 
 | |
| 	contractInfo, err := ioutil.ReadFile("info_test.json")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	checkEvalJSON(t, repl, `primary = eth.accounts[0]`, `"`+testAddress+`"`)
 | |
| 	checkEvalJSON(t, repl, `source = "`+source+`"`, `"`+source+`"`)
 | |
| 
 | |
| 	// if solc is found with right version, test it, otherwise read from file
 | |
| 	sol, err := compiler.New("")
 | |
| 	if err != nil {
 | |
| 		t.Logf("solc not found: mocking contract compilation step")
 | |
| 	} else if sol.Version() != solcVersion {
 | |
| 		t.Logf("WARNING: solc different version found (%v, test written for %v, may need to update)", sol.Version(), solcVersion)
 | |
| 	}
 | |
| 
 | |
| 	if err != nil {
 | |
| 		info, err := ioutil.ReadFile("info_test.json")
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("%v", err)
 | |
| 		}
 | |
| 		_, err = repl.re.Run(`contract = JSON.parse(` + strconv.Quote(string(info)) + `)`)
 | |
| 		if err != nil {
 | |
| 			t.Errorf("%v", err)
 | |
| 		}
 | |
| 	} else {
 | |
| 		checkEvalJSON(t, repl, `contract = eth.compile.solidity(source).test`, string(contractInfo))
 | |
| 	}
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `contract.code`, `"0x605880600c6000396000f3006000357c010000000000000000000000000000000000000000000000000000000090048063c6888fa114602e57005b603d6004803590602001506047565b8060005260206000f35b60006007820290506053565b91905056"`)
 | |
| 
 | |
| 	checkEvalJSON(
 | |
| 		t, repl,
 | |
| 		`contractaddress = eth.sendTransaction({from: primary, data: contract.code })`,
 | |
| 		`"0x5dcaace5982778b409c524873b319667eba5d074"`,
 | |
| 	)
 | |
| 
 | |
| 	callSetup := `abiDef = JSON.parse('[{"constant":false,"inputs":[{"name":"a","type":"uint256"}],"name":"multiply","outputs":[{"name":"d","type":"uint256"}],"type":"function"}]');
 | |
| Multiply7 = eth.contract(abiDef);
 | |
| multiply7 = Multiply7.at(contractaddress);
 | |
| `
 | |
| 	// time.Sleep(1500 * time.Millisecond)
 | |
| 	_, err = repl.re.Run(callSetup)
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error setting up contract, got %v", err)
 | |
| 	}
 | |
| 
 | |
| 	// checkEvalJSON(t, repl, `eth.getBlock("pending", true).transactions.length`, `3`)
 | |
| 
 | |
| 	// why is this sometimes failing?
 | |
| 	// checkEvalJSON(t, repl, `multiply7.multiply.call(6)`, `42`)
 | |
| 	expNotice := ""
 | |
| 	if repl.lastConfirm != expNotice {
 | |
| 		t.Errorf("incorrect confirmation message: expected %v, got %v", expNotice, repl.lastConfirm)
 | |
| 	}
 | |
| 
 | |
| 	txc, repl.xeth = repl.xeth.ApplyTestTxs(repl.stateDb, coinbase, txc)
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `admin.contractInfo.start()`, `true`)
 | |
| 	checkEvalJSON(t, repl, `multiply7.multiply.sendTransaction(6, { from: primary, gas: "1000000", gasPrice: "100000" })`, `undefined`)
 | |
| 	expNotice = `About to submit transaction (no NatSpec info found for contract: content hash not found for '0x87e2802265838c7f14bb69eecd2112911af6767907a702eeaa445239fb20711b'): {"params":[{"to":"0x5dcaace5982778b409c524873b319667eba5d074","data": "0xc6888fa10000000000000000000000000000000000000000000000000000000000000006"}]}`
 | |
| 	if repl.lastConfirm != expNotice {
 | |
| 		t.Errorf("incorrect confirmation message: expected %v, got %v", expNotice, repl.lastConfirm)
 | |
| 	}
 | |
| 
 | |
| 	var contenthash = `"0x86d2b7cf1e72e9a7a3f8d96601f0151742a2f780f1526414304fbe413dc7f9bd"`
 | |
| 	if sol != nil {
 | |
| 		modContractInfo := versionRE.ReplaceAll(contractInfo, []byte(`"compilerVersion":"`+sol.Version()+`"`))
 | |
| 		_ = modContractInfo
 | |
| 		// contenthash = crypto.Sha3(modContractInfo)
 | |
| 	}
 | |
| 	checkEvalJSON(t, repl, `filename = "/tmp/info.json"`, `"/tmp/info.json"`)
 | |
| 	checkEvalJSON(t, repl, `contenthash = admin.contractInfo.register(primary, contractaddress, contract, filename)`, contenthash)
 | |
| 	checkEvalJSON(t, repl, `admin.contractInfo.registerUrl(primary, contenthash, "file://"+filename)`, `true`)
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error registering, got %v", err)
 | |
| 	}
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `admin.contractInfo.start()`, `true`)
 | |
| 
 | |
| 	// update state
 | |
| 	txc, repl.xeth = repl.xeth.ApplyTestTxs(repl.stateDb, coinbase, txc)
 | |
| 
 | |
| 	checkEvalJSON(t, repl, `multiply7.multiply.sendTransaction(6, { from: primary, gas: "1000000", gasPrice: "100000" })`, `undefined`)
 | |
| 	expNotice = "Will multiply 6 by 7."
 | |
| 	if repl.lastConfirm != expNotice {
 | |
| 		t.Errorf("incorrect confirmation message: expected %v, got %v", expNotice, repl.lastConfirm)
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| func checkEvalJSON(t *testing.T, re *testjethre, expr, want string) error {
 | |
| 	val, err := re.re.Run("JSON.stringify(" + expr + ")")
 | |
| 	if err == nil && val.String() != want {
 | |
| 		err = fmt.Errorf("Output mismatch for `%s`:\ngot:  %s\nwant: %s", expr, val.String(), want)
 | |
| 	}
 | |
| 	if err != nil {
 | |
| 		_, file, line, _ := runtime.Caller(1)
 | |
| 		file = filepath.Base(file)
 | |
| 		fmt.Printf("\t%s:%d: %v\n", file, line, err)
 | |
| 		t.Fail()
 | |
| 	}
 | |
| 	return err
 | |
| }
 |