Use require.Eventually

pulsar/table_view_test.go

@@ -20,6 +20,7 @@ package pulsar
 import (
 	"context"
 	"fmt"
+	"github.com/stretchr/testify/require"
 	"reflect"
 	"testing"
 	"time"
@@ -114,9 +115,9 @@ func TestPublishNilValue(t *testing.T) {
 	})
 	assert.NoError(t, err)
 
-	for tv.Size() < 1 {
-		time.Sleep(time.Second * 1)
-	}
+	require.Eventually(t, func() bool {
+		return tv.Size() == 1
+	}, 5*time.Second, 100*time.Millisecond)
 
 	assert.Equal(t, *(tv.Get("key-1").(*string)), "value-1")
 
@@ -126,19 +127,19 @@ func TestPublishNilValue(t *testing.T) {
 	})
 	assert.NoError(t, err)
 
-	for tv.Size() >= 1 {
-		time.Sleep(time.Second * 1)
-	}
+	require.Eventually(t, func() bool {
+		return tv.Size() == 0
+	}, 5*time.Second, 100*time.Millisecond)
 
 	_, err = producer.Send(context.Background(), &ProducerMessage{
 		Key:   "key-2",
 		Value: "value-2",
 	})
 	assert.NoError(t, err)
 
-	for tv.Size() < 1 {
-		time.Sleep(time.Second * 1)
-	}
+	require.Eventually(t, func() bool {
+		return tv.Size() == 1
+	}, 5*time.Second, 100*time.Millisecond)
 
 	assert.Equal(t, *(tv.Get("key-2").(*string)), "value-2")
 }