143 lines
4.8 KiB
Nim
143 lines
4.8 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2022 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.used.}
|
|
|
|
{.push raises: [Defect].}
|
|
|
|
import
|
|
unittest2, stint,
|
|
eth/common/eth_types_rlp,
|
|
../data/history_data_parser,
|
|
../network/history/[history_content, accumulator],
|
|
./test_helpers
|
|
|
|
suite "Header Accumulator":
|
|
test "Header Accumulator Canonical Verification":
|
|
const
|
|
# Amount of headers to be created and added to the accumulator
|
|
amount = mergeBlockNumber
|
|
# Headers to test verification for.
|
|
# Note: This test assumes at least 5 epochs
|
|
headersToTest = [
|
|
0,
|
|
epochSize - 1,
|
|
epochSize,
|
|
epochSize*2 - 1,
|
|
epochSize*2,
|
|
epochSize*3 - 1,
|
|
epochSize*3,
|
|
epochSize*3 + 1,
|
|
int(amount) - 1]
|
|
|
|
var headers: seq[BlockHeader]
|
|
for i in 0..<amount:
|
|
# Note: These test headers will not be a blockchain, as the parent hashes
|
|
# are not properly filled in. That's fine however for this test, as that
|
|
# is not the way the headers are verified with the accumulator.
|
|
headers.add(BlockHeader(
|
|
blockNumber: i.stuint(256), difficulty: 1.stuint(256)))
|
|
|
|
let accumulatorRes = buildAccumulatorData(headers)
|
|
check accumulatorRes.isOk()
|
|
let (accumulator, epochAccumulators) = accumulatorRes.get()
|
|
|
|
block: # Test valid headers
|
|
for i in headersToTest:
|
|
let header = headers[i]
|
|
let proof = buildProof(header, epochAccumulators)
|
|
check:
|
|
proof.isOk()
|
|
verifyAccumulatorProof(accumulator, header, proof.get()).isOk()
|
|
|
|
block: # Test invalid headers
|
|
# Post merge block number must fail (> than latest header in accumulator)
|
|
var proof: AccumulatorProof
|
|
let header = BlockHeader(blockNumber: mergeBlockNumber.stuint(256))
|
|
check verifyAccumulatorProof(accumulator, header, proof).isErr()
|
|
|
|
# Test altered block headers by altering the difficulty
|
|
for i in headersToTest:
|
|
let proof = buildProof(headers[i], epochAccumulators)
|
|
check:
|
|
proof.isOk()
|
|
# Alter the block header so the proof no longer matches
|
|
let header = BlockHeader(
|
|
blockNumber: i.stuint(256), difficulty: 2.stuint(256))
|
|
|
|
check verifyAccumulatorProof(accumulator, header, proof.get()).isErr()
|
|
|
|
block: # Test invalid proofs
|
|
var proof: AccumulatorProof
|
|
|
|
for i in headersToTest:
|
|
check verifyAccumulatorProof(accumulator, headers[i], proof).isErr()
|
|
|
|
test "Header Accumulator - Not Finished":
|
|
# Less headers than needed to finish the accumulator
|
|
const amount = mergeBlockNumber - 1
|
|
|
|
var headers: seq[BlockHeader]
|
|
for i in 0..<amount:
|
|
headers.add(BlockHeader(
|
|
blockNumber: i.stuint(256), difficulty: 1.stuint(256)))
|
|
|
|
let accumulatorRes = buildAccumulator(headers)
|
|
|
|
check accumulatorRes.isErr()
|
|
|
|
test "Header BlockNumber to EpochAccumulator Root":
|
|
# Note: This test assumes at least 3 epochs
|
|
const amount = mergeBlockNumber
|
|
|
|
var
|
|
headerHashes: seq[Hash256] = @[]
|
|
headers: seq[BlockHeader]
|
|
|
|
for i in 0..<amount:
|
|
let header = BlockHeader(blockNumber: u256(i), difficulty: u256(1))
|
|
headers.add(header)
|
|
headerHashes.add(header.blockHash())
|
|
|
|
let accumulatorRes = buildAccumulator(headers)
|
|
check accumulatorRes.isOk()
|
|
let accumulator = accumulatorRes.get()
|
|
|
|
# Valid response for block numbers in epoch 0
|
|
block:
|
|
for i in 0..<epochSize:
|
|
let res = accumulator.getBlockEpochDataForBlockNumber(u256(i))
|
|
check:
|
|
res.isOk()
|
|
res.get().epochHash == accumulator.historicalEpochs[0]
|
|
|
|
# Valid response for block numbers in epoch 1
|
|
block:
|
|
for i in epochSize..<(2 * epochSize):
|
|
let res = accumulator.getBlockEpochDataForBlockNumber(u256(i))
|
|
check:
|
|
res.isOk()
|
|
res.get().epochHash == accumulator.historicalEpochs[1]
|
|
|
|
# Valid response for block numbers in the incomplete (= last) epoch
|
|
block:
|
|
const startIndex = mergeBlockNumber - (mergeBlockNumber mod epochSize)
|
|
for i in startIndex..<mergeBlockNumber:
|
|
let res = accumulator.getBlockEpochDataForBlockNumber(u256(i))
|
|
check:
|
|
res.isOk()
|
|
res.get().epochHash ==
|
|
accumulator.historicalEpochs[preMergeEpochs - 1]
|
|
|
|
# Error for block number at and past merge
|
|
block:
|
|
check:
|
|
accumulator.getBlockEpochDataForBlockNumber(
|
|
u256(mergeBlockNumber)).isErr()
|
|
accumulator.getBlockEpochDataForBlockNumber(
|
|
u256(mergeBlockNumber + 1)).isErr()
|