Skip to content

Commit

Permalink
fix: tests
Browse files Browse the repository at this point in the history
  • Loading branch information
rymnc committed Feb 9, 2024
1 parent 39e473b commit 2c4eaf0
Show file tree
Hide file tree
Showing 4 changed files with 20 additions and 20 deletions.
4 changes: 2 additions & 2 deletions apps/chat2/chat2.nim
Original file line number Diff line number Diff line change
Expand Up @@ -189,9 +189,9 @@ proc publish(c: Chat, line: string) =
# we should check the message content topic as well
let appendRes = c.node.wakuRlnRelay.appendRLNProof(message, float64(time))
if appendRes.isErr():
debug "could not append rate limit proof to the message", success=success
debug "could not append rate limit proof to the message"
else:
debug "rate limit proof is appended to the message", success=success
debug "rate limit proof is appended to the message"
let decodeRes = RateLimitProof.init(message.proof)
if decodeRes.isErr():
error "could not decode the RLN proof"
Expand Down
12 changes: 6 additions & 6 deletions tests/node/test_wakunode_relay_rln.nim
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ proc sendRlnMessage(
payload: seq[byte] = "Hello".toBytes(),
): Future[bool] {.async.} =
var message = WakuMessage(payload: payload, contentTopic: contentTopic)
doAssert(not client.wakuRlnRelay.appendRLNProof(message, epochTime()).isErr())
doAssert(client.wakuRlnRelay.appendRLNProof(message, epochTime()).isOk())
discard await client.publish(some(pubsubTopic), message)
let isCompleted = await completionFuture.withTimeout(FUTURE_TIMEOUT)
return isCompleted
Expand Down Expand Up @@ -249,18 +249,18 @@ suite "Waku RlnRelay - End to End":
WakuMessage(payload: @payload150kibPlus, contentTopic: contentTopic)

doAssert(
not client.wakuRlnRelay.appendRLNProof(message1b, epoch + EpochUnitSeconds * 0).isErr()
client.wakuRlnRelay.appendRLNProof(message1b, epoch + EpochUnitSeconds * 0).isOk()
)
doAssert(
not client.wakuRlnRelay.appendRLNProof(message1kib, epoch + EpochUnitSeconds * 1).isErr()
client.wakuRlnRelay.appendRLNProof(message1kib, epoch + EpochUnitSeconds * 1).isOk()
)
doAssert(
not client.wakuRlnRelay.appendRLNProof(message150kib, epoch + EpochUnitSeconds * 2).isErr()
client.wakuRlnRelay.appendRLNProof(message150kib, epoch + EpochUnitSeconds * 2).isOk()
)
doAssert(
not client.wakuRlnRelay.appendRLNProof(
client.wakuRlnRelay.appendRLNProof(
message151kibPlus, epoch + EpochUnitSeconds * 3
).isErr()
).isOk()
)

# When sending the 1B message
Expand Down
6 changes: 3 additions & 3 deletions tests/waku_rln_relay/test_waku_rln_relay.nim
Original file line number Diff line number Diff line change
Expand Up @@ -687,9 +687,9 @@ suite "Waku rln relay":

# ensure proofs are added
require:
proofAdded1
proofAdded2
proofAdded3
proofAdded1.isOk()
proofAdded2.isOk()
proofAdded3.isOk()

# validate messages
# validateMessage proc checks the validity of the message fields and adds it to the log (if valid)
Expand Down
18 changes: 9 additions & 9 deletions tests/waku_rln_relay/test_wakunode_rln_relay.nim
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ procSuite "WakuNode - RLN relay":

# prepare the epoch
var message = WakuMessage(payload: @payload, contentTopic: contentTopic)
doAssert(node1.wakuRlnRelay.appendRLNProof(message, epochTime()))
doAssert(node1.wakuRlnRelay.appendRLNProof(message, epochTime()).isOk())


## node1 publishes a message with a rate limit proof, the message is then relayed to node2 which in turn
Expand Down Expand Up @@ -155,12 +155,12 @@ procSuite "WakuNode - RLN relay":

for i in 0..<3:
var message = WakuMessage(payload: ("Payload_" & $i).toBytes(), contentTopic: contentTopics[0])
doAssert(nodes[0].wakuRlnRelay.appendRLNProof(message, epochTime))
doAssert(nodes[0].wakuRlnRelay.appendRLNProof(message, epochTime).isOk())
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))
doAssert(nodes[1].wakuRlnRelay.appendRLNProof(message, epochTime).isOk())
messages2.add(message)

# publish 3 messages from node[0] (last 2 are spam, window is 10 secs)
Expand Down Expand Up @@ -346,9 +346,9 @@ procSuite "WakuNode - RLN relay":

# check proofs are added correctly
check:
proofAdded1
proofAdded2
proofAdded3
proofAdded1.isOk()
proofAdded2.isOk()
proofAdded3.isOk()

# relay handler for node3
var completionFut1 = newFuture[bool]()
Expand Down Expand Up @@ -452,9 +452,9 @@ procSuite "WakuNode - RLN relay":

# check proofs are added correctly
check:
proofAdded1
proofAdded2
proofAdded3
proofAdded1.isOk()
proofAdded2.isOk()
proofAdded3.isOk()

# relay handler for node2
var completionFut1 = newFuture[bool]()
Expand Down

0 comments on commit 2c4eaf0

Please sign in to comment.