mirror of https://github.com/waku-org/nwaku.git
141 lines
3.5 KiB
Nim
141 lines
3.5 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/options,
|
|
stew/results,
|
|
stew/shims/net,
|
|
testutils/unittests
|
|
import
|
|
../../waku/common/enr,
|
|
../testlib/wakucore
|
|
|
|
|
|
suite "nim-eth ENR - builder and typed record":
|
|
|
|
test "Non-supported private key (ECDSA)":
|
|
## Given
|
|
let privateKey = generateEcdsaKey()
|
|
|
|
## Then
|
|
expect Defect:
|
|
discard EnrBuilder.init(privateKey)
|
|
|
|
test "Supported private key (Secp256k1)":
|
|
let
|
|
seqNum = 1u64
|
|
privateKey = generateSecp256k1Key()
|
|
|
|
let expectedPubKey = privateKey.getPublicKey().get().getRawBytes().get()
|
|
|
|
## When
|
|
var builder = EnrBuilder.init(privateKey, seqNum)
|
|
let enrRes = builder.build()
|
|
|
|
## Then
|
|
check enrRes.isOk()
|
|
|
|
let record = enrRes.tryGet().toTyped().get()
|
|
|
|
let id = record.id
|
|
check:
|
|
id == some(RecordId.V4)
|
|
|
|
let publicKey = record.secp256k1
|
|
check:
|
|
publicKey.isSome()
|
|
@(publicKey.get()) == expectedPubKey
|
|
|
|
|
|
suite "nim-eth ENR - Ext: IP address and TCP/UDP ports":
|
|
|
|
test "EIP-778 test vector":
|
|
## Given
|
|
# Test vector from EIP-778
|
|
# See: https://eips.ethereum.org/EIPS/eip-778#test-vectors
|
|
let expectedEnr = "-IS4QHCYrYZbAKWCBRlAy5zzaDZXJBGkcnh4MHcBFZntXNFrdvJjX04j" &
|
|
"RzjzCBOonrkTfj499SZuOh8R33Ls8RRcy5wBgmlkgnY0gmlwhH8AAAGJ" &
|
|
"c2VjcDI1NmsxoQPKY0yuDUmstAHYpMa2_oxVtw0RW_QAdpzBQA8yWM0x" &
|
|
"OIN1ZHCCdl8"
|
|
|
|
let
|
|
seqNum = 1u64
|
|
privateKey = ethSecp256k1Key("b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291")
|
|
|
|
enrIpAddr = ValidIpAddress.init("127.0.0.1")
|
|
enrUdpPort = Port(30303)
|
|
|
|
## When
|
|
var builder = EnrBuilder.init(privateKey, seqNum)
|
|
builder.withIpAddressAndPorts(ipAddr=some(enrIpAddr), udpPort=some(enrUdpPort))
|
|
|
|
let enrRes = builder.build()
|
|
|
|
## Then
|
|
check enrRes.isOk()
|
|
|
|
let record = enrRes.tryGet().toBase64()
|
|
check:
|
|
record == expectedEnr
|
|
|
|
test "IPv4 and TCP port":
|
|
let
|
|
seqNum = 1u64
|
|
privateKey = generateSecp256k1Key()
|
|
|
|
enrIpAddr = ValidIpAddress.init("127.0.0.1")
|
|
enrTcpPort = Port(30301)
|
|
|
|
let expectedPubKey = privateKey.getPublicKey().get().getRawBytes().get()
|
|
|
|
## When
|
|
var builder = EnrBuilder.init(privateKey, seqNum)
|
|
builder.withIpAddressAndPorts(
|
|
ipAddr=some(enrIpAddr),
|
|
tcpPort=some(enrTcpPort),
|
|
)
|
|
|
|
let enrRes = builder.build()
|
|
|
|
## Then
|
|
check enrRes.isOk()
|
|
|
|
let record = enrRes.tryGet().toTyped().get()
|
|
check:
|
|
@(record.secp256k1.get()) == expectedPubKey
|
|
record.ip == some(enrIpAddr.address_v4)
|
|
record.tcp == some(enrTcpPort.uint16)
|
|
record.udp == none(uint16)
|
|
record.ip6 == none(array[16, byte])
|
|
|
|
test "IPv6 and UDP port":
|
|
let
|
|
seqNum = 1u64
|
|
privateKey = generateSecp256k1Key()
|
|
|
|
enrIpAddr = ValidIpAddress.init("::1")
|
|
enrUdpPort = Port(30301)
|
|
|
|
let expectedPubKey = privateKey.getPublicKey().get().getRawBytes().get()
|
|
|
|
## When
|
|
var builder = EnrBuilder.init(privateKey, seqNum)
|
|
builder.withIpAddressAndPorts(
|
|
ipAddr=some(enrIpAddr),
|
|
udpPort=some(enrUdpPort),
|
|
)
|
|
|
|
let enrRes = builder.build()
|
|
|
|
## Then
|
|
check enrRes.isOk()
|
|
|
|
let record = enrRes.tryGet().toTyped().get()
|
|
check:
|
|
@(record.secp256k1.get()) == expectedPubKey
|
|
record.ip == none(array[4, byte])
|
|
record.tcp == none(uint16)
|
|
record.udp == none(uint16)
|
|
record.ip6 == some(enrIpAddr.address_v6)
|
|
record.tcp6 == none(uint16)
|
|
record.udp6 == some(enrUdpPort.uint16)
|