Skip to content

Commit b7db482

Browse files
dev: fix flap tests in PoolingSessionTests.cs
1 parent b238264 commit b7db482

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Session/PoolingSessionTests.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,7 @@ public async Task Open_WhenAttachStreamFirstMoveNextAsyncThrowException_IsBroken
101101
var ydbException = await Assert.ThrowsAsync<YdbException>(() => session.Open(CancellationToken.None));
102102
Assert.Equal("Transport RPC call error", ydbException.Message);
103103
Assert.Equal(StatusCode.ClientTransportTimeout, ydbException.Code);
104+
await Task.Delay(500);
104105
Assert.True(session.IsBroken);
105106
}
106107

@@ -113,6 +114,7 @@ public async Task Open_WhenAttachStreamFirstMoveNextIsFalse_IsBroken()
113114
var ydbException = await Assert.ThrowsAsync<YdbException>(() => session.Open(CancellationToken.None));
114115
Assert.Equal("Attach stream is not started!", ydbException.Message);
115116
Assert.Equal(StatusCode.Cancelled, ydbException.Code);
117+
await Task.Delay(500);
116118
Assert.True(session.IsBroken);
117119
}
118120

@@ -133,6 +135,7 @@ public async Task Open_WhenAttachStreamFirstCurrentIsBadSession_IsBroken()
133135
var ydbException = await Assert.ThrowsAsync<YdbException>(() => session.Open(CancellationToken.None));
134136
Assert.Equal("Status: BadSession, Issues:\n[1] Error: Ouch BadSession!", ydbException.Message);
135137
Assert.Equal(StatusCode.BadSession, ydbException.Code);
138+
await Task.Delay(500);
136139
Assert.True(session.IsBroken);
137140
}
138141

@@ -172,8 +175,6 @@ public async Task Open_WhenSuccessOpenThenAttachStreamSendBadSession_IsNotBroken
172175
await session.Open(CancellationToken.None);
173176
Assert.False(session.IsBroken);
174177
tcsSecondMoveAttachStream.SetResult(true); // attach stream is closed
175-
await Task.Delay(500);
176-
Assert.True(session.IsBroken);
177178
}
178179

179180
[Fact]

0 commit comments

Comments
 (0)