Export RemoveSub

smartcontractkit · Jan 10, 2025 · 25fe8d0 · 25fe8d0
1 parent a012866
commit 25fe8d0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/multinode/adaptor.go b/multinode/adaptor.go
@@ -80,7 +80,7 @@ func (m *Adapter[RPC, HEAD]) RegisterSub(sub *ManagedSubscription, stopInFLightC
 	return nil
 }
 
-func (m *Adapter[RPC, HEAD]) removeSub(sub Subscription) {
+func (m *Adapter[RPC, HEAD]) RemoveSub(sub Subscription) {
 	m.subsSliceMu.Lock()
 	defer m.subsSliceMu.Unlock()
 	delete(m.subs, sub)
@@ -143,7 +143,7 @@ func (m *Adapter[RPC, HEAD]) SubscribeToHeads(ctx context.Context) (<-chan HEAD,
 
 	sub := &ManagedSubscription{
 		Subscription:  &poller,
-		onUnsubscribe: m.removeSub,
+		onUnsubscribe: m.RemoveSub,
 	}
 
 	err := m.RegisterSub(sub, chStopInFlight)
@@ -176,7 +176,7 @@ func (m *Adapter[RPC, HEAD]) SubscribeToFinalizedHeads(ctx context.Context) (<-c
 
 	sub := &ManagedSubscription{
 		Subscription:  &poller,
-		onUnsubscribe: m.removeSub,
+		onUnsubscribe: m.RemoveSub,
 	}
 
 	err := m.RegisterSub(sub, chStopInFlight)

diff --git a/multinode/adaptor_test.go b/multinode/adaptor_test.go
@@ -136,7 +136,7 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) {
 		mockSub := newMockSub()
 		sub := &ManagedSubscription{
 			Subscription:  mockSub,
-			onUnsubscribe: c.removeSub,
+			onUnsubscribe: c.RemoveSub,
 		}
 		err := c.RegisterSub(sub, make(chan struct{}))
 		require.NoError(t, err)
@@ -151,7 +151,7 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) {
 		mockSub := newMockSub()
 		sub := &ManagedSubscription{
 			Subscription:  mockSub,
-			onUnsubscribe: c.removeSub,
+			onUnsubscribe: c.RemoveSub,
 		}
 		err := c.RegisterSub(sub, chStopInFlight)
 		require.Error(t, err)
@@ -164,12 +164,12 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) {
 		mockSub1 := newMockSub()
 		sub1 := &ManagedSubscription{
 			Subscription:  mockSub1,
-			onUnsubscribe: c.removeSub,
+			onUnsubscribe: c.RemoveSub,
 		}
 		mockSub2 := newMockSub()
 		sub2 := &ManagedSubscription{
 			Subscription:  mockSub2,
-			onUnsubscribe: c.removeSub,
+			onUnsubscribe: c.RemoveSub,
 		}
 		err := c.RegisterSub(sub1, chStopInFlight)
 		require.NoError(t, err)