waku-org
/
nwaku
mirror of https://github.com/waku-org/nwaku.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test(relay-filter): cleanup (#2138) 

* Fix some tests.
* Clean legacy tests.
* Fix imports.
This commit is contained in:
Álex Cabeza Romero 2023-11-15 16:15:38 +01:00 committed by GitHub
parent f5f431382b
commit bd25191a74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 541 additions and 718 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/all_tests_waku.nim
View File

@ -41,7 +41,7 @@ when defined(waku_exp_store_resume):
import import
./waku_relay/test_all ./waku_relay/test_all,
./waku_filter_v2/test_all ./waku_filter_v2/test_all

4
tests/testlib/futures.nim
View File

@ -1,9 +1,11 @@
import import
chronicles,
chronos chronos
import ../../../waku/waku_core/message import ../../../waku/waku_core/message
let FUTURE_TIMEOUT* = 1.seconds
proc newPushHandlerFuture*(): Future[(string, WakuMessage)] = proc newPushHandlerFuture*(): Future[(string, WakuMessage)] =
newFuture[(string, WakuMessage)]() newFuture[(string, WakuMessage)]()

766
tests/waku_filter_v2/test_waku_client.nim
View File

File diff suppressed because it is too large Load Diff

27
tests/waku_filter_v2/waku_filter_utils.nim
View File

@ -1,19 +1,24 @@
import import
std/[options,tables], std/[
std/[sequtils,sets,strutils], options,
testutils/unittests, tables,
sets
],
chronos, chronos,
chronicles chronicles
import import
../../../waku/node/peer_manager, ../../../waku/[
../../../waku/waku_filter_v2, node/peer_manager,
../../../waku/waku_filter_v2/client, waku_filter_v2,
../../../waku/waku_filter_v2/subscriptions, waku_filter_v2/client,
../../../waku/waku_filter_v2/rpc, waku_core
../../../waku/waku_core, ],
../testlib/common, ../testlib/[
../testlib/wakucore common,
wakucore
]
proc newTestWakuFilter*(switch: Switch): Future[WakuFilter] {.async.} = proc newTestWakuFilter*(switch: Switch): Future[WakuFilter] {.async.} =
let let

5
tests/waku_relay/test_all.nim
View File

@ -1,7 +1,6 @@
{.used.} {.used.}
import import
./test_waku_relay,
./test_wakunode_relay,
./test_message_id, ./test_message_id,
./test_protocol ./test_protocol,
./test_wakunode_relay

12
tests/waku_relay/test_message_id.nim
View File

@ -1,12 +1,14 @@
import import
unittest, unittest,
stew/shims/net, stew/[
stew/[results, byteutils] shims/net,
results,
byteutils
],
nimcrypto/sha2,
libp2p/protocols/pubsub/rpc/messages
import import
stew/results,
nimcrypto/sha2,
libp2p/protocols/pubsub/rpc/messages,
../../../waku/waku_relay/message_id, ../../../waku/waku_relay/message_id,
../testlib/sequtils ../testlib/sequtils

232
tests/waku_relay/test_protocol.nim
View File

@ -40,11 +40,14 @@ suite "Waku Relay":
var messageSeq {.threadvar.}: seq[(PubsubTopic, WakuMessage)] var messageSeq {.threadvar.}: seq[(PubsubTopic, WakuMessage)]
var handlerFuture {.threadvar.}: Future[(PubsubTopic, WakuMessage)] var handlerFuture {.threadvar.}: Future[(PubsubTopic, WakuMessage)]
var simpleFutureHandler {.threadvar.}: WakuRelayHandler var simpleFutureHandler {.threadvar.}: WakuRelayHandler
var switch {.threadvar.}: Switch var switch {.threadvar.}: Switch
var peerManager {.threadvar.}: PeerManager var peerManager {.threadvar.}: PeerManager
var node {.threadvar.}: WakuRelay var node {.threadvar.}: WakuRelay
var remotePeerInfo {.threadvar.}: RemotePeerInfo var remotePeerInfo {.threadvar.}: RemotePeerInfo
var peerId {.threadvar.}: PeerId var peerId {.threadvar.}: PeerId
var contentTopic {.threadvar.}: ContentTopic var contentTopic {.threadvar.}: ContentTopic
var pubsubTopic {.threadvar.}: PubsubTopic var pubsubTopic {.threadvar.}: PubsubTopic
var pubsubTopicSeq {.threadvar.}: seq[PubsubTopic] var pubsubTopicSeq {.threadvar.}: seq[PubsubTopic]
@ -85,7 +88,7 @@ suite "Waku Relay":
# Then the message is not published # Then the message is not published
check: check:
not await handlerFuture.withTimeout(3.seconds) not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
asyncTest "Publish with Subscription (Network Size: 1)": asyncTest "Publish with Subscription (Network Size: 1)":
# When subscribing to a Pubsub Topic # When subscribing to a Pubsub Topic
@ -100,7 +103,7 @@ suite "Waku Relay":
discard await node.publish(pubsubTopic, wakuMessage) discard await node.publish(pubsubTopic, wakuMessage)
# Then the message is published # Then the message is published
assert (await handlerFuture.withTimeout(3.seconds)) assert (await handlerFuture.withTimeout(FUTURE_TIMEOUT))
let (topic, msg) = handlerFuture.read() let (topic, msg) = handlerFuture.read()
check: check:
topic == pubsubTopic topic == pubsubTopic
@ -138,8 +141,8 @@ suite "Waku Relay":
# Then the message is published only in the subscribed node # Then the message is published only in the subscribed node
check: check:
not await handlerFuture.withTimeout(3.seconds) not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let (otherTopic1, otherMessage1) = otherHandlerFuture.read() let (otherTopic1, otherMessage1) = otherHandlerFuture.read()
check: check:
@ -154,8 +157,8 @@ suite "Waku Relay":
# Then the message is published only in the subscribed node # Then the message is published only in the subscribed node
check: check:
not await handlerFuture.withTimeout(3.seconds) not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let (otherTopic2, otherMessage2) = otherHandlerFuture.read() let (otherTopic2, otherMessage2) = otherHandlerFuture.read()
check: check:
@ -198,8 +201,8 @@ suite "Waku Relay":
# Then the message is published in both nodes # Then the message is published in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let let
(topic1, message1) = handlerFuture.read() (topic1, message1) = handlerFuture.read()
@ -219,8 +222,8 @@ suite "Waku Relay":
# Then the message is published in both nodes # Then the message is published in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let let
(topic2, message2) = handlerFuture.read() (topic2, message2) = handlerFuture.read()
@ -260,7 +263,7 @@ suite "Waku Relay":
discard await node.publish(pubsubTopic, wakuMessage) discard await node.publish(pubsubTopic, wakuMessage)
# Then the message is published # Then the message is published
check (await handlerFuture.withTimeout(3.seconds)) check (await handlerFuture.withTimeout(FUTURE_TIMEOUT))
let (topic, msg) = handlerFuture.read() let (topic, msg) = handlerFuture.read()
check: check:
topic == pubsubTopic topic == pubsubTopic
@ -315,14 +318,14 @@ suite "Waku Relay":
# Then the validator is ran in the other node, and fails # Then the validator is ran in the other node, and fails
# Not run in the self node # Not run in the self node
check: check:
await validatorFuture.withTimeout(3.seconds) await validatorFuture.withTimeout(FUTURE_TIMEOUT)
validatorFuture.read() == false validatorFuture.read() == false
# And the message is published in the self node, but not in the other node, # And the message is published in the self node, but not in the other node,
# because it doesn't pass the validator check. # because it doesn't pass the validator check.
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
not await otherHandlerFuture.withTimeout(3.seconds) not await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let (topic1, msg1) = handlerFuture.read() let (topic1, msg1) = handlerFuture.read()
# let (otherTopic1, otherMsg1) = otherHandlerFuture.read() # let (otherTopic1, otherMsg1) = otherHandlerFuture.read()
check: check:
@ -341,13 +344,13 @@ suite "Waku Relay":
# Then the validator is ran in the other node, and succeeds # Then the validator is ran in the other node, and succeeds
# Not run in the self node # Not run in the self node
check: check:
await validatorFuture.withTimeout(3.seconds) await validatorFuture.withTimeout(FUTURE_TIMEOUT)
validatorFuture.read() == true validatorFuture.read() == true
# And the message is published in both nodes # And the message is published in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let (topic2, msg2) = handlerFuture.read() let (topic2, msg2) = handlerFuture.read()
let (otherTopic2, otherMsg2) = otherHandlerFuture.read() let (otherTopic2, otherMsg2) = otherHandlerFuture.read()
check: check:
@ -403,8 +406,8 @@ suite "Waku Relay":
# Then the message is published in both nodes # Then the message is published in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
let (topic1, msg1) = handlerFuture.read() let (topic1, msg1) = handlerFuture.read()
let (otherTopic1, otherMsg1) = otherHandlerFuture.read() let (otherTopic1, otherMsg1) = otherHandlerFuture.read()
check: check:
@ -497,12 +500,12 @@ suite "Waku Relay":
# Then the messages are published in all nodes (because it's published in the center node) # Then the messages are published in all nodes (because it's published in the center node)
# Center meaning that all other nodes are connected to this one # Center meaning that all other nodes are connected to this one
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await handlerFuture2.withTimeout(3.seconds) await handlerFuture2.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture1.withTimeout(3.seconds) await otherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture2.withTimeout(3.seconds) await otherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture1.withTimeout(3.seconds) await anotherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture2.withTimeout(3.seconds) await anotherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
let let
(topic1, msg1) = handlerFuture.read() (topic1, msg1) = handlerFuture.read()
@ -553,12 +556,12 @@ suite "Waku Relay":
# Then the message is published in node and otherNode, # Then the message is published in node and otherNode,
# but not in anotherNode because it is not connected anymore # but not in anotherNode because it is not connected anymore
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await handlerFuture2.withTimeout(3.seconds) await handlerFuture2.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture1.withTimeout(3.seconds) await otherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture2.withTimeout(3.seconds) await otherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
not await anotherHandlerFuture1.withTimeout(3.seconds) not await anotherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
not await anotherHandlerFuture2.withTimeout(3.seconds) not await anotherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
let let
(topic3, msg3) = handlerFuture.read() (topic3, msg3) = handlerFuture.read()
@ -596,12 +599,12 @@ suite "Waku Relay":
# Then the messages are only published in anotherNode because it's disconnected from # Then the messages are only published in anotherNode because it's disconnected from
# the rest of the network # the rest of the network
check: check:
not await handlerFuture.withTimeout(3.seconds) not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
not await handlerFuture2.withTimeout(3.seconds) not await handlerFuture2.withTimeout(FUTURE_TIMEOUT)
not await otherHandlerFuture1.withTimeout(3.seconds) not await otherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
not await otherHandlerFuture2.withTimeout(3.seconds) not await otherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture1.withTimeout(3.seconds) await anotherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture2.withTimeout(3.seconds) await anotherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
let let
(anotherTopic3, anotherMsg3) = anotherHandlerFuture1.read() (anotherTopic3, anotherMsg3) = anotherHandlerFuture1.read()
@ -633,12 +636,12 @@ suite "Waku Relay":
# Then the messages are only published in otherNode and node, but not in anotherNode # Then the messages are only published in otherNode and node, but not in anotherNode
# because it's disconnected from the rest of the network # because it's disconnected from the rest of the network
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await handlerFuture2.withTimeout(3.seconds) await handlerFuture2.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture1.withTimeout(3.seconds) await otherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture2.withTimeout(3.seconds) await otherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
not await anotherHandlerFuture1.withTimeout(3.seconds) not await anotherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
not await anotherHandlerFuture2.withTimeout(3.seconds) not await anotherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
let let
(topic5, msg5) = handlerFuture.read() (topic5, msg5) = handlerFuture.read()
@ -683,12 +686,12 @@ suite "Waku Relay":
# even if they're connected like so AnotherNode <-> OtherNode <-> Node, # even if they're connected like so AnotherNode <-> OtherNode <-> Node,
# otherNode doesn't broadcast B topic messages because it's not subscribed to it # otherNode doesn't broadcast B topic messages because it's not subscribed to it
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
not await handlerFuture2.withTimeout(3.seconds) not await handlerFuture2.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture1.withTimeout(3.seconds) await otherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture2.withTimeout(3.seconds) await otherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture1.withTimeout(3.seconds) await anotherHandlerFuture1.withTimeout(FUTURE_TIMEOUT)
await anotherHandlerFuture2.withTimeout(3.seconds) await anotherHandlerFuture2.withTimeout(FUTURE_TIMEOUT)
let let
(topic7, msg7) = handlerFuture.read() (topic7, msg7) = handlerFuture.read()
@ -860,8 +863,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg1) == handlerFuture.read() (pubsubTopic, msg1) == handlerFuture.read()
(pubsubTopic, msg1) == otherHandlerFuture.read() (pubsubTopic, msg1) == otherHandlerFuture.read()
@ -872,8 +875,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg2) == handlerFuture.read() (pubsubTopic, msg2) == handlerFuture.read()
(pubsubTopic, msg2) == otherHandlerFuture.read() (pubsubTopic, msg2) == otherHandlerFuture.read()
@ -884,8 +887,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg3) == handlerFuture.read() (pubsubTopic, msg3) == handlerFuture.read()
(pubsubTopic, msg3) == otherHandlerFuture.read() (pubsubTopic, msg3) == otherHandlerFuture.read()
@ -896,8 +899,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg4) == handlerFuture.read() (pubsubTopic, msg4) == handlerFuture.read()
(pubsubTopic, msg4) == otherHandlerFuture.read() (pubsubTopic, msg4) == otherHandlerFuture.read()
@ -908,8 +911,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg5) == handlerFuture.read() (pubsubTopic, msg5) == handlerFuture.read()
(pubsubTopic, msg5) == otherHandlerFuture.read() (pubsubTopic, msg5) == otherHandlerFuture.read()
@ -920,8 +923,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg6) == handlerFuture.read() (pubsubTopic, msg6) == handlerFuture.read()
(pubsubTopic, msg6) == otherHandlerFuture.read() (pubsubTopic, msg6) == otherHandlerFuture.read()
@ -932,8 +935,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg7) == handlerFuture.read() (pubsubTopic, msg7) == handlerFuture.read()
(pubsubTopic, msg7) == otherHandlerFuture.read() (pubsubTopic, msg7) == otherHandlerFuture.read()
@ -944,8 +947,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg8) == handlerFuture.read() (pubsubTopic, msg8) == handlerFuture.read()
(pubsubTopic, msg8) == otherHandlerFuture.read() (pubsubTopic, msg8) == otherHandlerFuture.read()
@ -956,8 +959,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg9) == handlerFuture.read() (pubsubTopic, msg9) == handlerFuture.read()
(pubsubTopic, msg9) == otherHandlerFuture.read() (pubsubTopic, msg9) == otherHandlerFuture.read()
@ -968,8 +971,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg10) == handlerFuture.read() (pubsubTopic, msg10) == handlerFuture.read()
(pubsubTopic, msg10) == otherHandlerFuture.read() (pubsubTopic, msg10) == otherHandlerFuture.read()
@ -1004,10 +1007,10 @@ suite "Waku Relay":
msg1 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024)) # 1KiB msg1 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024)) # 1KiB
msg2 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(10*1024)) # 10KiB msg2 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(10*1024)) # 10KiB
msg3 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(100*1024)) # 100KiB msg3 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(100*1024)) # 100KiB
msg4 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1023*1024)) # 1MiB - 1B -> Max Size (Inclusive Limit) msg4 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024*1024 - 1)) # 1MiB - 1B -> Max Size (Inclusive Limit)
msg5 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024*1024)) # 1MiB -> Max Size (Exclusive Limit) msg5 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024*1024)) # 1MiB -> Max Size (Exclusive Limit)
msg6 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1025*1024)) # 1MiB + 1B -> Out of Max Size msg6 = fakeWakuMessage(contentTopic=contentTopic, payload=getByteSequence(1024*1024 + 1)) # 1MiB + 1B -> Out of Max Size
# When sending the 1KiB message # When sending the 1KiB message
handlerFuture = newPushHandlerFuture() handlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
@ -1015,11 +1018,11 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg1) == handlerFuture.read() (pubsubTopic, msg1) == handlerFuture.read()
(pubsubTopic, msg1) == otherHandlerFuture.read() (pubsubTopic, msg1) == otherHandlerFuture.read()
# When sending the 10KiB message # When sending the 10KiB message
handlerFuture = newPushHandlerFuture() handlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
@ -1027,23 +1030,23 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg2) == handlerFuture.read() (pubsubTopic, msg2) == handlerFuture.read()
(pubsubTopic, msg2) == otherHandlerFuture.read() (pubsubTopic, msg2) == otherHandlerFuture.read()
# When sending the 100KiB message # When sending the 100KiB message
handlerFuture = newPushHandlerFuture() handlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
discard await node.publish(pubsubTopic, msg3) discard await node.publish(pubsubTopic, msg3)
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg3) == handlerFuture.read() (pubsubTopic, msg3) == handlerFuture.read()
(pubsubTopic, msg3) == otherHandlerFuture.read() (pubsubTopic, msg3) == otherHandlerFuture.read()
# When sending the 1023KiB message # When sending the 1023KiB message
handlerFuture = newPushHandlerFuture() handlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
@ -1051,8 +1054,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg4) == handlerFuture.read() (pubsubTopic, msg4) == handlerFuture.read()
(pubsubTopic, msg4) == otherHandlerFuture.read() (pubsubTopic, msg4) == otherHandlerFuture.read()
@ -1063,8 +1066,8 @@ suite "Waku Relay":
# Then the message is received in self, because there's no checking, but not in other node # Then the message is received in self, because there's no checking, but not in other node
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
not await otherHandlerFuture.withTimeout(3.seconds) not await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg5) == handlerFuture.read() (pubsubTopic, msg5) == handlerFuture.read()
# When sending the 1025KiB message # When sending the 1025KiB message
@ -1074,8 +1077,8 @@ suite "Waku Relay":
# Then the message is received in self, because there's no checking, but not in other node # Then the message is received in self, because there's no checking, but not in other node
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
not await otherHandlerFuture.withTimeout(3.seconds) not await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg6) == handlerFuture.read() (pubsubTopic, msg6) == handlerFuture.read()
# Finally stop the other node # Finally stop the other node
@ -1120,31 +1123,31 @@ suite "Waku Relay":
msg4 = fakeWakuMessage("msg4", pubsubTopic) msg4 = fakeWakuMessage("msg4", pubsubTopic)
discard await node.publish(pubsubTopic, msg1) discard await node.publish(pubsubTopic, msg1)
check await thisHandlerFuture.withTimeout(3.seconds) check await thisHandlerFuture.withTimeout(FUTURE_TIMEOUT)
check await otherHandlerFuture.withTimeout(3.seconds) check await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
thisHandlerFuture = newPushHandlerFuture() thisHandlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
discard await node.publish(pubsubTopic, msg2) discard await node.publish(pubsubTopic, msg2)
check await thisHandlerFuture.withTimeout(3.seconds) check await thisHandlerFuture.withTimeout(FUTURE_TIMEOUT)
check await otherHandlerFuture.withTimeout(3.seconds) check await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
thisHandlerFuture = newPushHandlerFuture() thisHandlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
discard await node.publish(pubsubTopic, msg3) discard await node.publish(pubsubTopic, msg3)
check await thisHandlerFuture.withTimeout(3.seconds) check await thisHandlerFuture.withTimeout(FUTURE_TIMEOUT)
check await otherHandlerFuture.withTimeout(3.seconds) check await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
thisHandlerFuture = newPushHandlerFuture() thisHandlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
discard await node.publish(pubsubTopic, msg4) discard await node.publish(pubsubTopic, msg4)
check: check:
await thisHandlerFuture.withTimeout(3.seconds) await thisHandlerFuture.withTimeout(FUTURE_TIMEOUT)
thisMessageSeq == @[ thisMessageSeq == @[
(pubsubTopic, msg1), (pubsubTopic, msg1),
(pubsubTopic, msg2), (pubsubTopic, msg2),
(pubsubTopic, msg3), (pubsubTopic, msg3),
(pubsubTopic, msg4) (pubsubTopic, msg4)
] ]
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
otherMessageSeq == @[ otherMessageSeq == @[
(pubsubTopic, msg1), (pubsubTopic, msg1),
(pubsubTopic, msg2), (pubsubTopic, msg2),
@ -1194,8 +1197,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg1) == handlerFuture.read() (pubsubTopic, msg1) == handlerFuture.read()
(pubsubTopic, msg1) == otherHandlerFuture.read() (pubsubTopic, msg1) == otherHandlerFuture.read()
@ -1207,8 +1210,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg2) == handlerFuture.read() (pubsubTopic, msg2) == handlerFuture.read()
(pubsubTopic, msg2) == otherHandlerFuture.read() (pubsubTopic, msg2) == otherHandlerFuture.read()
@ -1224,8 +1227,8 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg3) == handlerFuture.read() (pubsubTopic, msg3) == handlerFuture.read()
(pubsubTopic, msg3) == otherHandlerFuture.read() (pubsubTopic, msg3) == otherHandlerFuture.read()
@ -1237,15 +1240,15 @@ suite "Waku Relay":
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg4) == handlerFuture.read() (pubsubTopic, msg4) == handlerFuture.read()
(pubsubTopic, msg4) == otherHandlerFuture.read() (pubsubTopic, msg4) == otherHandlerFuture.read()
# Finally stop the other node # Finally stop the other node
await allFutures(otherSwitch.stop(), otherNode.stop()) await allFutures(otherSwitch.stop(), otherNode.stop())
xasyncTest "Relay can receive messages after subscribing and stopping without unsubscribing": asyncTest "Relay can't receive messages after subscribing and stopping without unsubscribing":
# Given a second node connected to the first one # Given a second node connected to the first one
let let
otherSwitch = newTestSwitch() otherSwitch = newTestSwitch()
@ -1272,7 +1275,6 @@ suite "Waku Relay":
await sleepAsync(500.millis) await sleepAsync(500.millis)
# FIXME: Inconsistent behaviour with Filter protocol.
# Given other node is stopped without unsubscribing # Given other node is stopped without unsubscribing
await allFutures(otherSwitch.stop(), otherNode.stop()) await allFutures(otherSwitch.stop(), otherNode.stop())
@ -1280,22 +1282,20 @@ suite "Waku Relay":
let msg1 = fakeWakuMessage(testMessage, pubsubTopic) let msg1 = fakeWakuMessage(testMessage, pubsubTopic)
discard await node.publish(pubsubTopic, msg1) discard await node.publish(pubsubTopic, msg1)
# Then the message is received in both nodes # Then the message is not received in any node
check: check:
await handlerFuture.withTimeout(3.seconds) await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) not await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg1) == handlerFuture.read() (pubsubTopic, msg1) == handlerFuture.read()
(pubsubTopic, msg1) == otherHandlerFuture.read()
# When sending a message from other node # When sending a message from other node
handlerFuture = newPushHandlerFuture() handlerFuture = newPushHandlerFuture()
otherHandlerFuture = newPushHandlerFuture() otherHandlerFuture = newPushHandlerFuture()
let msg2 = fakeWakuMessage(testMessage, pubsubTopic) let msg2 = fakeWakuMessage(testMessage, pubsubTopic)
discard await otherNode.publish(pubsubTopic, msg2) discard await otherNode.publish(pubsubTopic, msg2)
# Then the message is received in both nodes # Then the message is received in both nodes
check: check:
await handlerFuture.withTimeout(3.seconds) not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
await otherHandlerFuture.withTimeout(3.seconds) await otherHandlerFuture.withTimeout(FUTURE_TIMEOUT)
(pubsubTopic, msg2) == handlerFuture.read()
(pubsubTopic, msg2) == otherHandlerFuture.read() (pubsubTopic, msg2) == otherHandlerFuture.read()

211
tests/waku_relay/test_waku_relay.nim
View File

@ -1,211 +0,0 @@
{.used.}
import
std/[options, sequtils, strutils],
stew/shims/net as stewNet,
testutils/unittests,
chronicles,
chronos,
libp2p/protocols/pubsub/pubsub,
libp2p/protocols/pubsub/rpc/messages
import
../../../waku/node/peer_manager,
../../../waku/waku_core,
../../../waku/waku_relay,
../testlib/common,
../testlib/wakucore,
./utils
suite "Waku Relay":
asyncTest "subscribe and add handler to topics":
## Setup
let nodeA = await newTestWakuRelay()
## Given
let
networkA = "test-network1"
networkB = "test-network2"
## when
discard nodeA.subscribe(networkA, noopRawHandler())
discard nodeA.subscribe(networkB, noopRawHandler())
## Then
check:
nodeA.isSubscribed(networkA)
nodeA.isSubscribed(networkB)
let subscribedTopics = toSeq(nodeA.subscribedTopics)
check:
subscribedTopics.len == 2
subscribedTopics.contains(networkA)
subscribedTopics.contains(networkB)
## Cleanup
await nodeA.stop()
asyncTest "unsubscribe all handlers from topic":
## Setup
let nodeA = await newTestWakuRelay()
## Given
let
networkA = "test-network1"
networkB = "test-network2"
networkC = "test-network3"
discard nodeA.subscribe(networkA, noopRawHandler())
discard nodeA.subscribe(networkB, noopRawHandler())
discard nodeA.subscribe(networkC, noopRawHandler())
let topics = toSeq(nodeA.subscribedTopics)
require:
topics.len == 3
topics.contains(networkA)
topics.contains(networkB)
topics.contains(networkC)
## When
nodeA.unsubscribeAll(networkA)
## Then
check:
nodeA.isSubscribed(networkB)
nodeA.isSubscribed(networkC)
not nodeA.isSubscribed(networkA)
let subscribedTopics = toSeq(nodeA.subscribedTopics)
check:
subscribedTopics.len == 2
subscribedTopics.contains(networkB)
subscribedTopics.contains(networkC)
not subscribedTopics.contains(networkA)
## Cleanup
await nodeA.stop()
asyncTest "publish a message into a topic":
## Setup
let
srcSwitch = newTestSwitch()
srcPeerManager = PeerManager.new(srcSwitch)
srcNode = await newTestWakuRelay(srcSwitch)
dstSwitch = newTestSwitch()
dstPeerManager = PeerManager.new(dstSwitch)
dstNode = await newTestWakuRelay(dstSwitch)
await allFutures(srcSwitch.start(), dstSwitch.start())
let dstPeerInfo = dstPeerManager.switch.peerInfo.toRemotePeerInfo()
let connOk = await srcPeerManager.connectRelay(dstPeerInfo)
require:
connOk == true
## Given
let networkTopic = "test-network1"
let message = fakeWakuMessage()
# Self subscription (triggerSelf = true)
let srcSubsFut = newFuture[(PubsubTopic, WakuMessage)]()
proc srcSubsHandler(topic: PubsubTopic, message: WakuMessage) {.async, gcsafe.} =
srcSubsFut.complete((topic, message))
discard srcNode.subscribe(networkTopic, srcSubsHandler)
# Subscription
let dstSubsFut = newFuture[(PubsubTopic, WakuMessage)]()
proc dstSubsHandler(topic: PubsubTopic, message: WakuMessage) {.async, gcsafe.} =
dstSubsFut.complete((topic, message))
discard dstNode.subscribe(networkTopic, dstSubsHandler)
await sleepAsync(500.millis)
## When
discard await srcNode.publish(networkTopic, message)
## Then
require:
await srcSubsFut.withTimeout(5.seconds)
await dstSubsFut.withTimeout(5.seconds)
let (srcTopic, srcMsg) = srcSubsFut.read()
check:
srcTopic == networkTopic
srcMsg == message
let (dstTopic, dstMsg) = dstSubsFut.read()
check:
dstTopic == networkTopic
dstMsg == message
## Cleanup
await allFutures(srcSwitch.stop(), dstSwitch.stop())
asyncTest "content topic validator as a message subscription filter":
## Setup
let
srcSwitch = newTestSwitch()
srcPeerManager = PeerManager.new(srcSwitch)
srcNode = await newTestWakuRelay(srcSwitch)
dstSwitch = newTestSwitch()
dstPeerManager = PeerManager.new(dstSwitch)
dstNode = await newTestWakuRelay(dstSwitch)
await allFutures(srcSwitch.start(), dstSwitch.start())
let dstPeerInfo = dstPeerManager.switch.peerInfo.toRemotePeerInfo()
let connOk = await srcPeerManager.connectRelay(dstPeerInfo)
require:
connOk == true
## Given
let networkTopic = "test-network1"
let contentTopic = "test-content1"
let message = fakeWakuMessage(contentTopic=contentTopic)
let messages = @[
fakeWakuMessage(contentTopic="any"),
fakeWakuMessage(contentTopic="any"),
fakeWakuMessage(contentTopic="any"),
message,
fakeWakuMessage(contentTopic="any"),
]
# Subscription
let dstSubsFut = newFuture[(PubsubTopic, WakuMessage)]()
proc dstSubsHandler(topic: PubsubTopic, message: WakuMessage) {.async, gcsafe.} =
dstSubsFut.complete((topic, message))
discard dstNode.subscribe(networkTopic, dstSubsHandler)
await sleepAsync(500.millis)
# Validator
proc validator(topic: PubsubTopic, msg: WakuMessage): Future[ValidationResult] {.async.} =
# only relay messages with contentTopic1
if msg.contentTopic != contentTopic:
return ValidationResult.Reject
return ValidationResult.Accept
dstNode.addValidator(networkTopic, validator)
## When
for msg in messages:
discard await srcNode.publish(networkTopic, msg)
## Then
require:
await dstSubsFut.withTimeout(5.seconds)
let (dstTopic, dstMsg) = dstSubsFut.read()
check:
dstTopic == networkTopic
dstMsg == message
## Cleanup
await allFutures(srcSwitch.stop(), dstSwitch.stop())