mirror of https://github.com/poanetwork/quorum.git
254 lines
7.0 KiB
Go
254 lines
7.0 KiB
Go
|
package natspec
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/ethereum/go-ethereum/accounts"
|
||
|
"github.com/ethereum/go-ethereum/common"
|
||
|
"github.com/ethereum/go-ethereum/common/docserver"
|
||
|
"github.com/ethereum/go-ethereum/common/registrar"
|
||
|
"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"
|
||
|
xe "github.com/ethereum/go-ethereum/xeth"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
testBalance = "10000000000000000000"
|
||
|
|
||
|
testFileName = "long_file_name_for_testing_registration_of_URLs_longer_than_32_bytes.content"
|
||
|
|
||
|
testNotice = "Register key `utils.toHex(_key)` <- content `utils.toHex(_content)`"
|
||
|
|
||
|
testExpNotice = "Register key 0xadd1a7d961cff0242089674ec2ef6fca671ab15e1fe80e38859fc815b98d88ab <- content 0xb3a2dea218de5d8bbe6c4645aadbf67b5ab00ecb1a9ec95dbdad6a0eed3e41a7"
|
||
|
|
||
|
testExpNotice2 = `About to submit transaction (NatSpec notice error: abi key does not match any method): {"params":[{"to":"%s","data": "0x31e12c20"}]}`
|
||
|
|
||
|
testExpNotice3 = `About to submit transaction (no NatSpec info found for contract: content hash not found for '0x1392c62d05b2d149e22a339c531157ae06b44d39a674cce500064b12b9aeb019'): {"params":[{"to":"%s","data": "0x300a3bbfb3a2dea218de5d8bbe6c4645aadbf67b5ab00ecb1a9ec95dbdad6a0eed3e41a7000000000000000000000000000000000000000000000000000000000000000000000000000000000000000066696c653a2f2f2f746573742e636f6e74656e74"}]}`
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
testUserDoc = `
|
||
|
{
|
||
|
"methods": {
|
||
|
"register(uint256,uint256)": {
|
||
|
"notice": "` + testNotice + `"
|
||
|
}
|
||
|
},
|
||
|
"invariants": [
|
||
|
{ "notice": "" }
|
||
|
],
|
||
|
"construction": [
|
||
|
{ "notice": "" }
|
||
|
]
|
||
|
}
|
||
|
`
|
||
|
testAbiDefinition = `
|
||
|
[{
|
||
|
"name": "register",
|
||
|
"constant": false,
|
||
|
"type": "function",
|
||
|
"inputs": [{
|
||
|
"name": "_key",
|
||
|
"type": "uint256"
|
||
|
}, {
|
||
|
"name": "_content",
|
||
|
"type": "uint256"
|
||
|
}],
|
||
|
"outputs": []
|
||
|
}]
|
||
|
`
|
||
|
|
||
|
testContractInfo = `
|
||
|
{
|
||
|
"userDoc": ` + testUserDoc + `,
|
||
|
"abiDefinition": ` + testAbiDefinition + `
|
||
|
}
|
||
|
`
|
||
|
)
|
||
|
|
||
|
type testFrontend struct {
|
||
|
t *testing.T
|
||
|
ethereum *eth.Ethereum
|
||
|
xeth *xe.XEth
|
||
|
coinbase common.Address
|
||
|
stateDb *state.StateDB
|
||
|
txc uint64
|
||
|
lastConfirm string
|
||
|
wantNatSpec bool
|
||
|
}
|
||
|
|
||
|
func (self *testFrontend) UnlockAccount(acc []byte) bool {
|
||
|
self.ethereum.AccountManager().Unlock(common.BytesToAddress(acc), "password")
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func (self *testFrontend) ConfirmTransaction(tx string) bool {
|
||
|
if self.wantNatSpec {
|
||
|
ds := docserver.New("/tmp/")
|
||
|
self.lastConfirm = GetNotice(self.xeth, tx, ds)
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func testEth(t *testing.T) (ethereum *eth.Ethereum, err error) {
|
||
|
|
||
|
os.RemoveAll("/tmp/eth-natspec/")
|
||
|
|
||
|
err = os.MkdirAll("/tmp/eth-natspec/keystore", os.ModePerm)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
// create a testAddress
|
||
|
ks := crypto.NewKeyStorePassphrase("/tmp/eth-natspec/keystore")
|
||
|
am := accounts.NewManager(ks)
|
||
|
testAccount, err := am.NewAccount("password")
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
testAddress := strings.TrimPrefix(testAccount.Address.Hex(), "0x")
|
||
|
|
||
|
// set up mock genesis with balance on the testAddress
|
||
|
core.GenesisAccounts = []byte(`{
|
||
|
"` + testAddress + `": {"balance": "` + testBalance + `"}
|
||
|
}`)
|
||
|
|
||
|
// only use minimalistic stack with no networking
|
||
|
ethereum, err = eth.New(ð.Config{
|
||
|
DataDir: "/tmp/eth-natspec",
|
||
|
AccountManager: am,
|
||
|
MaxPeers: 0,
|
||
|
})
|
||
|
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func testInit(t *testing.T) (self *testFrontend) {
|
||
|
// initialise and start minimal ethereum stack
|
||
|
ethereum, err := testEth(t)
|
||
|
if err != nil {
|
||
|
t.Errorf("error creating ethereum: %v", err)
|
||
|
return
|
||
|
}
|
||
|
err = ethereum.Start()
|
||
|
if err != nil {
|
||
|
t.Errorf("error starting ethereum: %v", err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// mock frontend
|
||
|
self = &testFrontend{t: t, ethereum: ethereum}
|
||
|
self.xeth = xe.New(ethereum, self)
|
||
|
|
||
|
addr, _ := ethereum.Etherbase()
|
||
|
self.coinbase = addr
|
||
|
self.stateDb = self.ethereum.ChainManager().State().Copy()
|
||
|
|
||
|
// initialise the registry contracts
|
||
|
reg := registrar.New(self.xeth)
|
||
|
err = reg.SetHashReg("", addr)
|
||
|
if err != nil {
|
||
|
t.Errorf("error creating HashReg: %v", err)
|
||
|
}
|
||
|
err = reg.SetUrlHint("", addr)
|
||
|
if err != nil {
|
||
|
t.Errorf("error creating UrlHint: %v", err)
|
||
|
}
|
||
|
self.applyTxs()
|
||
|
|
||
|
return
|
||
|
|
||
|
}
|
||
|
|
||
|
// this is needed for transaction to be applied to the state in testing
|
||
|
// the heavy lifing is done in XEth.ApplyTestTxs
|
||
|
// this is fragile,
|
||
|
// and does process leaking since xeth loops cannot quit safely
|
||
|
// should be replaced by proper mining with testDAG for easy full integration tests
|
||
|
func (self *testFrontend) applyTxs() {
|
||
|
self.txc, self.xeth = self.xeth.ApplyTestTxs(self.stateDb, self.coinbase, self.txc)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// end to end test
|
||
|
func TestNatspecE2E(t *testing.T) {
|
||
|
t.Skip()
|
||
|
|
||
|
tf := testInit(t)
|
||
|
defer tf.ethereum.Stop()
|
||
|
|
||
|
// create a contractInfo file (mock cloud-deployed contract metadocs)
|
||
|
// incidentally this is the info for the registry contract itself
|
||
|
ioutil.WriteFile("/tmp/"+testFileName, []byte(testContractInfo), os.ModePerm)
|
||
|
dochash := common.BytesToHash(crypto.Sha3([]byte(testContractInfo)))
|
||
|
|
||
|
// take the codehash for the contract we wanna test
|
||
|
// codehex := tf.xeth.CodeAt(registar.HashRegAddr)
|
||
|
codeb := tf.xeth.CodeAtBytes(registrar.HashRegAddr)
|
||
|
codehash := common.BytesToHash(crypto.Sha3(codeb))
|
||
|
|
||
|
// use resolver to register codehash->dochash->url
|
||
|
// test if globalregistry works
|
||
|
// registrar.HashRefAddr = "0x0"
|
||
|
// registrar.UrlHintAddr = "0x0"
|
||
|
reg := registrar.New(tf.xeth)
|
||
|
_, err := reg.SetHashToHash(tf.coinbase, codehash, dochash)
|
||
|
if err != nil {
|
||
|
t.Errorf("error registering: %v", err)
|
||
|
}
|
||
|
_, err = reg.SetUrlToHash(tf.coinbase, dochash, "file:///"+testFileName)
|
||
|
if err != nil {
|
||
|
t.Errorf("error registering: %v", err)
|
||
|
}
|
||
|
// apply txs to the state
|
||
|
tf.applyTxs()
|
||
|
|
||
|
// NatSpec info for register method of HashReg contract installed
|
||
|
// now using the same transactions to check confirm messages
|
||
|
|
||
|
tf.wantNatSpec = true // this is set so now the backend uses natspec confirmation
|
||
|
_, err = reg.SetHashToHash(tf.coinbase, codehash, dochash)
|
||
|
if err != nil {
|
||
|
t.Errorf("error calling contract registry: %v", err)
|
||
|
}
|
||
|
|
||
|
fmt.Printf("GlobalRegistrar: %v, HashReg: %v, UrlHint: %v\n", registrar.GlobalRegistrarAddr, registrar.HashRegAddr, registrar.UrlHintAddr)
|
||
|
if tf.lastConfirm != testExpNotice {
|
||
|
t.Errorf("Wrong confirm message. expected '%v', got '%v'", testExpNotice, tf.lastConfirm)
|
||
|
}
|
||
|
|
||
|
// test unknown method
|
||
|
exp := fmt.Sprintf(testExpNotice2, registrar.HashRegAddr)
|
||
|
_, err = reg.SetOwner(tf.coinbase)
|
||
|
if err != nil {
|
||
|
t.Errorf("error setting owner: %v", err)
|
||
|
}
|
||
|
|
||
|
if tf.lastConfirm != exp {
|
||
|
t.Errorf("Wrong confirm message, expected '%v', got '%v'", exp, tf.lastConfirm)
|
||
|
}
|
||
|
|
||
|
// test unknown contract
|
||
|
exp = fmt.Sprintf(testExpNotice3, registrar.UrlHintAddr)
|
||
|
|
||
|
_, err = reg.SetUrlToHash(tf.coinbase, dochash, "file:///test.content")
|
||
|
if err != nil {
|
||
|
t.Errorf("error registering: %v", err)
|
||
|
}
|
||
|
|
||
|
if tf.lastConfirm != exp {
|
||
|
t.Errorf("Wrong confirm message, expected '%v', got '%v'", exp, tf.lastConfirm)
|
||
|
}
|
||
|
|
||
|
}
|