workaround for flaky test

Author: rekby

internal/topic/topicwriterinternal/writer_reconnector_test.go

@@ -263,47 +263,46 @@ func TestWriterImpl_WriteCodecs(t *testing.T) {
 }
 
 func TestWriterReconnector_Write_QueueLimit(t *testing.T) {
-	ctx := xtest.Context(t)
-	w := newWriterReconnectorStopped(newWriterReconnectorConfig(
-		WithAutoSetSeqNo(false),
-		WithMaxQueueLen(2),
-	))
-	w.firstConnectionHandled.Store(true)
-
-	waitStartQueueWait := func() {
-		xtest.SpinWaitCondition(t, nil, func() bool {
-			if w.semaphore.TryAcquire(1) {
-				w.semaphore.Release(1)
-				return false
-			}
-			return true
-		})
-	}
+	xtest.TestManyTimes(t, func(t testing.TB) {
+		ctx := xtest.Context(t)
+		w := newWriterReconnectorStopped(newWriterReconnectorConfig(
+			WithAutoSetSeqNo(false),
+			WithMaxQueueLen(2),
+		))
+		w.firstConnectionHandled.Store(true)
 
-	err := w.Write(ctx, newTestMessages(1, 2))
-	require.NoError(t, err)
+		waitStartQueueWait := func() {
+			// TODO: fix with reflection and unsafe for count internal waiters
+			time.Sleep(time.Second / 10)
+		}
 
-	ctxNoQueueSpace, ctxNoQueueSpaceCancel := context.WithCancel(ctx)
+		err := w.Write(ctx, newTestMessages(1, 2))
+		require.NoError(t, err)
 
-	go func() {
-		waitStartQueueWait()
-		ctxNoQueueSpaceCancel()
-	}()
-	err = w.Write(ctxNoQueueSpace, newTestMessages(3))
-	require.ErrorIs(t, err, PublicErrQueueIsFull)
+		ctxNoQueueSpace, ctxNoQueueSpaceCancel := context.WithCancel(ctx)
 
-	go func() {
-		waitStartQueueWait()
-		ackErr := w.queue.AcksReceived([]rawtopicwriter.WriteAck{
-			{
-				SeqNo: 1,
-			},
-		})
-		require.NoError(t, ackErr)
-	}()
+		go func() {
+			waitStartQueueWait()
+			ctxNoQueueSpaceCancel()
+		}()
+		err = w.Write(ctxNoQueueSpace, newTestMessages(3))
+		if !errors.Is(err, PublicErrQueueIsFull) {
+			require.ErrorIs(t, err, PublicErrQueueIsFull)
+		}
+
+		go func() {
+			waitStartQueueWait()
+			ackErr := w.queue.AcksReceived([]rawtopicwriter.WriteAck{
+				{
+					SeqNo: 1,
+				},
+			})
+			require.NoError(t, ackErr)
+		}()
 
-	err = w.Write(ctx, newTestMessages(3))
-	require.NoError(t, err)
+		err = w.Write(ctx, newTestMessages(3))
+		require.NoError(t, err)
+	})
 }
 
 func TestEnv(t *testing.T) {