124 lines
3.4 KiB
Nim
124 lines
3.4 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at http://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at http://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import
|
|
unittest, nimcrypto, eth_common, sequtils, options, milagro_crypto,
|
|
../beacon_chain/ssz, ../beacon_chain/spec/datatypes
|
|
|
|
func filled[N: static[int], T](typ: type array[N, T], value: T): array[N, T] =
|
|
for val in result.mitems:
|
|
val = value
|
|
|
|
func filled(T: type MDigest, value: byte): T =
|
|
for val in result.data.mitems:
|
|
val = value
|
|
|
|
suite "Simple serialization":
|
|
# pending spec updates in
|
|
# - https://github.com/ethereum/eth2.0-specs
|
|
type
|
|
Foo = object
|
|
f0: uint8
|
|
f1: uint32
|
|
f2: EthAddress
|
|
f3: MDigest[256]
|
|
f4: seq[byte]
|
|
f5: Uint24
|
|
|
|
let expected_deser = Foo(
|
|
f0: 5,
|
|
f1: 0'u32 - 3,
|
|
f2: EthAddress.filled(byte 35),
|
|
f3: MDigest[256].filled(byte 35),
|
|
f4: @[byte 'c'.ord, 'o'.ord, 'w'.ord],
|
|
f5: Uint24(79)
|
|
)
|
|
|
|
var expected_ser = @[
|
|
byte 0, 0, 0, 67, # length
|
|
5,
|
|
'\xFF'.ord, '\xFF'.ord, '\xFF'.ord, '\xFD'.ord,
|
|
]
|
|
expected_ser &= EthAddress.filled(byte 35)
|
|
expected_ser &= MDigest[256].filled(byte 35).data
|
|
expected_ser &= [byte 0, 0, 0, 3, 'c'.ord, 'o'.ord, 'w'.ord]
|
|
expected_ser &= [byte 0, 0, 79]
|
|
|
|
test "Object deserialization":
|
|
let deser = expected_ser.deserialize(Foo).get()
|
|
check: expected_deser == deser
|
|
|
|
test "Object serialization":
|
|
let ser = expected_deser.serialize()
|
|
check: expected_ser == ser
|
|
|
|
test "Not enough data":
|
|
check:
|
|
expected_ser[0..^2].deserialize(Foo).isNone()
|
|
expected_ser[1..^1].deserialize(Foo).isNone()
|
|
|
|
test "Uint24 roundtrip":
|
|
# https://github.com/nim-lang/Nim/issues/10027
|
|
let v = 79.Uint24
|
|
let ser = v.serialize()
|
|
check:
|
|
ser.len() == 3
|
|
deserialize(ser, type(v)).get() == v
|
|
|
|
test "Array roundtrip":
|
|
let v = [1, 2, 3]
|
|
let ser = v.serialize()
|
|
check:
|
|
deserialize(ser, type(v)).get() == v
|
|
|
|
test "Seq roundtrip":
|
|
let v = @[1, 2, 3]
|
|
let ser = v.serialize()
|
|
|
|
check:
|
|
deserialize(ser, type(v)).get() == v
|
|
|
|
test "Key roundtrip":
|
|
let v = newSigKey().fromSigKey()
|
|
let ser = v.serialize()
|
|
|
|
check:
|
|
deserialize(ser, type(v)).get() == v
|
|
|
|
# Just to see that we can serialize stuff at all
|
|
test "Roundtrip main eth2 types":
|
|
let
|
|
bb = BeaconBlock(
|
|
slot: 42,
|
|
signature: signMessage(newSigKey(), "")
|
|
)
|
|
bs = BeaconState(slot: 42)
|
|
|
|
check:
|
|
bb.serialize().deserialize(BeaconBlock).get() == bb
|
|
bs.serialize().deserialize(BeaconState).get() == bs
|
|
|
|
suite "Tree hashing":
|
|
# TODO Nothing but smoke tests for now..
|
|
|
|
test "Hash ValidatorRecord":
|
|
let vr = ValidatorRecord()
|
|
check: hash_tree_root(vr).len > 0
|
|
|
|
test "Hash ShardCommittee":
|
|
let sc = ShardCommittee()
|
|
check: hash_tree_root(sc).len > 0
|
|
|
|
test "Hash BeaconBlock":
|
|
## TODO: Test genesis hash when spec is updated
|
|
let bb = BeaconBlock()
|
|
check: hash_tree_root(bb).len > 0
|
|
|
|
test "Hash integer":
|
|
check: hash_tree_root(0x01'u32) == [0'u8, 0, 0, 1] # big endian!
|
|
check: hash_tree_root(Uint24(0x01)) == [0'u8, 0, 1] # big endian!
|