Test SafeSubscription concurrent unsubscribe

Nuffle-Labs · May 31, 2024 · 6f0b117 · 6f0b117
1 parent 7a1bc30
commit 6f0b117
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/core/safeclient/client_test.go b/core/safeclient/client_test.go
@@ -679,3 +679,21 @@ func TestSubscribeFilterLogs_ErrorInSubscription_Resubscribe(t *testing.T) {
 
 	triggerError()
 }
+
+func TestSafeSubscription_ConcurrentUnsubscribe(t *testing.T) {
+	mockCtrl := gomock.NewController(t)
+	sub := mocks.NewMockSubscription(mockCtrl)
+	sub.EXPECT().Unsubscribe().Times(1)
+
+	safeSub := safeclient.NewSafeSubscription(sub)
+
+	var wg sync.WaitGroup
+	for i := 1; i <= 10; i++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			safeSub.Unsubscribe()
+		}()
+	}
+	wg.Wait()
+}