nimbus-eth1/tests/test_graphql.nim

111 lines
2.9 KiB
Nim
Raw Normal View History

2021-04-24 04:01:56 +00:00
# nim-graphql
# Copyright (c) 2021-2024 Status Research & Development GmbH
2021-04-24 04:01:56 +00:00
# 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
std/[json],
2023-06-12 04:29:03 +00:00
stew/byteutils,
2022-12-02 04:39:12 +00:00
eth/[p2p, rlp],
2021-04-24 04:01:56 +00:00
graphql, ../nimbus/graphql/ethapi, graphql/test_common,
../nimbus/sync/protocol,
2022-12-02 04:39:12 +00:00
../nimbus/config,
../nimbus/core/[chain, tx_pool],
../nimbus/common/[common, context],
./test_helpers
2021-04-24 04:01:56 +00:00
type
EthBlock = object
header: BlockHeader
transactions: seq[Transaction]
uncles: seq[BlockHeader]
const
caseFolder = "tests/graphql"
dataFolder = "tests/fixtures/eth_tests/BlockchainTests/ValidBlocks/bcUncleTest"
2021-04-24 04:01:56 +00:00
proc toBlock(n: JsonNode, key: string): EthBlock =
let rlpBlob = hexToSeqByte(n[key].str)
rlp.decode(rlpBlob, EthBlock)
2022-12-02 04:39:12 +00:00
proc setupChain(): CommonRef =
let config = ChainConfig(
chainId : MainNet.ChainId,
2022-12-02 04:39:12 +00:00
byzantiumBlock : some(0.toBlockNumber),
constantinopleBlock : some(0.toBlockNumber),
petersburgBlock : some(0.toBlockNumber),
istanbulBlock : some(0.toBlockNumber),
muirGlacierBlock : some(0.toBlockNumber),
berlinBlock : some(10.toBlockNumber)
)
var jn = json.parseFile(dataFolder & "/oneUncle.json")
2021-04-24 04:01:56 +00:00
for k, v in jn:
if v["network"].str == "Istanbul":
jn = v
break
let gen = jn.toBlock("genesisRLP")
var genesis = Genesis(
nonce : gen.header.nonce,
extraData : gen.header.extraData,
gasLimit : gen.header.gasLimit,
difficulty: gen.header.difficulty,
mixHash : gen.header.mixDigest,
coinBase : gen.header.coinbase,
timestamp : gen.header.timestamp,
baseFeePerGas: gen.header.fee
)
if not parseGenesisAlloc($(jn["pre"]), genesis.alloc):
quit(QuitFailure)
let customNetwork = NetworkParams(
config: config,
genesis: genesis
)
2022-12-02 04:39:12 +00:00
let com = CommonRef.new(
newCoreDbRef DefaultDbMemory,
CustomNet,
customNetwork
)
2022-12-02 04:39:12 +00:00
com.initializeEmptyDb()
2021-04-24 04:01:56 +00:00
let blocks = jn["blocks"]
var headers: seq[BlockHeader]
var bodies: seq[BlockBody]
for n in blocks:
let ethBlock = n.toBlock("rlp")
headers.add ethBlock.header
bodies.add BlockBody(
transactions: ethBlock.transactions,
uncles: ethBlock.uncles
)
2022-12-02 04:39:12 +00:00
let chain = newChain(com)
2021-04-24 04:01:56 +00:00
let res = chain.persistBlocks(headers, bodies)
assert res.isOk(), res.error()
2021-04-24 04:01:56 +00:00
2022-12-02 04:39:12 +00:00
com
2021-04-24 04:01:56 +00:00
proc graphqlMain*() =
2021-04-24 04:01:56 +00:00
let
conf = makeTestConfig()
ethCtx = newEthContext()
ethNode = setupEthNode(conf, ethCtx, eth)
2022-12-02 04:39:12 +00:00
com = setupChain()
txPool = TxPoolRef.new(com, conf.engineSigner)
2021-04-24 04:01:56 +00:00
2022-12-02 04:39:12 +00:00
let ctx = setupGraphqlContext(com, ethNode, txPool)
2021-04-24 04:01:56 +00:00
when isMainModule:
ctx.main(caseFolder, purgeSchema = false)
else:
ctx.executeCases(caseFolder, purgeSchema = false)
when isMainModule:
graphqlMain()