118 lines
3.4 KiB
Nim
118 lines
3.4 KiB
Nim
import
|
|
std/[strutils, tables, os, json],
|
|
unittest2,
|
|
stew/byteutils,
|
|
../nimbus/core/pow/difficulty,
|
|
../nimbus/constants,
|
|
../nimbus/common/common,
|
|
./test_helpers
|
|
|
|
type
|
|
Tester = object
|
|
parentTimestamp: int64
|
|
parentDifficulty: Uint256
|
|
parentUncles: Hash256
|
|
currentTimestamp: int64
|
|
currentBlockNumber: Uint256
|
|
currentDifficulty: Uint256
|
|
|
|
Tests = Table[string, Tester]
|
|
|
|
const
|
|
inputPath = "tests" / "fixtures" / "eth_tests" / "DifficultyTests"
|
|
|
|
proc hexOrInt64(data: JsonNode, key: string, hex: static[bool]): int64 =
|
|
when hex:
|
|
getHexadecimalInt data[key]
|
|
else:
|
|
int64(parseInt data[key].getStr)
|
|
|
|
proc hexOrInt256(data: JsonNode, key: string, hex: static[bool]): Uint256 =
|
|
when hex:
|
|
UInt256.fromHex data[key].getStr
|
|
else:
|
|
parse(data[key].getStr, Uint256)
|
|
|
|
proc parseHash(data: string): Hash256 =
|
|
case data
|
|
of "0x00": result = EMPTY_UNCLE_HASH
|
|
of "0x01": result.data[0] = 1.byte
|
|
else:
|
|
doAssert(false, "invalid uncle hash")
|
|
|
|
proc parseTests(testData: JSonNode): Tests =
|
|
const hex = true
|
|
result = initTable[string, Tester]()
|
|
var t: Tester
|
|
for title, data in testData:
|
|
t.parentTimestamp = hexOrInt64(data, "parentTimestamp", hex)
|
|
t.parentDifficulty = hexOrInt256(data, "parentDifficulty", hex)
|
|
let pu = data.fields.getOrDefault("parentUncles")
|
|
if pu.isNil:
|
|
t.parentUncles = EMPTY_UNCLE_HASH
|
|
else:
|
|
t.parentUncles = parseHash(pu.getStr)
|
|
t.currentTimestamp = hexOrInt64(data, "currentTimestamp", hex)
|
|
t.currentBlockNumber = hexOrInt256(data, "currentBlockNumber", hex)
|
|
t.currentDifficulty = hexOrInt256(data, "currentDifficulty", hex)
|
|
result[title] = t
|
|
|
|
proc calculator(revision: string, timestamp: EthTime, header: BlockHeader): DifficultyInt =
|
|
case revision
|
|
of "Homestead": result = calcDifficultyHomestead(timestamp, header)
|
|
of "GrayGlacier": result = calcDifficultyGrayGlacier(timestamp, header)
|
|
of "Frontier": result = calcDifficultyFrontier(timestamp, header)
|
|
of "Berlin": result = calcDifficultyMuirGlacier(timestamp, header)
|
|
of "Constantinople": result = calcDifficultyConstantinople(timestamp, header)
|
|
of "Byzantium": result = calcDifficultyByzantium(timestamp, header)
|
|
of "ArrowGlacier": result = calcDifficultyArrowGlacier(timestamp, header)
|
|
else:
|
|
doAssert(false, "unknown revision: " & revision)
|
|
|
|
proc testFixture(fixtures: JsonNode, testStatusIMPL: var TestStatus) =
|
|
var fixture: JsonNode
|
|
for _, child in fixtures:
|
|
fixture = child
|
|
break
|
|
|
|
for revision, child in fixture:
|
|
if revision == "_info":
|
|
continue
|
|
|
|
let tests = parseTests(child)
|
|
|
|
for title, t in tests:
|
|
let p = BlockHeader(
|
|
difficulty : t.parentDifficulty,
|
|
timestamp : EthTime(t.parentTimestamp),
|
|
blockNumber: t.currentBlockNumber - 1,
|
|
ommersHash : t.parentUncles
|
|
)
|
|
|
|
let timestamp = EthTime(t.currentTimeStamp)
|
|
|
|
let diff = calculator(revision, timestamp, p)
|
|
check diff == t.currentDifficulty
|
|
|
|
template runTest() =
|
|
var filenames: seq[string] = @[]
|
|
for filename in walkDirRec(inputPath):
|
|
if not filename.endsWith(".json"):
|
|
continue
|
|
filenames.add filename
|
|
|
|
doAssert(filenames.len > 0)
|
|
|
|
for fname in filenames:
|
|
let filename = fname
|
|
test fname.subStr(inputPath.len + 1):
|
|
let fixtures = parseJson(readFile(filename))
|
|
testFixture(fixtures, testStatusIMPL)
|
|
|
|
proc difficultyMain*() =
|
|
suite "DifficultyTest":
|
|
runTest()
|
|
|
|
when isMainModule:
|
|
difficultyMain()
|