mirror of https://github.com/status-im/nim-rlp.git
75 lines
1.9 KiB
Nim
75 lines
1.9 KiB
Nim
|
import
|
||
|
|
json, strutils, rlp
|
||
|
|
|
||
|
|
proc append(output: var RlpWriter, js: JsonNode) =
|
||
|
|
case js.kind
|
||
|
|
of JNull, JFloat, JObject:
|
||
|
|
raise newException(ValueError, "Unsupported JSON value type " & $js.kind)
|
||
|
|
of JBool:
|
||
|
|
output.append js.bval.int
|
||
|
|
of JInt:
|
||
|
|
output.append int(js.num)
|
||
|
|
of JString:
|
||
|
|
output.append js.str
|
||
|
|
of JArray:
|
||
|
|
output.append js.elems
|
||
|
|
|
||
|
|
proc hexRepr*(bytes: BytesRange): string =
|
||
|
|
result = newStringOfCap(bytes.len * 2)
|
||
|
|
for byte in bytes:
|
||
|
|
result.add(toHex(int(byte), 2).toLowerAscii)
|
||
|
|
|
||
|
|
proc `==`(lhs: JsonNode, rhs: string): bool =
|
||
|
|
lhs.kind == JString and lhs.str == rhs
|
||
|
|
|
||
|
|
proc runTests*(filename: string) =
|
||
|
|
let js = json.parseFile(filename)
|
||
|
|
|
||
|
|
for testname, testdata in js:
|
||
|
|
template testStatus(status: string) =
|
||
|
|
echo status, " ", filename, " :: ", testname
|
||
|
|
|
||
|
|
let
|
||
|
|
input = testdata{"in"}
|
||
|
|
output = testdata{"out"}
|
||
|
|
|
||
|
|
if input.isNil or output.isNil or output.kind != JString:
|
||
|
|
testStatus "IGNORED"
|
||
|
|
continue
|
||
|
|
|
||
|
|
if input == "VALID":
|
||
|
|
var rlp = rlpFromHex(output.str)
|
||
|
|
discard rlp.inspect
|
||
|
|
elif input == "INVALID":
|
||
|
|
var success = false
|
||
|
|
var inspectOutput = ""
|
||
|
|
try:
|
||
|
|
var rlp = rlpFromHex(output.str)
|
||
|
|
inspectOutput = rlp.inspect(1)
|
||
|
|
discard rlp.getType
|
||
|
|
while rlp.hasData: discard rlp.toNodes
|
||
|
|
except MalformedRlpError, ValueError:
|
||
|
|
success = true
|
||
|
|
if not success:
|
||
|
|
testStatus "FAILED"
|
||
|
|
echo " ACCEPTED MALFORMED BYTES: ", output.str
|
||
|
|
echo " INTERPRETATION:\n", inspectOutput
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
if input.kind == JString and input.str[0] == '#':
|
||
|
|
continue
|
||
|
|
|
||
|
|
var outRlp = initRlpWriter()
|
||
|
|
outRlp.append input
|
||
|
|
let
|
||
|
|
actual = outRlp.finish.hexRepr
|
||
|
|
expected = output.str
|
||
|
|
if actual != expected:
|
||
|
|
testStatus "FAILED"
|
||
|
|
echo " EXPECTED BYTES: ", expected
|
||
|
|
echo " ACTUAL BYTES: ", actual
|
||
|
|
continue
|
||
|
|
|
||
|
|
testStatus "OK"
|
||
|
|
|