mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-03 14:33:12 +00:00
9f68c83fed
* properly pass userMessageLimit to OnchainGroupManager * waku.nimble 2.2.4 Nim compiler * rm stew/shims/net import * change ValidIpAddress.init with parseIpAddress * fix serialize for zerokit * group_manager: separate if statements * protocol_types: add encode UInt32 with zeros up to 32 bytes * windows build: skip libunwind build and rm libunwind.a inlcusion step * bump nph to overcome the compilation issues with 2.2.x * bump nim-libp2p to v1.10.1
132 lines
5.1 KiB
Nim
132 lines
5.1 KiB
Nim
import chronicles, std/[options, tables, strutils], chronos, testutils/unittests
|
|
|
|
import
|
|
waku/node/waku_node,
|
|
waku/waku_core,
|
|
../../waku_lightpush/[lightpush_utils],
|
|
../../testlib/[wakucore, wakunode, futures, testasync],
|
|
waku/node/peer_manager/peer_manager
|
|
|
|
suite "Peer Manager":
|
|
suite "onPeerMetadata":
|
|
var
|
|
listenPort {.threadvar.}: Port
|
|
listenAddress {.threadvar.}: IpAddress
|
|
serverKey {.threadvar.}: PrivateKey
|
|
clientKey {.threadvar.}: PrivateKey
|
|
clusterId {.threadvar.}: uint64
|
|
|
|
asyncSetup:
|
|
listenPort = Port(0)
|
|
listenAddress = parseIpAddress("0.0.0.0")
|
|
serverKey = generateSecp256k1Key()
|
|
clientKey = generateSecp256k1Key()
|
|
clusterId = 1
|
|
|
|
asyncTest "light client is not disconnected":
|
|
# Given two nodes with the same shardId
|
|
let
|
|
server = newTestWakuNode(
|
|
serverKey, listenAddress, listenPort, clusterId = clusterId, shards = @[0]
|
|
)
|
|
client = newTestWakuNode(
|
|
clientKey, listenAddress, listenPort, clusterId = clusterId, shards = @[1]
|
|
)
|
|
|
|
# And both mount metadata and filter
|
|
discard client.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
discard server.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
await client.mountFilterClient()
|
|
await server.mountFilter()
|
|
|
|
# And both nodes are started
|
|
await allFutures(server.start(), client.start())
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# And the nodes are connected
|
|
let serverRemotePeerInfo = server.switch.peerInfo.toRemotePeerInfo()
|
|
await client.connectToNodes(@[serverRemotePeerInfo])
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# When making an operation that triggers onPeerMetadata
|
|
discard await client.filterSubscribe(
|
|
some("/waku/2/rs/0/0"), "waku/lightpush/1", serverRemotePeerInfo
|
|
)
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
check:
|
|
server.switch.isConnected(client.switch.peerInfo.toRemotePeerInfo().peerId)
|
|
client.switch.isConnected(server.switch.peerInfo.toRemotePeerInfo().peerId)
|
|
|
|
asyncTest "relay with same shardId is not disconnected":
|
|
# Given two nodes with the same shardId
|
|
let
|
|
server = newTestWakuNode(
|
|
serverKey, listenAddress, listenPort, clusterId = clusterId, shards = @[0]
|
|
)
|
|
client = newTestWakuNode(
|
|
clientKey, listenAddress, listenPort, clusterId = clusterId, shards = @[1]
|
|
)
|
|
|
|
# And both mount metadata and relay
|
|
discard client.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
discard server.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
(await client.mountRelay()).isOkOr:
|
|
assert false, "Failed to mount relay"
|
|
(await server.mountRelay()).isOkOr:
|
|
assert false, "Failed to mount relay"
|
|
|
|
# And both nodes are started
|
|
await allFutures(server.start(), client.start())
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# And the nodes are connected
|
|
let serverRemotePeerInfo = server.switch.peerInfo.toRemotePeerInfo()
|
|
await client.connectToNodes(@[serverRemotePeerInfo])
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# When making an operation that triggers onPeerMetadata
|
|
client.subscribe((kind: SubscriptionKind.PubsubSub, topic: "newTopic")).isOkOr:
|
|
assert false, "Failed to subscribe to relay"
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
check:
|
|
server.switch.isConnected(client.switch.peerInfo.toRemotePeerInfo().peerId)
|
|
client.switch.isConnected(server.switch.peerInfo.toRemotePeerInfo().peerId)
|
|
|
|
asyncTest "relay with different shardId is disconnected":
|
|
# Given two nodes with different shardIds
|
|
let
|
|
server = newTestWakuNode(
|
|
serverKey, listenAddress, listenPort, clusterId = clusterId, shards = @[0]
|
|
)
|
|
client = newTestWakuNode(
|
|
clientKey, listenAddress, listenPort, clusterId = clusterId, shards = @[1]
|
|
)
|
|
|
|
# And both mount metadata and relay
|
|
discard client.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
discard server.mountMetadata(0) # clusterId irrelevant, overridden by topic
|
|
(await client.mountRelay()).isOkOr:
|
|
assert false, "Failed to mount relay"
|
|
(await server.mountRelay()).isOkOr:
|
|
assert false, "Failed to mount relay"
|
|
|
|
# And both nodes are started
|
|
await allFutures(server.start(), client.start())
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# And the nodes are connected
|
|
let serverRemotePeerInfo = server.switch.peerInfo.toRemotePeerInfo()
|
|
await client.connectToNodes(@[serverRemotePeerInfo])
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
# When making an operation that triggers onPeerMetadata
|
|
client.subscribe((kind: SubscriptionKind.PubsubSub, topic: "newTopic")).isOkOr:
|
|
assert false, "Failed to subscribe to relay"
|
|
await sleepAsync(FUTURE_TIMEOUT)
|
|
|
|
check:
|
|
not server.switch.isConnected(client.switch.peerInfo.toRemotePeerInfo().peerId)
|
|
not client.switch.isConnected(server.switch.peerInfo.toRemotePeerInfo().peerId)
|