2022-02-22 08:55:04 +00:00
|
|
|
# Nimbus
|
|
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
|
|
# Licensed under either of
|
|
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
|
|
# at your option.
|
|
|
|
# This file may not be copied, modified, or distributed except according to
|
|
|
|
# those terms.
|
|
|
|
|
|
|
|
import
|
2022-07-12 08:56:01 +00:00
|
|
|
std/[json, os, sets, strutils, typetraits],
|
|
|
|
unittest2, eth/common as eth_common,
|
|
|
|
json_rpc/[rpcserver, rpcclient], web3/[engine_api_types],
|
2022-02-22 08:55:04 +00:00
|
|
|
eth/[trie/db, p2p/private/p2p_types],
|
2022-05-13 16:30:10 +00:00
|
|
|
../nimbus/sync/protocol,
|
2022-07-12 08:56:01 +00:00
|
|
|
../nimbus/rpc/[p2p, engine_api],
|
2022-02-22 08:55:04 +00:00
|
|
|
../nimbus/db/[db_chain],
|
2022-07-12 08:56:01 +00:00
|
|
|
../nimbus/[config, context, genesis, sealer],
|
2022-02-22 08:55:04 +00:00
|
|
|
../nimbus/utils/[tx_pool],
|
2022-03-21 17:12:07 +00:00
|
|
|
../nimbus/p2p/chain,
|
2022-02-22 08:55:04 +00:00
|
|
|
../nimbus/merge/mergetypes,
|
|
|
|
./test_helpers
|
|
|
|
|
|
|
|
const
|
|
|
|
baseDir = "tests" / "merge"
|
|
|
|
paramsFile = baseDir / "params.json"
|
|
|
|
stepsFile = baseDir / "steps.json"
|
|
|
|
|
|
|
|
type
|
|
|
|
Step = ref object
|
|
|
|
name: string
|
|
|
|
meth: string
|
|
|
|
params: JSonNode
|
|
|
|
expect: JsonNode
|
|
|
|
error : bool
|
|
|
|
|
|
|
|
Steps = ref object
|
|
|
|
list: seq[Step]
|
|
|
|
|
|
|
|
proc parseStep(s: Step, node: JsonNode) =
|
|
|
|
for k, v in node:
|
|
|
|
case k
|
|
|
|
of "name": s.name = v.getStr()
|
|
|
|
of "method": s.meth = v.getStr()
|
|
|
|
of "params": s.params = v
|
|
|
|
of "expect": s.expect = v
|
|
|
|
of "error": s.error = true
|
|
|
|
else:
|
|
|
|
doAssert(false, "unknown key: " & k)
|
|
|
|
|
|
|
|
proc parseSteps(node: JsonNode): Steps =
|
|
|
|
let ss = Steps(list: @[])
|
|
|
|
for n in node:
|
|
|
|
let s = Step()
|
|
|
|
s.parseStep(n)
|
|
|
|
ss.list.add s
|
|
|
|
ss
|
|
|
|
|
|
|
|
proc forkChoiceUpdate(step: Step, client: RpcClient, testStatusIMPL: var TestStatus) =
|
|
|
|
let arg = step.params[1]
|
|
|
|
if arg.kind == JNull:
|
|
|
|
step.params.elems.setLen(1)
|
|
|
|
|
|
|
|
let res = waitFor client.call(step.meth, step.params)
|
|
|
|
check toLowerAscii($res) == toLowerAscii($step.expect)
|
|
|
|
|
|
|
|
proc getPayload(step: Step, client: RpcClient, testStatusIMPL: var TestStatus) =
|
|
|
|
try:
|
|
|
|
let res = waitFor client.call(step.meth, step.params)
|
|
|
|
check toLowerAscii($res) == toLowerAscii($step.expect)
|
|
|
|
except:
|
|
|
|
check step.error == true
|
|
|
|
|
|
|
|
proc newPayload(step: Step, client: RpcClient, testStatusIMPL: var TestStatus) =
|
|
|
|
let res = waitFor client.call(step.meth, step.params)
|
|
|
|
check toLowerAscii($res) == toLowerAscii($step.expect)
|
|
|
|
|
|
|
|
proc runTest(steps: Steps) =
|
|
|
|
let
|
|
|
|
conf = makeConfig(@["--custom-network:" & paramsFile])
|
|
|
|
ctx = newEthContext()
|
|
|
|
ethNode = setupEthNode(conf, ctx, eth)
|
|
|
|
chainDB = newBaseChainDB(
|
|
|
|
newMemoryDb(),
|
|
|
|
conf.pruneMode == PruneMode.Full,
|
|
|
|
conf.networkId,
|
|
|
|
conf.networkParams
|
|
|
|
)
|
|
|
|
chainRef = newChain(chainDB)
|
|
|
|
|
|
|
|
initializeEmptyDb(chainDB)
|
|
|
|
|
|
|
|
var
|
|
|
|
rpcServer = newRpcSocketServer(["localhost:" & $conf.rpcPort])
|
|
|
|
client = newRpcSocketClient()
|
|
|
|
txPool = TxPoolRef.new(chainDB, conf.engineSigner)
|
|
|
|
sealingEngine = SealingEngineRef.new(
|
|
|
|
chainRef, ctx, conf.engineSigner,
|
|
|
|
txPool, EnginePostMerge
|
|
|
|
)
|
|
|
|
|
|
|
|
setupEthRpc(ethNode, ctx, chainDB, txPool, rpcServer)
|
|
|
|
setupEngineAPI(sealingEngine, rpcServer)
|
|
|
|
|
|
|
|
sealingEngine.start()
|
|
|
|
rpcServer.start()
|
2022-10-10 02:31:28 +00:00
|
|
|
waitFor client.connect("localhost", conf.rpcPort)
|
2022-02-22 08:55:04 +00:00
|
|
|
|
|
|
|
suite "Engine API tests":
|
|
|
|
for i, step in steps.list:
|
|
|
|
test $i & " " & step.name:
|
|
|
|
case step.meth
|
|
|
|
of "engine_forkchoiceUpdatedV1":
|
|
|
|
forkChoiceUpdate(step, client, testStatusIMPL)
|
|
|
|
of "engine_getPayloadV1":
|
|
|
|
getPayload(step, client, testStatusIMPL)
|
|
|
|
of "engine_newPayloadV1":
|
|
|
|
newPayload(step, client, testStatusIMPL)
|
|
|
|
else:
|
|
|
|
doAssert(false, "unknown method: " & step.meth)
|
|
|
|
|
|
|
|
waitFor client.close()
|
|
|
|
waitFor sealingEngine.stop()
|
|
|
|
rpcServer.stop()
|
|
|
|
waitFor rpcServer.closeWait()
|
|
|
|
|
|
|
|
proc testEngineAPI() =
|
|
|
|
let node = parseJSON(readFile(stepsFile))
|
|
|
|
let steps = parseSteps(node)
|
|
|
|
runTest(steps)
|
|
|
|
|
|
|
|
proc toId(x: int): PayloadId =
|
|
|
|
var id: distinctBase PayloadId
|
|
|
|
id[^1] = x.byte
|
|
|
|
PayloadId(id)
|
|
|
|
|
|
|
|
proc `==`(a, b: Quantity): bool =
|
|
|
|
uint64(a) == uint64(b)
|
|
|
|
|
2022-03-11 09:25:23 +00:00
|
|
|
proc testEngineApiSupport() =
|
2022-02-22 08:55:04 +00:00
|
|
|
var api = EngineAPI.new()
|
|
|
|
let
|
|
|
|
id1 = toId(1)
|
|
|
|
id2 = toId(2)
|
|
|
|
ep1 = ExecutionPayloadV1(gasLimit: Quantity 100)
|
|
|
|
ep2 = ExecutionPayloadV1(gasLimit: Quantity 101)
|
|
|
|
hdr1 = EthBlockHeader(gasLimit: 100)
|
|
|
|
hdr2 = EthBlockHeader(gasLimit: 101)
|
|
|
|
hash1 = hdr1.blockHash
|
|
|
|
hash2 = hdr2.blockHash
|
|
|
|
|
|
|
|
suite "Test engine api support":
|
|
|
|
test "test payload queue":
|
|
|
|
api.put(id1, ep1)
|
|
|
|
api.put(id2, ep2)
|
|
|
|
var eep1, eep2: ExecutionPayloadV1
|
|
|
|
check api.get(id1, eep1)
|
|
|
|
check api.get(id2, eep2)
|
|
|
|
check eep1.gasLimit == ep1.gasLimit
|
|
|
|
check eep2.gasLimit == ep2.gasLimit
|
|
|
|
|
|
|
|
test "test header queue":
|
|
|
|
api.put(hash1, hdr1)
|
|
|
|
api.put(hash2, hdr2)
|
|
|
|
var eh1, eh2: EthBlockHeader
|
|
|
|
check api.get(hash1, eh1)
|
|
|
|
check api.get(hash2, eh2)
|
|
|
|
check eh1.gasLimit == hdr1.gasLimit
|
|
|
|
check eh2.gasLimit == hdr2.gasLimit
|
|
|
|
|
|
|
|
proc mergeMain*() =
|
2022-03-11 09:25:23 +00:00
|
|
|
# temporary disable it until engine API more stable
|
2022-03-17 05:54:04 +00:00
|
|
|
testEngineAPI()
|
2022-03-11 09:25:23 +00:00
|
|
|
testEngineApiSupport()
|
2022-02-22 08:55:04 +00:00
|
|
|
|
|
|
|
when isMainModule:
|
|
|
|
mergeMain()
|