From 7231fa9bb63322b7c052b8359cb14a31403f3b31 Mon Sep 17 00:00:00 2001
From: Antonio Pancorbo <48168255+apancorb@users.noreply.github.com>
Date: Wed, 11 Dec 2024 12:01:40 +0000
Subject: [PATCH] feat(watcher): update unit tests to accomdate BitcoinWrapper
 listener logic to register svp spend tx

---
 .../watcher/FederationWatcherListenerImplTest.java         | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/test/java/co/rsk/federate/watcher/FederationWatcherListenerImplTest.java b/src/test/java/co/rsk/federate/watcher/FederationWatcherListenerImplTest.java
index 7e1ffde5b..b7cd365fe 100644
--- a/src/test/java/co/rsk/federate/watcher/FederationWatcherListenerImplTest.java
+++ b/src/test/java/co/rsk/federate/watcher/FederationWatcherListenerImplTest.java
@@ -10,6 +10,7 @@
 import co.rsk.bitcoinj.core.BtcECKey;
 import co.rsk.bitcoinj.core.NetworkParameters;
 import co.rsk.federate.BtcToRskClient;
+import co.rsk.federate.bitcoin.BitcoinWrapper;
 import co.rsk.federate.btcreleaseclient.BtcReleaseClient;
 import co.rsk.peg.federation.Federation;
 import co.rsk.peg.federation.FederationArgs;
@@ -38,6 +39,7 @@ class FederationWatcherListenerImplTest {
     private BtcToRskClient btcToRskClientActive;
     private BtcToRskClient btcToRskClientRetiring;
     private BtcReleaseClient btcReleaseClient;
+    private BitcoinWrapper bitcoinWrapper;
     private FederationWatcherListener federationWatcherListener;
 
     @BeforeEach
@@ -45,8 +47,9 @@ void setUp() {
         btcToRskClientActive = mock(BtcToRskClient.class);
         btcToRskClientRetiring = mock(BtcToRskClient.class);
         btcReleaseClient = mock(BtcReleaseClient.class);
+        bitcoinWrapper = mock(BitcoinWrapper.class);
         federationWatcherListener = new FederationWatcherListenerImpl(
-            btcToRskClientActive, btcToRskClientRetiring, btcReleaseClient);
+            btcToRskClientActive, btcToRskClientRetiring, btcReleaseClient, bitcoinWrapper);
     }
    
     @Test
@@ -99,6 +102,7 @@ void onProposedFederationChange_whenNewProposedFederationIsNull_shouldNotStartCl
 
         // Assert
         verify(btcReleaseClient, never()).start(any(Federation.class));
+        verify(bitcoinWrapper, never()).addFederationListener(any(Federation.class), any(BtcToRskClient.class));
     }
 
     @Test
@@ -108,6 +112,7 @@ void onProposedFederationChange_whenNewProposedFederationIsValid_shouldStartClie
 
         // Assert
         verify(btcReleaseClient).start(FEDERATION);
+        verify(bitcoinWrapper).addFederationListener(FEDERATION, btcToRskClientActive);
     }
 
     @Test