fix(tests): fix flaky test (#1972)

This commit is contained in:
Alvaro Revuelta 2023-08-31 12:24:41 +02:00 committed by GitHub
parent d262837ea5
commit f262397d71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 9 deletions

View File

@ -150,17 +150,28 @@ procSuite "WakuNode - RLN relay":
nodes[2].subscribe(pubsubTopics[1], relayHandler) nodes[2].subscribe(pubsubTopics[1], relayHandler)
await sleepAsync(1000.millis) await sleepAsync(1000.millis)
# publish 3 messages from node[0] (last 2 are spam, window is 10 secs) # generate some messages with rln proofs first. generating
for i in 0..<3: # the proof takes some time, so this is done before publishing
var message1 = WakuMessage(payload: ("Payload_" & $i).toBytes(), contentTopic: contentTopics[0]) # to avoid blocking the test
doAssert(nodes[0].wakuRlnRelay.appendRLNProof(message1, epochTime())) var messages1: seq[WakuMessage] = @[]
await nodes[0].publish(pubsubTopics[0], message1) var messages2: seq[WakuMessage] = @[]
let epochTime = epochTime()
# publish 3 messages from node[1] (last 2 are spam, window is 10 secs)
for i in 0..<3: for i in 0..<3:
var message2 = WakuMessage(payload: ("Payload_" & $i).toBytes(), contentTopic: contentTopics[1]) var message = WakuMessage(payload: ("Payload_" & $i).toBytes(), contentTopic: contentTopics[0])
doAssert(nodes[1].wakuRlnRelay.appendRLNProof(message2, epochTime())) doAssert(nodes[0].wakuRlnRelay.appendRLNProof(message, epochTime))
await nodes[1].publish(pubsubTopics[1], message2) messages1.add(message)
for i in 0..<3:
var message = WakuMessage(payload: ("Payload_" & $i).toBytes(), contentTopic: contentTopics[1])
doAssert(nodes[1].wakuRlnRelay.appendRLNProof(message, epochTime))
messages2.add(message)
# publish 3 messages from node[0] (last 2 are spam, window is 10 secs)
# publish 3 messages from node[1] (last 2 are spam, window is 10 secs)
for msg in messages1: await nodes[0].publish(pubsubTopics[0], msg)
for msg in messages2: await nodes[1].publish(pubsubTopics[1], msg)
# wait for gossip to propagate # wait for gossip to propagate
await sleepAsync(5000.millis) await sleepAsync(5000.millis)