diff --git a/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs b/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs index a2c1f667..2d9ca649 100644 --- a/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs +++ b/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs @@ -130,9 +130,9 @@ public async Task Initialize_WhenFailWriteMessage_ShouldRetryInitializeAndReturn Assert.Equal(PersistenceStatus.Written, (await writer.WriteAsync("abacaba")).Status); // check attempt repeated!!! - _mockStream.Verify(stream => stream.Write(It.IsAny()), + _mockStream.Verify(stream => stream.Write(It.IsAny()), Times.Exactly(3)); + _mockStream.Verify(stream => stream.MoveNextAsync(), Times.AtLeast(2)); // run processing ack may not be able to start on time - _mockStream.Verify(stream => stream.MoveNextAsync(), Times.Exactly(3)); _mockStream.Verify(stream => stream.Current, Times.Exactly(2)); }