In-line basic floodsub test

protocol/v2/waku_protocol.nim

@@ -3,11 +3,15 @@
 ## This file should eventually correspond to waku_protocol as RLPx subprotocol.
 ## Instead, it should likely be on top of GossipSub with a similar interface.
 
+import unittest
 import sequtils, tables, options, sets, strutils
 import chronos, chronicles
 import ../../vendor/nim-libp2p/libp2p/protocols/pubsub/pubsub,
        ../../vendor/nim-libp2p/libp2p/protocols/pubsub/pubsubpeer,
        ../../vendor/nim-libp2p/libp2p/protocols/pubsub/floodsub
+import ../../vendor/nim-libp2p/tests/pubsub/utils
+
+# XXX: Hacky in-line test for now
 
 logScope:
     topic = "WakuSub"
@@ -36,3 +40,56 @@ type
 # procCall PubSub(f).publish(topic, data)
 
 # Then we should be able to write tests like floodsub test
+
+# Can also do in-line here
+
+
+# XXX: Testing in-line
+proc waitSub(sender, receiver: auto; key: string) {.async, gcsafe.} =
+  # turn things deterministic
+  # this is for testing purposes only
+  var ceil = 15
+  # TODO: Cast WakuSub
+  let fsub = cast[FloodSub](sender.pubSub.get())
+  while not fsub.floodsub.hasKey(key) or
+        not fsub.floodsub[key].contains(receiver.peerInfo.id):
+    await sleepAsync(100.millis)
+    dec ceil
+    doAssert(ceil > 0, "waitSub timeout!")
+
+proc test(): Future[bool] {.async.} =
+  var completionFut = newFuture[bool]()
+  proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
+    echo "HIT HANDLER", topic
+    check topic == "foobar"
+    completionFut.complete(true)
+
+  let
+    nodes = generateNodes(2)
+    nodesFut = await allFinished(
+      nodes[0].start(),
+      nodes[1].start()
+    )
+
+  await subscribeNodes(nodes)
+
+  await nodes[1].subscribe("foobar", handler)
+  await waitSub(nodes[0], nodes[1], "foobar")
+
+  await nodes[0].publish("foobar", cast[seq[byte]]("Hello!"))
+
+  echo "ok"
+  result = await completionFut.wait(5.seconds)
+
+#  await allFuturesThrowing(
+#    nodes[0].stop(),
+#    nodes[1].stop()
+#  )
+#
+#  for fut in nodesFut:
+#    let res = fut.read()
+#    await allFuturesThrowing(res)
+
+echo "Starting"
+var res = waitFor test()
+echo "Done with res: ", $res