mirror of https://github.com/waku-org/nwaku.git
77 lines
2.4 KiB
Nim
77 lines
2.4 KiB
Nim
|
{.used.}
|
||
|
|
|
||
|
|
import
|
||
|
|
std/[options, tables, sets],
|
||
|
|
testutils/unittests, chronos, chronicles,
|
||
|
|
libp2p/switch,
|
||
|
|
libp2p/protobuf/minprotobuf,
|
||
|
|
libp2p/stream/[bufferstream, connection],
|
||
|
|
libp2p/crypto/crypto,
|
||
|
|
libp2p/multistream,
|
||
|
|
../../waku/v2/node/peer_manager/peer_manager,
|
||
|
|
../../waku/v2/protocol/message_notifier,
|
||
|
|
../../waku/v2/protocol/waku_lightpush/waku_lightpush,
|
||
|
|
../test_helpers, ./utils
|
||
|
|
|
||
|
|
procSuite "Waku Light Push":
|
||
|
|
|
||
|
|
asyncTest "handle light push request":
|
||
|
|
const defaultTopic = "/waku/2/default-waku/proto"
|
||
|
|
|
||
|
|
let
|
||
|
|
key = PrivateKey.random(ECDSA, rng[]).get()
|
||
|
|
peer = PeerInfo.init(key)
|
||
|
|
contentTopic = ContentTopic("/waku/2/default-content/proto")
|
||
|
|
post = WakuMessage(payload: @[byte 1, 2, 3], contentTopic: contentTopic)
|
||
|
|
|
||
|
|
var dialSwitch = newStandardSwitch()
|
||
|
|
discard await dialSwitch.start()
|
||
|
|
|
||
|
|
var listenSwitch = newStandardSwitch(some(key))
|
||
|
|
discard await listenSwitch.start()
|
||
|
|
|
||
|
|
var responseRequestIdFuture = newFuture[string]()
|
||
|
|
var completionFut = newFuture[bool]()
|
||
|
|
|
||
|
|
proc handle(requestId: string, msg: PushRequest) {.gcsafe, closure.} =
|
||
|
|
# TODO Success return here
|
||
|
|
debug "handle push req"
|
||
|
|
check:
|
||
|
|
1 == 0
|
||
|
|
responseRequestIdFuture.complete(requestId)
|
||
|
|
|
||
|
|
# FIXME Unclear how we want to use subscriptions, if at all
|
||
|
|
let
|
||
|
|
proto = WakuLightPush.init(PeerManager.new(dialSwitch), crypto.newRng(), handle)
|
||
|
|
wm = WakuMessage(payload: @[byte 1, 2, 3], contentTopic: contentTopic)
|
||
|
|
rpc = PushRequest(pubSubTopic: defaultTopic, message: wm)
|
||
|
|
|
||
|
|
dialSwitch.mount(proto)
|
||
|
|
proto.setPeer(listenSwitch.peerInfo)
|
||
|
|
|
||
|
|
|
||
|
|
# TODO Can possibly get rid of this if it isn't dynamic
|
||
|
|
proc requestHandle(requestId: string, msg: PushRequest) {.gcsafe, closure.} =
|
||
|
|
debug "push request handler"
|
||
|
|
# TODO: Also relay message
|
||
|
|
# TODO: Here we want to send back response with is_success true
|
||
|
|
discard
|
||
|
|
|
||
|
|
let
|
||
|
|
proto2 = WakuLightPush.init(PeerManager.new(listenSwitch), crypto.newRng(), requestHandle)
|
||
|
|
|
||
|
|
listenSwitch.mount(proto2)
|
||
|
|
|
||
|
|
# FIXME Don't think this will be hit yet
|
||
|
|
proc handler(response: PushResponse) {.gcsafe, closure.} =
|
||
|
|
debug "push response handler, expecting false"
|
||
|
|
check:
|
||
|
|
response.isSuccess == false
|
||
|
|
completionFut.complete(true)
|
||
|
|
|
||
|
|
await proto.request(rpc, handler)
|
||
|
|
await sleepAsync(2.seconds)
|
||
|
|
|
||
|
|
check:
|
||
|
|
(await completionFut.withTimeout(5.seconds)) == true
|