swarmsim/tests/engine/tpeer.nim
Jaremy Creechley 3a750decc4
move around
2023-08-18 16:04:32 -07:00

21 lines
448 B
Nim

import std/unittest
import std/sets
import pkg/swarmsim/engine/peer
suite "peer":
test "should allow inclusion and membership tests on a HashSet":
var peerSet = HashSet[Peer]()
let p1 = Peer.new(protocols = @[], peerId = 1.some)
let p2 = Peer.new(protocols = @[], peerId = 2.some)
peerSet.incl(p1)
check(peerSet.contains(p1))
check(not peerSet.contains(p2))
peerSet.excl(p1)
check(not peerSet.contains(p1))