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

fix: fix flaky test for dos topic (#1747)

This commit is contained in:
Alvaro Revuelta 2023-05-19 12:34:02 +02:00 committed by GitHub
parent 5eae60e817
commit 46e231d03f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/wakunode2/test_validators.nim
View File

@ -136,9 +136,9 @@ suite "WakuNode2 - Validators":
for node in nodes: node.wakuRelay.subscribe(spamProtectedTopic, handler) for node in nodes: node.wakuRelay.subscribe(spamProtectedTopic, handler)
await sleepAsync(500.millis) await sleepAsync(500.millis)
# Each node sends 10 messages, signed but with a non-whitelisted key (total = 50) # Each node sends 5 messages, signed but with a non-whitelisted key (total = 25)
for i in 0..<5: for i in 0..<5:
for j in 0..<10: for j in 0..<5:
var msg = WakuMessage( var msg = WakuMessage(
payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic, payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic,
version: 2, timestamp: now(), ephemeral: true) version: 2, timestamp: now(), ephemeral: true)
@ -148,34 +148,34 @@ suite "WakuNode2 - Validators":
await nodes[i].publish(spamProtectedTopic, msg) await nodes[i].publish(spamProtectedTopic, msg)
# Each node sends 10 messages that are not signed (total = 50) # Each node sends 5 messages that are not signed (total = 25)
for i in 0..<5: for i in 0..<5:
for j in 0..<10: for j in 0..<5:
let unsignedMessage = WakuMessage( let unsignedMessage = WakuMessage(
payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic, payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic,
version: 2, timestamp: now(), ephemeral: true) version: 2, timestamp: now(), ephemeral: true)
await nodes[i].publish(spamProtectedTopic, unsignedMessage) await nodes[i].publish(spamProtectedTopic, unsignedMessage)
# Each node sends 10 messages that dont contain timestamp (total = 50) # Each node sends 5 messages that dont contain timestamp (total = 25)
for i in 0..<5: for i in 0..<5:
for j in 0..<10: for j in 0..<5:
let unsignedMessage = WakuMessage( let unsignedMessage = WakuMessage(
payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic, payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic,
version: 2, timestamp: 0, ephemeral: true) version: 2, timestamp: 0, ephemeral: true)
await nodes[i].publish(spamProtectedTopic, unsignedMessage) await nodes[i].publish(spamProtectedTopic, unsignedMessage)
# Each node sends 10 messages way BEFORE than the current timestmap (total = 50) # Each node sends 5 messages way BEFORE than the current timestmap (total = 25)
for i in 0..<5: for i in 0..<5:
for j in 0..<10: for j in 0..<5:
let beforeTimestamp = now() - getNanosecondTime(6*60) let beforeTimestamp = now() - getNanosecondTime(6*60)
let unsignedMessage = WakuMessage( let unsignedMessage = WakuMessage(
payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic, payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic,
version: 2, timestamp: beforeTimestamp, ephemeral: true) version: 2, timestamp: beforeTimestamp, ephemeral: true)
await nodes[i].publish(spamProtectedTopic, unsignedMessage) await nodes[i].publish(spamProtectedTopic, unsignedMessage)
# Each node sends 10 messages way LATER than the current timestmap (total = 50) # Each node sends 5 messages way LATER than the current timestmap (total = 25)
for i in 0..<5: for i in 0..<5:
for j in 0..<10: for j in 0..<5:
let afterTimestamp = now() - getNanosecondTime(6*60) let afterTimestamp = now() - getNanosecondTime(6*60)
let unsignedMessage = WakuMessage( let unsignedMessage = WakuMessage(
payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic, payload: urandom(1*(10^3)), contentTopic: spamProtectedTopic,
@ -185,12 +185,12 @@ suite "WakuNode2 - Validators":
# Wait for gossip # Wait for gossip
await sleepAsync(2.seconds) await sleepAsync(2.seconds)
# Since we have a full mesh with 5 nodes and each one publishes 50+50+50+50+50 msgs # Since we have a full mesh with 5 nodes and each one publishes 25+25+25+25+25 msgs
# there are 1250 messages being sent. # there are 625 messages being sent.
# 250 are received ok in the handler (first hop) # 125 are received ok in the handler (first hop)
# 1000 are are wrong so rejected (rejected not relayed) # 500 are are wrong so rejected (rejected not relayed)
check: check:
msgReceived == 250 msgReceived == 125
var msgRejected = 0 var msgRejected = 0
for i in 0..<5: for i in 0..<5:
@ -198,7 +198,7 @@ suite "WakuNode2 - Validators":
msgRejected += v.topicInfos[spamProtectedTopic].invalidMessageDeliveries.int msgRejected += v.topicInfos[spamProtectedTopic].invalidMessageDeliveries.int
check: check:
msgRejected == 1000 msgRejected == 500
await allFutures(nodes.mapIt(it.stop())) await allFutures(nodes.mapIt(it.stop()))