253 lines
9.5 KiB
Nim
253 lines
9.5 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-2023 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.push raises: [].}
|
|
{.used.}
|
|
|
|
import std/strutils
|
|
import unittest2
|
|
import ../beacon_chain/validator_client/[common, scoring]
|
|
|
|
const
|
|
HostNames = [
|
|
"[2001:db8::1]",
|
|
"127.0.0.1",
|
|
"hostname.com",
|
|
"localhost",
|
|
"username:password@[2001:db8::1]",
|
|
"username:password@127.0.0.1",
|
|
"username:password@hostname.com",
|
|
"username:password@localhost",
|
|
]
|
|
|
|
GoodTestVectors = [
|
|
("http://$1",
|
|
"ok(http://$1)"),
|
|
("http://$1?q=query",
|
|
"ok(http://$1?q=query)"),
|
|
("http://$1?q=query#anchor",
|
|
"ok(http://$1?q=query#anchor)"),
|
|
("http://$1/subpath/",
|
|
"ok(http://$1/subpath/)"),
|
|
("http://$1/subpath/q=query",
|
|
"ok(http://$1/subpath/q=query)"),
|
|
("http://$1/subpath/q=query#anchor",
|
|
"ok(http://$1/subpath/q=query#anchor)"),
|
|
("http://$1/subpath",
|
|
"ok(http://$1/subpath)"),
|
|
("http://$1/subpath?q=query",
|
|
"ok(http://$1/subpath?q=query)"),
|
|
("http://$1/subpath?q=query#anchor",
|
|
"ok(http://$1/subpath?q=query#anchor)"),
|
|
|
|
("https://$1",
|
|
"ok(https://$1)"),
|
|
("https://$1?q=query",
|
|
"ok(https://$1?q=query)"),
|
|
("https://$1?q=query#anchor",
|
|
"ok(https://$1?q=query#anchor)"),
|
|
("https://$1/subpath/",
|
|
"ok(https://$1/subpath/)"),
|
|
("https://$1/subpath/q=query",
|
|
"ok(https://$1/subpath/q=query)"),
|
|
("https://$1/subpath/q=query#anchor",
|
|
"ok(https://$1/subpath/q=query#anchor)"),
|
|
("https://$1/subpath",
|
|
"ok(https://$1/subpath)"),
|
|
("https://$1/subpath?q=query",
|
|
"ok(https://$1/subpath?q=query)"),
|
|
("https://$1/subpath?q=query#anchor",
|
|
"ok(https://$1/subpath?q=query#anchor)"),
|
|
|
|
("$1:5052",
|
|
"ok(http://$1:5052)"),
|
|
("$1:5052?q=query",
|
|
"ok(http://$1:5052?q=query)"),
|
|
("$1:5052?q=query#anchor",
|
|
"ok(http://$1:5052?q=query#anchor)"),
|
|
("$1:5052/subpath/",
|
|
"ok(http://$1:5052/subpath/)"),
|
|
("$1:5052/subpath/q=query",
|
|
"ok(http://$1:5052/subpath/q=query)"),
|
|
("$1:5052/subpath/q=query#anchor",
|
|
"ok(http://$1:5052/subpath/q=query#anchor)"),
|
|
("$1:5052/subpath",
|
|
"ok(http://$1:5052/subpath)"),
|
|
("$1:5052/subpath?q=query",
|
|
"ok(http://$1:5052/subpath?q=query)"),
|
|
("$1:5052/subpath?q=query#anchor",
|
|
"ok(http://$1:5052/subpath?q=query#anchor)"),
|
|
|
|
("bnode://$1:5052",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052?q=query",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052?q=query#anchor",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath/",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath/q=query",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath/q=query#anchor",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath?q=query",
|
|
"err(Unknown scheme value)"),
|
|
("bnode://$1:5052/subpath?q=query#anchor",
|
|
"err(Unknown scheme value)"),
|
|
|
|
("//$1:5052",
|
|
"ok(http://$1:5052)"),
|
|
("//$1:5052?q=query",
|
|
"ok(http://$1:5052?q=query)"),
|
|
("//$1:5052?q=query#anchor",
|
|
"ok(http://$1:5052?q=query#anchor)"),
|
|
("//$1:5052/subpath/",
|
|
"ok(http://$1:5052/subpath/)"),
|
|
("//$1:5052/subpath/q=query",
|
|
"ok(http://$1:5052/subpath/q=query)"),
|
|
("//$1:5052/subpath/q=query#anchor",
|
|
"ok(http://$1:5052/subpath/q=query#anchor)"),
|
|
("//$1:5052/subpath",
|
|
"ok(http://$1:5052/subpath)"),
|
|
("//$1:5052/subpath?q=query",
|
|
"ok(http://$1:5052/subpath?q=query)"),
|
|
("//$1:5052/subpath?q=query#anchor",
|
|
"ok(http://$1:5052/subpath?q=query#anchor)"),
|
|
|
|
("//$1", "err(Missing port number)"),
|
|
("//$1?q=query", "err(Missing port number)"),
|
|
("//$1?q=query#anchor", "err(Missing port number)"),
|
|
("//$1/subpath/", "err(Missing port number)"),
|
|
("//$1/subpath/q=query", "err(Missing port number)"),
|
|
("//$1/subpath/q=query#anchor", "err(Missing port number)"),
|
|
("//$1/subpath", "err(Missing port number)"),
|
|
("//$1/subpath?q=query", "err(Missing port number)"),
|
|
("//$1/subpath?q=query#anchor", "err(Missing port number)"),
|
|
|
|
("$1", "err(Missing port number)"),
|
|
("$1?q=query", "err(Missing port number)"),
|
|
("$1?q=query#anchor", "err(Missing port number)"),
|
|
("$1/subpath/", "err(Missing port number)"),
|
|
("$1/subpath/q=query", "err(Missing port number)"),
|
|
("$1/subpath/q=query#anchor", "err(Missing port number)"),
|
|
("$1/subpath", "err(Missing port number)"),
|
|
("$1/subpath?q=query", "err(Missing port number)"),
|
|
("$1/subpath?q=query#anchor", "err(Missing port number)"),
|
|
|
|
("", "err(Missing hostname)")
|
|
]
|
|
|
|
type
|
|
AttestationDataTuple* = tuple[
|
|
slot: uint64,
|
|
index: uint64,
|
|
beacon_block_root: string,
|
|
source: uint64,
|
|
target: uint64
|
|
]
|
|
|
|
const
|
|
AttestationDataVectors = [
|
|
# Attestation score with block monitoring enabled (perfect).
|
|
((6002798'u64, 10'u64, "22242212", 187586'u64, 187587'u64),
|
|
("22242212", 6002798'u64), "<perfect>"),
|
|
((6002811'u64, 24'u64, "26ec78d6", 187586'u64, 187587'u64),
|
|
("26ec78d6", 6002811'u64), "<perfect>"),
|
|
((6002821'u64, 11'u64, "10c6d1a2", 187587'u64, 187588'u64),
|
|
("10c6d1a2", 6002821'u64), "<perfect>"),
|
|
((6002836'u64, 15'u64, "42354ded", 187587'u64, 187588'u64),
|
|
("42354ded", 6002836'u64), "<perfect>"),
|
|
((6002859'u64, 10'u64, "97d8ac69", 187588'u64, 187589'u64),
|
|
("97d8ac69", 6002859'u64), "<perfect>"),
|
|
# Attestation score with block monitoring enabled #1 (not perfect).
|
|
((6002871'u64, 25'u64, "524a9e2b", 187588'u64, 187589'u64),
|
|
("524a9e2b", 6002870'u64), "375177.5000"),
|
|
((6002871'u64, 25'u64, "524a9e2b", 187588'u64, 187589'u64),
|
|
("524a9e2b", 6002869'u64), "375177.3333"),
|
|
((6002871'u64, 25'u64, "524a9e2b", 187588'u64, 187589'u64),
|
|
("524a9e2b", 6002868'u64), "375177.2500"),
|
|
((6002871'u64, 25'u64, "524a9e2b", 187588'u64, 187589'u64),
|
|
("524a9e2b", 6002867'u64), "375177.2000"),
|
|
((6002871'u64, 25'u64, "524a9e2b", 187588'u64, 187589'u64),
|
|
("524a9e2b", 6002866'u64), "375177.1667"),
|
|
# Attestation score with block monitoring enabled #2 (not perfect).
|
|
((6002962'u64, 14'u64, "22a19d87", 187591'u64, 187592'u64),
|
|
("22a19d87", 6002961'u64), "375183.5000"),
|
|
((6002962'u64, 14'u64, "22a19d87", 187591'u64, 187592'u64),
|
|
("22a19d87", 6002960'u64), "375183.3333"),
|
|
((6002962'u64, 14'u64, "22a19d87", 187591'u64, 187592'u64),
|
|
("22a19d87", 6002959'u64), "375183.2500"),
|
|
((6002962'u64, 14'u64, "22a19d87", 187591'u64, 187592'u64),
|
|
("22a19d87", 6002958'u64), "375183.2000"),
|
|
((6002962'u64, 14'u64, "22a19d87", 187591'u64, 187592'u64),
|
|
("22a19d87", 6002957'u64), "375183.1667"),
|
|
# Attestation score with block monitoring disabled #1.
|
|
((6003217'u64, 52'u64, "5e945218", 187599'u64, 187600'u64),
|
|
("00000000", 0'u64), "375199.0000"),
|
|
((6003217'u64, 52'u64, "5e945218", 187598'u64, 187600'u64),
|
|
("00000000", 0'u64), "375198.0000"),
|
|
((6003217'u64, 52'u64, "5e945218", 187597'u64, 187600'u64),
|
|
("00000000", 0'u64), "375197.0000"),
|
|
((6003217'u64, 52'u64, "5e945218", 187596'u64, 187600'u64),
|
|
("00000000", 0'u64), "375196.0000"),
|
|
((6003217'u64, 52'u64, "5e945218", 187595'u64, 187600'u64),
|
|
("00000000", 0'u64), "375195.0000"),
|
|
# Attestation score with block monitoring disabled #2.
|
|
((6003257'u64, 9'u64, "7bfa464e", 187600'u64, 187601'u64),
|
|
("00000000", 0'u64), "375201.0000"),
|
|
((6003257'u64, 9'u64, "7bfa464e", 187599'u64, 187601'u64),
|
|
("00000000", 0'u64), "375200.0000"),
|
|
((6003257'u64, 9'u64, "7bfa464e", 187598'u64, 187601'u64),
|
|
("00000000", 0'u64), "375199.0000"),
|
|
((6003257'u64, 9'u64, "7bfa464e", 187597'u64, 187601'u64),
|
|
("00000000", 0'u64), "375198.0000"),
|
|
((6003257'u64, 9'u64, "7bfa464e", 187596'u64, 187601'u64),
|
|
("00000000", 0'u64), "375197.0000"),
|
|
]
|
|
|
|
proc init(t: typedesc[Eth2Digest], data: string): Eth2Digest =
|
|
let length = len(data)
|
|
var dst = Eth2Digest()
|
|
try:
|
|
hexToByteArray(data.toOpenArray(0, len(data) - 1),
|
|
dst.data.toOpenArray(0, (length div 2) - 1))
|
|
except ValueError:
|
|
discard
|
|
dst
|
|
|
|
proc init*(t: typedesc[ProduceAttestationDataResponse],
|
|
ad: AttestationDataTuple): ProduceAttestationDataResponse =
|
|
ProduceAttestationDataResponse(data: AttestationData(
|
|
slot: Slot(ad.slot), index: ad.index,
|
|
beacon_block_root: Eth2Digest.init(ad.beacon_block_root),
|
|
source: Checkpoint(epoch: Epoch(ad.source)),
|
|
target: Checkpoint(epoch: Epoch(ad.target))
|
|
))
|
|
|
|
proc createRootsSeen(
|
|
root: tuple[root: string, slot: uint64]): Table[Eth2Digest, Slot] =
|
|
var res: Table[Eth2Digest, Slot]
|
|
res[Eth2Digest.init(root.root)] = Slot(root.slot)
|
|
res
|
|
|
|
suite "Validator Client test suite":
|
|
test "normalizeUri() test vectors":
|
|
for hostname in HostNames:
|
|
for vector in GoodTestVectors:
|
|
let expect = vector[1] % (hostname)
|
|
check $normalizeUri(parseUri(vector[0] % (hostname))) == expect
|
|
|
|
test "getAttestationDataScore() test vectors":
|
|
for vector in AttestationDataVectors:
|
|
let
|
|
adata = ProduceAttestationDataResponse.init(vector[0])
|
|
roots = createRootsSeen(vector[1])
|
|
score = shortScore(roots.getAttestationDataScore(adata))
|
|
check score == vector[2]
|