diff --git a/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs b/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs index 1b402702..57297a62 100644 --- a/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs +++ b/src/Ydb.Sdk/tests/Topic/WriterUnitTests.cs @@ -485,7 +485,7 @@ public async Task WriteAsync_WhenTransportExceptionOnProcessingWriteAck_ShouldRe IBidirectionalStream.MoveNextAsync() IBidirectionalStream.Write({ "writeRequest": { "messages": [ { "seqNo": "1", "createdAt": "2024-12-03T14:12:59.548210Z", "data": "ZAAAAAAAAAA=", "uncompressedSize": "8" } ], "codec": 1 } }) IBidirectionalStream.Current - IBidirectionalStream.MoveNextAsync() + IBidirectionalStream.MoveNextAsync() [Maybe] */ [Fact] public async Task WriteAsync_WhenStreamIsClosingOnProcessingWriteAck_ShouldReconnectThenReturnWriteResult() @@ -543,7 +543,7 @@ public async Task WriteAsync_WhenStreamIsClosingOnProcessingWriteAck_ShouldRecon Assert.Equal(PersistenceStatus.Written, (await writer.WriteAsync(100L)).Status); _mockStream.Verify(stream => stream.Write(It.IsAny()), Times.Exactly(3)); - _mockStream.Verify(stream => stream.MoveNextAsync(), Times.Exactly(5)); + _mockStream.Verify(stream => stream.MoveNextAsync(), Times.AtLeast(4)); _mockStream.Verify(stream => stream.Current, Times.Exactly(3)); }