diff --git a/tests/waku_lightpush/lightpush_utils.nim b/tests/waku_lightpush/lightpush_utils.nim index 4be10ea2b..5c3e78764 100644 --- a/tests/waku_lightpush/lightpush_utils.nim +++ b/tests/waku_lightpush/lightpush_utils.nim @@ -30,9 +30,4 @@ proc newTestWakuLightpushClient*( reputationEnabled: bool = false ): WakuLightPushClient = let peerManager = PeerManager.new(switch) - let reputationManager = - if reputationEnabled: - some(ReputationManager.new()) - else: - none(ReputationManager) - WakuLightPushClient.new(peerManager, rng, reputationManager) + WakuLightPushClient.new(peerManager, rng, reputationEnabled) diff --git a/waku/waku_lightpush/client.nim b/waku/waku_lightpush/client.nim index ab513ab8f..a7f32076c 100644 --- a/waku/waku_lightpush/client.nim +++ b/waku/waku_lightpush/client.nim @@ -27,8 +27,13 @@ proc new*( T: type WakuLightPushClient, peerManager: PeerManager, rng: ref rand.HmacDrbgContext, - reputationManager: Option[ReputationManager], + reputationEnabled: bool, ): T = + let reputationManager = + if reputationEnabled: + some(ReputationManager.new()) + else: + none(ReputationManager) WakuLightPushClient( peerManager: peerManager, rng: rng, reputationManager: reputationManager )