85 lines
3.1 KiB
Nim
85 lines
3.1 KiB
Nim
import unittest2, strutils, tables, os, json,
|
|
../nimbus/utils/difficulty, stint, times,
|
|
eth/common, test_helpers, stew/byteutils,
|
|
../nimbus/constants, ../nimbus/chain_config
|
|
|
|
type
|
|
Tester = object
|
|
parentTimestamp: int64
|
|
parentDifficulty: Uint256
|
|
parentUncles: Hash256
|
|
currentTimestamp: int64
|
|
currentBlockNumber: Uint256
|
|
currentDifficulty: Uint256
|
|
|
|
Tests = Table[string, Tester]
|
|
|
|
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 parseTests(name: string, hex: static[bool]): Tests =
|
|
let fileName = "tests" / "fixtures" / "DifficultyTests" / "difficulty" & name & ".json"
|
|
let fixtures = parseJSON(readFile(fileName))
|
|
|
|
result = initTable[string, Tester]()
|
|
var t: Tester
|
|
for title, data in fixtures:
|
|
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:
|
|
hexToByteArray(pu.getStr, t.parentUncles.data)
|
|
t.currentTimestamp = hexOrInt64(data, "currentTimestamp", hex)
|
|
t.currentBlockNumber = hexOrInt256(data, "currentBlockNumber", hex)
|
|
t.currentDifficulty = hexOrInt256(data, "currentDifficulty", hex)
|
|
result[title] = t
|
|
|
|
template runTests(name: string, hex: bool, calculator: typed) =
|
|
let testTitle = if name == "": "Difficulty" else: name
|
|
test testTitle:
|
|
let data = parseTests(name, hex)
|
|
for title, t in data:
|
|
var p = BlockHeader(
|
|
difficulty: t.parentDifficulty,
|
|
timestamp: times.fromUnix(t.parentTimestamp),
|
|
blockNumber: t.currentBlockNumber - 1,
|
|
ommersHash: t.parentUncles)
|
|
|
|
let diff = calculator(times.fromUnix(t.currentTimeStamp), p)
|
|
check diff == t.currentDifficulty
|
|
|
|
proc difficultyMain*() =
|
|
let mainnetConfig = networkParams(MainNet).config
|
|
func calcDifficultyMainNetWork(timeStamp: EthTime, parent: BlockHeader): DifficultyInt =
|
|
mainnetConfig.calcDifficulty(timeStamp, parent)
|
|
|
|
let ropstenConfig = networkParams(RopstenNet).config
|
|
func calcDifficultyRopsten(timeStamp: EthTime, parent: BlockHeader): DifficultyInt =
|
|
ropstenConfig.calcDifficulty(timeStamp, parent)
|
|
|
|
suite "DifficultyTest":
|
|
runTests("EIP2384_random_to20M", true, calcDifficultyMuirGlacier)
|
|
runTests("EIP2384_random", true, calcDifficultyMuirGlacier)
|
|
runTests("EIP2384", true, calcDifficultyMuirGlacier)
|
|
runTests("Byzantium", true, calcDifficultyByzantium)
|
|
runTests("Constantinople", true, calcDifficultyConstantinople)
|
|
runTests("Homestead", true, calcDifficultyHomestead)
|
|
runTests("MainNetwork", true, calcDifficultyMainNetwork)
|
|
runTests("Frontier", true, calcDifficultyFrontier)
|
|
runTests("", false, calcDifficultyMainNetWork)
|
|
runTests("Ropsten", true, calcDifficultyRopsten)
|
|
|
|
when isMainModule:
|
|
difficultyMain()
|