mirror of https://github.com/status-im/nim-eth.git
126 lines
2.9 KiB
Nim
126 lines
2.9 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/[os, json, tables, strutils, algorithm],
|
|
stew/byteutils,
|
|
../../eth/trie/[db, hexary]
|
|
|
|
type
|
|
TestOp = object
|
|
idx: int
|
|
key: seq[byte]
|
|
value: seq[byte]
|
|
|
|
proc cmp(lhs, rhs: TestOp): int = cmp(lhs.idx, rhs.idx)
|
|
proc `<=`(lhs, rhs: TestOp): bool = lhs.idx <= rhs.idx
|
|
|
|
proc runSingleTest(testSequence: openArray[TestOp],
|
|
secureMode: bool,
|
|
expectedRootHash: string): bool =
|
|
var
|
|
db = newMemoryDB()
|
|
t = initHexaryTrie(db)
|
|
|
|
for op in testSequence:
|
|
let
|
|
k = op.key
|
|
v = op.value
|
|
|
|
if v.len > 0:
|
|
if secureMode:
|
|
t.SecureHexaryTrie.put k, v
|
|
else:
|
|
t.put k, v
|
|
else:
|
|
if secureMode:
|
|
t.SecureHexaryTrie.del k
|
|
else:
|
|
t.del k
|
|
|
|
return t.rootHashHex == expectedRootHash
|
|
|
|
proc runTests*(filename: string) =
|
|
let js = json.parseFile(filename)
|
|
|
|
for testname, testdata in js:
|
|
template testStatus(status: string) =
|
|
echo status, " ", filename, " :: ", testname
|
|
|
|
template invalidTest =
|
|
testStatus "IGNORED"
|
|
continue
|
|
|
|
let
|
|
input = testdata{"in"}
|
|
root = testdata{"root"}
|
|
secureMode = "secure" in filename
|
|
permuteOrder = "anyorder" in filename
|
|
|
|
if input.isNil or root.isNil or root.kind != JString:
|
|
invalidTest()
|
|
|
|
var inputs = newSeq[TestOp](0)
|
|
|
|
case input.kind
|
|
of JArray:
|
|
for pair in input.elems:
|
|
if pair.kind != JArray or pair.elems.len != 2:
|
|
invalidTest()
|
|
|
|
let
|
|
k = pair.elems[0]
|
|
v = pair.elems[1]
|
|
|
|
if k.kind == JString:
|
|
case v.kind
|
|
of JString:
|
|
inputs.add(TestOp(idx: inputs.len,
|
|
key: k.str.toBytes,
|
|
value: v.str.toBytes))
|
|
of JNull:
|
|
inputs.add(TestOp(idx: inputs.len,
|
|
key: k.str.toBytes,
|
|
value: @[]))
|
|
|
|
else: invalidTest()
|
|
else: invalidTest()
|
|
|
|
of JObject:
|
|
for k, v in input.fields:
|
|
case v.kind
|
|
of JString:
|
|
inputs.add(TestOp(idx: inputs.len,
|
|
key: k.toBytes,
|
|
value: v.str.toBytes))
|
|
of JNull:
|
|
inputs.add(TestOp(idx: inputs.len,
|
|
key: k.toBytes,
|
|
value: @[]))
|
|
|
|
else: invalidTest()
|
|
else: invalidTest()
|
|
|
|
let expectedRootHash = root.str.substr(2).toUpperAscii
|
|
|
|
if permuteOrder:
|
|
sort(inputs, cmp)
|
|
while true:
|
|
if not runSingleTest(inputs, secureMode, expectedRootHash):
|
|
testStatus "FAILED"
|
|
break
|
|
|
|
if not nextPermutation(inputs):
|
|
testStatus "OK"
|
|
break
|
|
|
|
else:
|
|
if runSingleTest(inputs, secureMode, expectedRootHash):
|
|
testStatus "OK"
|
|
else:
|
|
testStatus "FAILED"
|
|
|
|
for file in walkDirRec("tests/cases"):
|
|
if file.endsWith("json"):
|
|
runTests(file)
|
|
|