mirror of https://github.com/status-im/nim-eth.git
109 lines
3.4 KiB
Nim
109 lines
3.4 KiB
Nim
# nim-eth
|
|
# Copyright (c) 2021-2024 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.
|
|
|
|
{.used.}
|
|
|
|
import
|
|
std/net,
|
|
unittest2,
|
|
../../eth/common/keys, ../../eth/p2p/discoveryv5/[node, ip_vote]
|
|
|
|
suite "IP vote":
|
|
let rng = newRng()
|
|
|
|
test "Majority vote":
|
|
var
|
|
votes = IpVote.init(2)
|
|
let
|
|
addr1 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(1))
|
|
addr2 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(2))
|
|
addr3 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(3))
|
|
|
|
votes.insert(NodeId.random(rng[]), addr1);
|
|
votes.insert(NodeId.random(rng[]), addr1);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
votes.insert(NodeId.random(rng[]), addr3);
|
|
votes.insert(NodeId.random(rng[]), addr3);
|
|
|
|
check votes.majority() == Opt.some(addr2)
|
|
|
|
test "Votes below threshold":
|
|
const threshold = 10
|
|
|
|
var
|
|
votes = IpVote.init(threshold)
|
|
let
|
|
addr1 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(1))
|
|
addr2 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(2))
|
|
addr3 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(3))
|
|
|
|
votes.insert(NodeId.random(rng[]), addr1);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
|
|
for i in 0..<(threshold - 1):
|
|
votes.insert(NodeId.random(rng[]), addr3);
|
|
|
|
check votes.majority().isNone()
|
|
|
|
test "Votes at threshold":
|
|
const threshold = 10
|
|
|
|
var
|
|
votes = IpVote.init(threshold)
|
|
let
|
|
addr1 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(1))
|
|
addr2 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(2))
|
|
addr3 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(3))
|
|
|
|
votes.insert(NodeId.random(rng[]), addr1);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
|
|
for i in 0..<(threshold):
|
|
votes.insert(NodeId.random(rng[]), addr3);
|
|
|
|
check votes.majority() == Opt.some(addr3)
|
|
|
|
test "Double votes with same address":
|
|
const threshold = 2
|
|
|
|
var
|
|
votes = IpVote.init(threshold)
|
|
let
|
|
addr1 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(1))
|
|
addr2 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(2))
|
|
|
|
let nodeIdA = NodeId.random(rng[])
|
|
votes.insert(nodeIdA, addr1);
|
|
votes.insert(nodeIdA, addr1);
|
|
votes.insert(nodeIdA, addr1);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
|
|
check votes.majority() == Opt.some(addr2)
|
|
|
|
test "Double votes with different address":
|
|
const threshold = 2
|
|
|
|
var
|
|
votes = IpVote.init(threshold)
|
|
let
|
|
addr1 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(1))
|
|
addr2 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(2))
|
|
addr3 = Address(ip: parseIpAddress("127.0.0.1"), port: Port(3))
|
|
|
|
let nodeIdA = NodeId.random(rng[])
|
|
votes.insert(nodeIdA, addr1);
|
|
votes.insert(nodeIdA, addr2);
|
|
votes.insert(nodeIdA, addr3);
|
|
votes.insert(NodeId.random(rng[]), addr1);
|
|
votes.insert(NodeId.random(rng[]), addr2);
|
|
votes.insert(NodeId.random(rng[]), addr3);
|
|
|
|
check votes.majority() == Opt.some(addr3)
|