nwaku/tests/v2/test_waku_bridge.nim

{.used.}

import
  std/[unittest, strutils],
  chronicles, chronos, stew/shims/net as stewNet, stew/byteutils,
  libp2p/crypto/crypto,
  libp2p/crypto/secp,
  libp2p/peerid,
  libp2p/multiaddress,
  libp2p/switch,
  libp2p/protocols/pubsub/rpc/messages,
  libp2p/protocols/pubsub/pubsub,
  eth/p2p,
  eth/keys,
  ../../waku/common/wakubridge,
  ../../waku/v1/protocol/waku_protocol,
  ../../waku/v2/protocol/waku_message,
  ../../waku/v2/protocol/waku_store/waku_store,
  ../../waku/v2/protocol/waku_filter/waku_filter,
  ../../waku/v2/node/[wakunode2, waku_payload],
  ../test_helpers

procSuite "WakuBridge":
  let rng = keys.newRng()

  asyncTest "Messages are bridged between Waku v1 and Waku v2":
    let
      # Bridge
      nodev1Key = keys.KeyPair.random(rng[])
      nodev2Key = crypto.PrivateKey.random(Secp256k1, rng[])[]
      bridge = WakuBridge.new(
          nodev1Key= nodev1Key,
          nodev1Address = localAddress(30303),
          powRequirement = 0.002,
          rng = rng,
          nodev2Key = nodev2Key,
          nodev2BindIp = ValidIpAddress.init("0.0.0.0"), nodev2BindPort= Port(60000))
      
      # Waku v1 node
      v1Node = setupTestNode(rng, Waku)

      # Waku v2 node
      v2NodeKey = crypto.PrivateKey.random(Secp256k1, rng[])[]
      v2Node = WakuNode.init(v2NodeKey, ValidIpAddress.init("0.0.0.0"), Port(60002))

      topic = [byte 0x00, 0, 0, byte 0x01]
      contentTopic = ContentTopic(1)
      payloadV1 = "hello from V1".toBytes()
      payloadV2 = "hello from V2".toBytes()
      message = WakuMessage(payload: payloadV2, contentTopic: contentTopic)

    await bridge.start()

    await v2Node.start()
    v2Node.mountRelay(@[defaultBridgeTopic])

    discard await v1Node.rlpxConnect(newNode(bridge.nodev1.toENode()))
    await v2Node.connectToNodes(@[bridge.nodev2.peerInfo])

    var completionFut = newFuture[bool]()
    proc relayHandler(topic: string, data: seq[byte]) {.async, gcsafe.} =
      let msg = WakuMessage.init(data)
      if msg.isOk() and msg.value().version == 1:
        check:
          # Message fields are as expected
          msg.value().contentTopic == contentTopic # Topic translation worked
          string.fromBytes(msg.value().payload).contains("from V1")
        completionFut.complete(true)

    v2Node.subscribe(defaultBridgeTopic, relayHandler)

    await sleepAsync(2000.millis)

    # Test bridging from V2 to V1
    await v2Node.publish(defaultBridgeTopic, message)

    await sleepAsync(2000.millis)

    check:
      # v1Node received message published by v2Node
      v1Node.protocolState(Waku).queue.items.len == 1

    let msg = v1Node.protocolState(Waku).queue.items[0]

    check:
      # Message fields are as expected
      msg.env.topic == topic # Topic translation worked
      string.fromBytes(msg.env.data).contains("from V2")
    
    # Test bridging from V1 to V2
    check:
      v1Node.postMessage(ttl = 5,
                         topic = topic,
                         payload = payloadV1) == true

      # v2Node received payload published by v1Node
      await completionFut.withTimeout(5.seconds)

    await bridge.stop()