70 lines
2.4 KiB
Nim
70 lines
2.4 KiB
Nim
|
# Nimbus
|
||
|
|
# Copyright (c) 2018-2024 Status Research & Development GmbH
|
||
|
|
# Licensed under either of
|
||
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
|
||
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
|
||
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
||
|
|
|
||
|
|
import
|
||
|
|
std/[json, os, tables, strutils],
|
||
|
|
unittest2,
|
||
|
|
stew/byteutils,
|
||
|
|
./test_helpers,
|
||
|
|
../nimbus/core/chain,
|
||
|
|
../nimbus/common/common,
|
||
|
|
../stateless/[witness_verification, witness_types]
|
||
|
|
|
||
|
|
# use tracerTestGen.nim to generate additional test data
|
||
|
|
proc testFixture(node: JsonNode, testStatusIMPL: var TestStatus) =
|
||
|
|
var
|
||
|
|
blockNumber = UInt256.fromHex(node["blockNumber"].getStr())
|
||
|
|
memoryDB = newCoreDbRef LegacyDbMemory
|
||
|
|
config = chainConfigForNetwork(MainNet)
|
||
|
|
com = CommonRef.new(memoryDB, config, pruneTrie = false)
|
||
|
|
state = node["state"]
|
||
|
|
|
||
|
|
for k, v in state:
|
||
|
|
let key = hexToSeqByte(k)
|
||
|
|
let value = hexToSeqByte(v.getStr())
|
||
|
|
memoryDB.kvt.put(key, value)
|
||
|
|
|
||
|
|
let
|
||
|
|
parentNumber = blockNumber - 1
|
||
|
|
parent = com.db.getBlockHeader(parentNumber)
|
||
|
|
header = com.db.getBlockHeader(blockNumber)
|
||
|
|
headerHash = header.blockHash
|
||
|
|
blockBody = com.db.getBlockBody(headerHash)
|
||
|
|
chain = newChain(com)
|
||
|
|
headers = @[header]
|
||
|
|
bodies = @[blockBody]
|
||
|
|
|
||
|
|
chain.generateWitness = true # Enable code to generate and store witness in the db
|
||
|
|
|
||
|
|
# it's ok if setHead fails here because of missing ancestors
|
||
|
|
discard com.db.setHead(parent, true)
|
||
|
|
let validationResult = chain.persistBlocks(headers, bodies)
|
||
|
|
check validationResult == ValidationResult.OK
|
||
|
|
|
||
|
|
let
|
||
|
|
blockHash = memoryDB.getBlockHash(blockNumber)
|
||
|
|
witness = memoryDB.getBlockWitness(blockHash).value()
|
||
|
|
verifyWitnessResult = verifyWitness(parent.stateRoot, witness, {wfNoFlag})
|
||
|
|
|
||
|
|
check verifyWitnessResult.isOk()
|
||
|
|
let witnessData = verifyWitnessResult.value()
|
||
|
|
|
||
|
|
if blockBody.transactions.len() > 0:
|
||
|
|
check:
|
||
|
|
witness.len() > 0
|
||
|
|
witnessData.len() > 0
|
||
|
|
|
||
|
|
proc persistBlockWitnessJsonMain*() =
|
||
|
|
suite "persist block json tests":
|
||
|
|
jsonTest("PersistBlockTests", testFixture)
|
||
|
|
#var testStatusIMPL: TestStatus
|
||
|
|
#let n = json.parseFile("tests" / "fixtures" / "PersistBlockTests" / "block420301.json")
|
||
|
|
#testFixture(n, testStatusIMPL)
|
||
|
|
|
||
|
|
when isMainModule:
|
||
|
|
persistBlockWitnessJsonMain()
|