diff --git a/tests/waku_lightpush/lightpush_utils.nim b/tests/waku_lightpush/lightpush_utils.nim index a2a13a415..9fe9542a1 100644 --- a/tests/waku_lightpush/lightpush_utils.nim +++ b/tests/waku_lightpush/lightpush_utils.nim @@ -27,5 +27,7 @@ proc newTestWakuLightpushNode*( proc newTestWakuLightpushClient*(switch: Switch): WakuLightPushClient = let peerManager = PeerManager.new(switch) - let reputationManager = ReputationManager.new() - WakuLightPushClient.new(peerManager, reputationManager, rng) + var reputationManager = none(ReputationManager) + when defined(reputation): + reputationManager = some(ReputationManager.new()) + WakuLightPushClient.new(peerManager, rng, reputationManager) diff --git a/waku/waku_lightpush/client.nim b/waku/waku_lightpush/client.nim index 66e9fa856..e63411d3c 100644 --- a/waku/waku_lightpush/client.nim +++ b/waku/waku_lightpush/client.nim @@ -26,12 +26,15 @@ type WakuLightPushClient* = ref object proc new*( T: type WakuLightPushClient, peerManager: PeerManager, - reputationManager: ReputationManager, rng: ref rand.HmacDrbgContext, + reputationManager: Option[ReputationManager] = none(ReputationManager), ): T = - WakuLightPushClient( - peerManager: peerManager, reputationManager: reputationManager, rng: rng - ) + if reputationManager.isSome: + WakuLightPushClient( + peerManager: peerManager, rng: rng, reputationManager: reputationManager.get() + ) + else: + WakuLightPushClient(peerManager: peerManager, rng: rng) proc addPublishObserver*(wl: WakuLightPushClient, obs: PublishObserver) = wl.publishObservers.add(obs)