logos-messaging-nim/tests/waku_lightpush/lightpush_utils.nim

{.used.}

import std/options, chronicles, chronos, libp2p/crypto/crypto

import
  waku/node/peer_manager,
  waku/waku_core,
  waku/waku_lightpush,
  waku/waku_lightpush/[client, common],
  waku/common/rate_limit/setting,
  ../testlib/[common, wakucore],
  waku/incentivization/reputation_manager

proc newTestWakuLightpushNode*(
    switch: Switch,
    handler: PushMessageHandler,
    rateLimitSetting: Option[RateLimitSetting] = none[RateLimitSetting](),
): Future[WakuLightPush] {.async.} =
  let
    peerManager = PeerManager.new(switch)
    proto = WakuLightPush.new(peerManager, rng, handler, rateLimitSetting)

  await proto.start()
  switch.mount(proto)

  return proto

proc newTestWakuLightpushClient*(
  switch: Switch,
  reputationEnabled: bool = false
  ): WakuLightPushClient =
  let peerManager = PeerManager.new(switch)
  let reputationManager =
    if reputationEnabled:
      some(ReputationManager.new())
    else:
      none(ReputationManager)
  WakuLightPushClient.new(peerManager, rng, reputationManager)