Fix pubsub reconnect failure handling

magicblock-chainlink/src/remote_account_provider/chain_pubsub_client.rs

@@ -286,7 +286,7 @@ impl ReconnectableClient for ChainPubsubClientImpl {
 pub mod mock {
     use std::{
         collections::HashSet,
-        sync::atomic::{AtomicU64, Ordering as AtomicOrdering},
+        sync::atomic::{AtomicU16, AtomicU64, Ordering as AtomicOrdering},
         time::Duration,
     };
 
@@ -319,13 +319,16 @@ pub mod mock {
         subscribe_blocked: Arc<Mutex<bool>>,
         subscribe_attempts: Arc<AtomicU64>,
         subscribe_notify: Arc<Notify>,
+        client_id: String,
     }
 
     impl ChainPubsubClientMock {
         pub fn new(
             updates_sndr: mpsc::Sender<SubscriptionUpdate>,
             updates_rcvr: mpsc::Receiver<SubscriptionUpdate>,
         ) -> Self {
+            static CLIENT_ID: AtomicU16 = AtomicU16::new(0);
+
             Self {
                 updates_sndr,
                 updates_rcvr: Arc::new(Mutex::new(Some(updates_rcvr))),
@@ -339,6 +342,10 @@ pub mod mock {
                 subscribe_blocked: Arc::new(Mutex::new(false)),
                 subscribe_attempts: Arc::new(AtomicU64::new(0)),
                 subscribe_notify: Arc::new(Notify::new()),
+                client_id: format!(
+                    "mock:{}",
+                    CLIENT_ID.fetch_add(1, AtomicOrdering::SeqCst)
+                ),
             }
         }
 
@@ -430,6 +437,10 @@ pub mod mock {
             }
         }
 
+        pub fn subscribe_attempts(&self) -> u64 {
+            self.subscribe_attempts.load(AtomicOrdering::SeqCst)
+        }
+
         pub fn is_connected_and_resubscribed(&self) -> bool {
             *self.connected.lock()
                 && self.subscribed_pubkeys.lock().len()
@@ -544,7 +555,7 @@ pub mod mock {
         }
 
         fn id(&self) -> &str {
-            "ChainPubsubClientMock"
+            &self.client_id
         }
     }