status-im
/
nim-libp2p
mirror of https://github.com/status-im/nim-libp2p.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Refactor discovery tests for reusability.

This commit is contained in:
Alejandro Cabeza Romero 2024-08-07 21:06:13 +02:00
parent a85be0463e
commit c8bc839c81
No known key found for this signature in database
GPG Key ID: DA3D14AE478030FD
1 changed files with 32 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
tests/testdiscovery.nim
View File

@ -18,7 +18,7 @@ import
discovery/discoverymngr,
discovery/rendezvousinterface,
]
import ./helpers
import ./helpers, ./asyncunit
proc createSwitch(rdv: RendezVous = RendezVous.new()): Switch =
SwitchBuilder
@ -32,17 +32,30 @@ proc createSwitch(rdv: RendezVous = RendezVous.new()): Switch =
.build()
suite "Discovery":
teardown:
checkTrackers()
asyncTest "RendezVous test":
let
rdvA = RendezVous.new()
rdvB = RendezVous.new()
clientA = createSwitch(rdvA)
clientB = createSwitch(rdvB)
remoteNode = createSwitch()
dmA = DiscoveryManager()
dmB = DiscoveryManager()
const
namespace = "foo"
rdvNamespace = RdvNamespace(namespace)
var
rdvA {.threadvar.}: RendezVous
rdvB {.threadvar.}: RendezVous
rdvRemote {.threadvar.}: RendezVous
clientA {.threadvar.}: Switch
clientB {.threadvar.}: Switch
remoteNode {.threadvar.}: Switch
dmA {.threadvar.}: DiscoveryManager
dmB {.threadvar.}: DiscoveryManager
asyncSetup:
rdvA = RendezVous.new()
rdvB = RendezVous.new()
rdvRemote = RendezVous.new()
clientA = createSwitch(rdvA)
clientB = createSwitch(rdvB)
remoteNode = createSwitch(rdvRemote)
dmA = DiscoveryManager()
dmB = DiscoveryManager()
dmA.add(RendezVousInterface.new(rdvA, ttr = 500.milliseconds))
dmB.add(RendezVousInterface.new(rdvB))
await allFutures(clientA.start(), clientB.start(), remoteNode.start())
@ -50,14 +63,18 @@ suite "Discovery":
await clientB.connect(remoteNode.peerInfo.peerId, remoteNode.peerInfo.addrs)
await clientA.connect(remoteNode.peerInfo.peerId, remoteNode.peerInfo.addrs)
dmB.advertise(RdvNamespace("foo"))
asyncTeardown:
await allFutures(clientA.stop(), clientB.stop(), remoteNode.stop())
checkTrackers()
asyncTest "RendezVous test":
dmB.advertise(rdvNamespace)
let
query = dmA.request(RdvNamespace("foo"))
query = dmA.request(rdvNamespace)
res = await query.getPeer()
check:
res{PeerId}.get() == clientB.peerInfo.peerId
res[PeerId] == clientB.peerInfo.peerId
res.getAll(PeerId) == @[clientB.peerInfo.peerId]
toHashSet(res.getAll(MultiAddress)) == toHashSet(clientB.peerInfo.addrs)
await allFutures(clientA.stop(), clientB.stop(), remoteNode.stop())