nimbus-eth1/tests/test_aristo/test_merge_proof.nim
Jacek Sieka 58cde36656
Remove RawData from possible leaf payload types (#2794)
This kind of data is not used except in tests where it is used only to
create databases that don't match actual usage of aristo.

Removing simplifies future optimizations that can focus on processing
specific leaf types more efficiently.

A casualty of this removal is some test code as well as some proof
generation code that is unused - on the surface, it looks like it should
be possible to port both of these to the more specific data types -
doing so would ensure that a database written by one part of the
codebase can interact with the other - as it stands, there is confusion
on this point since using the proof generation code will result in a
database of a shape that is incompatible with the rest of eth1.
2024-11-02 10:29:16 +01:00

155 lines
4.1 KiB
Nim

# Nimbus
# Copyright (c) 2023-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
eth/common,
results,
unittest2,
../../nimbus/db/opts,
../../nimbus/db/core_db/backend/aristo_rocksdb,
../../nimbus/db/aristo/[
aristo_check,
aristo_desc,
aristo_init/persistent,
aristo_part,
aristo_part/part_debug,
aristo_tx],
../replay/xcheck,
./test_helpers
const
testRootVid = VertexID(2)
## Need to reconfigure for the test, root ID 1 cannot be deleted as a trie
# ------------------------------------------------------------------------------
# Private helper
# ------------------------------------------------------------------------------
proc innerCleanUp(ps: var PartStateRef) =
if not ps.isNil:
ps.db.finish(eradicate=true)
ps = PartStateRef(nil)
# -----------------------
proc saveToBackend(
tx: var AristoTxRef;
noisy: bool;
debugID: int;
): bool =
var db = tx.to(AristoDbRef)
# Verify context: nesting level must be 2 (i.e. two transactions)
xCheck tx.level == 2
# Commit and hashify the current layer
block:
let rc = tx.commit()
xCheckRc rc.error == 0
block:
let rc = db.txTop()
xCheckRc rc.error == 0
tx = rc.value
# Verify context: nesting level must be 1 (i.e. one transaction)
xCheck tx.level == 1
block:
let rc = db.checkBE()
xCheckRc rc.error == (0,0)
# Commit and save to backend
block:
let rc = tx.commit()
xCheckRc rc.error == 0
block:
let rc = db.txTop()
xCheckErr rc.value.level < 0 # force error
block:
let rc = db.schedStow()
xCheckRc rc.error == 0
# Update layers to original level
tx = db.txBegin().value.to(AristoDbRef).txBegin().value
true
# ------------------------------------------------------------------------------
# Public test function
# ------------------------------------------------------------------------------
proc testMergeProofAndKvpList*(
noisy: bool;
list: openArray[ProofTrieData];
rdbPath: string; # Rocks DB storage directory
idPfx = "";
): bool {.deprecated.} =
# TODO update for non-generic data
# var
# ps = PartStateRef(nil)
# tx = AristoTxRef(nil)
# rootKey: Hash32
# defer:
# if not ps.isNil:
# ps.db.finish(eradicate=true)
# for n,w in list:
# # Start new database upon request
# if w.root != rootKey or w.proof.len == 0:
# ps.innerCleanUp()
# let db = block:
# # New DB with disabled filter slots management
# if 0 < rdbPath.len:
# let (dbOpts, cfOpts) = DbOptions.init().toRocksDb()
# let rc = AristoDbRef.init(RdbBackendRef, rdbPath, DbOptions.init(), dbOpts, cfOpts, [])
# xCheckRc rc.error == 0
# rc.value()[0]
# else:
# AristoDbRef.init(MemBackendRef)
# ps = PartStateRef.init(db)
# # Start transaction (double frame for testing)
# tx = ps.db.txBegin().value.to(AristoDbRef).txBegin().value
# xCheck tx.isTop()
# # Update root
# rootKey = w.root
# if 0 < w.proof.len:
# let rc = ps.partPut(w.proof, ForceGenericPayload)
# xCheckRc rc.error == 0
# block:
# let rc = ps.check()
# xCheckRc rc.error == (0,0)
# for ltp in w.kvpLst:
# block:
# let rc = ps.partMergeGenericData(
# testRootVid, @(ltp.leafTie.path), ltp.payload.rawBlob)
# xCheckRc rc.error == 0
# block:
# let rc = ps.check()
# xCheckRc rc.error == (0,0)
# block:
# let saveBeOk = tx.saveToBackend(noisy=noisy, debugID=n)
# xCheck saveBeOk
true
# ------------------------------------------------------------------------------
# End
# ------------------------------------------------------------------------------