Update publisher_test.go

configcat · Oct 30, 2024 · fcdf106 · fcdf106
1 parent 6ca7ffa
commit fcdf106
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pubsub/publisher_test.go b/pubsub/publisher_test.go
@@ -12,8 +12,10 @@ func TestPubSub_Sub_Unsub(t *testing.T) {
 
 	sub := make(chan struct{})
 	pubSub.Subscribe(sub)
-	assert.NotEmpty(t, pubSub.subscriptions)
-	assert.NotNil(t, pubSub.subscriptions[sub])
+	utils.WaitUntil(2*time.Second, func() bool {
+		_, ok := pubSub.subscriptions[sub]
+		return ok
+	})
 
 	msg := struct{}{}
 	pubSub.Publish(msg)
@@ -28,5 +30,6 @@ func TestPubSub_Sub_Unsub(t *testing.T) {
 	pubSub.Close()
 	sub2 := make(chan struct{})
 	pubSub.Subscribe(sub2)
-	assert.Empty(t, pubSub.subscriptions)
+	_, ok := pubSub.subscriptions[sub2]
+	assert.False(t, ok)
 }