swarmsim/tests/engine/tpeer.nim
Jaremy Creechley 4cf1b793e2
import tweaks
2023-08-18 16:09:04 -07:00

21 lines
444 B
Nim

import std/unittest
import std/sets
import 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))