mirror of
https://github.com/waku-org/nwaku.git
synced 2025-01-28 15:46:33 +00:00
77 lines
2.7 KiB
Nim
77 lines
2.7 KiB
Nim
|
import
|
||
|
|
os, strformat, chronicles, json_rpc/[rpcclient, rpcserver], nimcrypto/sysrand,
|
||
|
|
eth/common as eth_common, eth/keys,
|
||
|
|
../protocol/waku_protocol, ./rpc/[hexstrings, rpc_types],
|
||
|
|
options as what # TODO: Huh? Redefinition?
|
||
|
|
|
||
|
|
from os import DirSep
|
||
|
|
from strutils import rsplit
|
||
|
|
template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]
|
||
|
|
|
||
|
|
const sigWakuPath = &"{sourceDir}{DirSep}rpc{DirSep}wakucallsigs.nim"
|
||
|
|
createRpcSigs(RpcHttpClient, sigWakuPath)
|
||
|
|
|
||
|
|
const topicAmount = 100
|
||
|
|
|
||
|
|
let
|
||
|
|
trafficNode = newRpcHttpClient()
|
||
|
|
lightNode = newRpcHttpClient()
|
||
|
|
lightNode2 = newRpcHttpClient()
|
||
|
|
|
||
|
|
waitFor lightNode.connect("localhost", Port(8545))
|
||
|
|
waitFor lightNode2.connect("localhost", Port(8546))
|
||
|
|
waitFor trafficNode.connect("localhost", Port(8548))
|
||
|
|
|
||
|
|
proc generateTopics(amount = topicAmount): seq[waku_protocol.Topic] =
|
||
|
|
var topic: waku_protocol.Topic
|
||
|
|
for i in 0..<amount:
|
||
|
|
if randomBytes(topic) != 4:
|
||
|
|
raise newException(ValueError, "Generation of random topic failed.")
|
||
|
|
result.add(topic)
|
||
|
|
|
||
|
|
let
|
||
|
|
symKey = "0x0000000000000000000000000000000000000000000000000000000000000001"
|
||
|
|
topics = generateTopics()
|
||
|
|
symKeyID = waitFor lightNode.waku_addSymKey(symKey)
|
||
|
|
options = WakuFilterOptions(symKeyID: some(symKeyID),
|
||
|
|
topics: some(topics))
|
||
|
|
filterID = waitFor lightNode.waku_newMessageFilter(options)
|
||
|
|
|
||
|
|
symKeyID2 = waitFor lightNode2.waku_addSymKey(symKey)
|
||
|
|
options2 = WakuFilterOptions(symKeyID: some(symKeyID2),
|
||
|
|
topics: some(topics))
|
||
|
|
filterID2 = waitFor lightNode2.waku_newMessageFilter(options2)
|
||
|
|
|
||
|
|
symkeyID3 = waitFor trafficNode.waku_addSymKey(symKey)
|
||
|
|
|
||
|
|
var message = WakuPostMessage(symKeyID: some(symkeyID3),
|
||
|
|
ttl: 30,
|
||
|
|
topic: some(topics[0]),
|
||
|
|
payload: "0x45879632".HexDataStr,
|
||
|
|
powTime: 1.0,
|
||
|
|
powTarget: 0.002)
|
||
|
|
|
||
|
|
info "Posting envelopes on all subscribed topics"
|
||
|
|
for i in 0..<topicAmount:
|
||
|
|
message.topic = some(topics[i])
|
||
|
|
discard waitFor trafficNode.waku_post(message)
|
||
|
|
|
||
|
|
# Check if the subscription for the topics works
|
||
|
|
|
||
|
|
waitFor sleepAsync(1000.milliseconds) # This is a bit brittle
|
||
|
|
|
||
|
|
let
|
||
|
|
messages = waitFor lightNode.waku_getFilterMessages(filterID)
|
||
|
|
messages2 = waitFor lightNode2.waku_getFilterMessages(filterID2)
|
||
|
|
|
||
|
|
if messages.len != topicAmount or messages2.len != topicAmount:
|
||
|
|
error "Light node did not receive envelopes on all subscribed topics",
|
||
|
|
lightnode1=messages.len, lightnode2=messages2.len
|
||
|
|
quit 1
|
||
|
|
|
||
|
|
info "Received envelopes on all subscribed topics"
|
||
|
|
|
||
|
|
# Generate test traffic on node
|
||
|
|
discard waitFor trafficNode.wakusim_generateRandomTraffic(10_000)
|
||
|
|
info "Started random traffic generation"
|