@@ -26,7 +26,7 @@ public void Ctor_EnsureDefaults()
26
26
public async Task IsolateAsync_Ok ( )
27
27
{
28
28
// arrange
29
- var cancellationToken = CancellationToken . None ;
29
+ var cancellationToken = TestContext . Current . CancellationToken ;
30
30
bool called = false ;
31
31
_options . OnOpened = args =>
32
32
{
@@ -69,7 +69,7 @@ public async Task IsolateAsync_Ok()
69
69
public async Task BreakAsync_Ok ( )
70
70
{
71
71
// arrange
72
- var cancellationToken = CancellationToken . None ;
72
+ var cancellationToken = TestContext . Current . CancellationToken ;
73
73
bool called = false ;
74
74
_options . OnClosed = args =>
75
75
{
@@ -102,7 +102,7 @@ public async Task BreakAsync_Ok()
102
102
[ Fact ]
103
103
public async Task Disposed_EnsureThrows ( )
104
104
{
105
- var context = ResilienceContextPool . Shared . Get ( ) ;
105
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
106
106
107
107
var controller = CreateController ( ) ;
108
108
controller . Dispose ( ) ;
@@ -121,7 +121,7 @@ public async Task Disposed_EnsureThrows()
121
121
[ Fact ]
122
122
public async Task OnActionPreExecute_CircuitOpenedByValue ( )
123
123
{
124
- var context = ResilienceContextPool . Shared . Get ( ) ;
124
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
125
125
using var controller = CreateController ( ) ;
126
126
127
127
await OpenCircuit ( controller , Outcome . FromResult ( 99 ) ) ;
@@ -138,7 +138,7 @@ public async Task OnActionPreExecute_CircuitOpenedByValue()
138
138
public async Task OnActionPreExecute_CircuitOpened_EnsureExceptionStackTraceDoesNotGrow ( bool innerException )
139
139
{
140
140
var stacks = new List < string > ( ) ;
141
- var context = ResilienceContextPool . Shared . Get ( ) ;
141
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
142
142
using var controller = CreateController ( ) ;
143
143
144
144
await OpenCircuit (
@@ -175,7 +175,7 @@ await OpenCircuit(
175
175
[ Fact ]
176
176
public async Task HalfOpen_EnsureBreakDuration ( )
177
177
{
178
- var context = ResilienceContextPool . Shared . Get ( ) ;
178
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
179
179
using var controller = CreateController ( ) ;
180
180
181
181
await TransitionToState ( controller , CircuitState . HalfOpen , context ) ;
@@ -187,7 +187,7 @@ public async Task HalfOpen_EnsureBreakDuration()
187
187
[ Theory ]
188
188
public async Task HalfOpen_EnsureCorrectStateTransitionAfterExecution ( bool success )
189
189
{
190
- var context = ResilienceContextPool . Shared . Get ( ) ;
190
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
191
191
using var controller = CreateController ( ) ;
192
192
193
193
await TransitionToState ( controller , CircuitState . HalfOpen , context ) ;
@@ -213,7 +213,7 @@ public async Task HalfOpen_EnsureCorrectStateTransitionAfterExecution(bool succe
213
213
[ Fact ]
214
214
public async Task OnActionPreExecute_CircuitOpenedByException ( )
215
215
{
216
- var context = ResilienceContextPool . Shared . Get ( ) ;
216
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
217
217
using var controller = CreateController ( ) ;
218
218
219
219
await OpenCircuit ( controller , Outcome . FromException < int > ( new InvalidOperationException ( ) ) ) ;
@@ -227,7 +227,7 @@ public async Task OnActionPreExecute_CircuitOpenedByException()
227
227
public async Task OnActionFailure_EnsureLock ( )
228
228
{
229
229
// arrange
230
- var cancellationToken = CancellationToken . None ;
230
+ var cancellationToken = TestContext . Current . CancellationToken ;
231
231
var context = ResilienceContextPool . Shared . Get ( cancellationToken ) ;
232
232
233
233
using var executing = new ManualResetEvent ( false ) ;
@@ -261,7 +261,7 @@ public async Task OnActionFailure_EnsureLock()
261
261
public async Task OnActionPreExecute_HalfOpen ( )
262
262
{
263
263
// arrange
264
- var context = ResilienceContextPool . Shared . Get ( ) ;
264
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
265
265
var called = false ;
266
266
_options . OnHalfOpened = _ =>
267
267
{
@@ -293,7 +293,7 @@ public async Task OnActionPreExecute_HalfOpen()
293
293
public async Task OnActionSuccess_EnsureCorrectBehavior ( CircuitState state , CircuitState expectedState )
294
294
{
295
295
// arrange
296
- var context = ResilienceContextPool . Shared . Get ( ) ;
296
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
297
297
var called = false ;
298
298
_options . OnClosed = args =>
299
299
{
@@ -329,7 +329,7 @@ public async Task OnActionSuccess_EnsureCorrectBehavior(CircuitState state, Circ
329
329
public async Task OnActionFailureAsync_EnsureCorrectBehavior ( CircuitState state , CircuitState expectedState , bool shouldBreak )
330
330
{
331
331
// arrange
332
- var context = ResilienceContextPool . Shared . Get ( ) ;
332
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
333
333
var called = false ;
334
334
_options . OnOpened = args =>
335
335
{
@@ -370,7 +370,7 @@ public async Task OnActionFailureAsync_EnsureCorrectBehavior(CircuitState state,
370
370
public async Task OnActionFailureAsync_EnsureBreakDurationGeneration ( )
371
371
{
372
372
// arrange
373
- var context = ResilienceContextPool . Shared . Get ( ) ;
373
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
374
374
_options . BreakDurationGenerator = static args =>
375
375
{
376
376
args . FailureCount . Should ( ) . Be ( 1 ) ;
@@ -402,7 +402,7 @@ public async Task OnActionFailureAsync_EnsureBreakDurationGeneration()
402
402
public async Task BreakDurationGenerator_EnsureHalfOpenAttempts ( )
403
403
{
404
404
// arrange
405
- var context = ResilienceContextPool . Shared . Get ( ) ;
405
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
406
406
var halfOpenAttempts = new List < int > ( ) ;
407
407
408
408
_options . BreakDurationGenerator = args =>
@@ -440,7 +440,7 @@ public async Task BreakDurationGenerator_EnsureHalfOpenAttempts()
440
440
public async Task OnActionFailureAsync_EnsureBreakDurationNotOverflow ( bool overflow )
441
441
{
442
442
// arrange
443
- var context = ResilienceContextPool . Shared . Get ( ) ;
443
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
444
444
using var controller = CreateController ( ) ;
445
445
var shouldBreak = true ;
446
446
await TransitionToState ( controller , CircuitState . HalfOpen , context ) ;
@@ -475,7 +475,7 @@ public async Task OnActionFailureAsync_EnsureBreakDurationNotOverflow(bool overf
475
475
public async Task OnActionFailureAsync_VoidResult_EnsureBreakingExceptionNotSet ( )
476
476
{
477
477
// arrange
478
- var context = ResilienceContextPool . Shared . Get ( ) ;
478
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
479
479
using var controller = CreateController ( ) ;
480
480
bool shouldBreak = true ;
481
481
await TransitionToState ( controller , CircuitState . Open , context ) ;
@@ -497,7 +497,7 @@ public async Task OnActionFailureAsync_VoidResult_EnsureBreakingExceptionNotSet(
497
497
[ Fact ]
498
498
public async Task Flow_Closed_HalfOpen_Closed ( )
499
499
{
500
- var context = ResilienceContextPool . Shared . Get ( ) ;
500
+ var context = ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) ;
501
501
using var controller = CreateController ( ) ;
502
502
503
503
await TransitionToState ( controller , CircuitState . HalfOpen , context ) ;
@@ -512,7 +512,7 @@ public async Task Flow_Closed_HalfOpen_Closed()
512
512
[ Fact ]
513
513
public async Task Flow_Closed_HalfOpen_Open_HalfOpen_Closed ( )
514
514
{
515
- var cancellationToken = CancellationToken . None ;
515
+ var cancellationToken = TestContext . Current . CancellationToken ;
516
516
var context = ResilienceContextPool . Shared . Get ( cancellationToken ) ;
517
517
518
518
using var controller = CreateController ( ) ;
@@ -550,7 +550,7 @@ public async Task Flow_Closed_HalfOpen_Open_HalfOpen_Closed()
550
550
[ Fact ]
551
551
public async Task ExecuteScheduledTask_Async_Ok ( )
552
552
{
553
- var cancellationToken = CancellationToken . None ;
553
+ var cancellationToken = TestContext . Current . CancellationToken ;
554
554
var context = ResilienceContextPool . Shared . Get ( cancellationToken ) ;
555
555
556
556
var source = new TaskCompletionSource < string > ( ) ;
@@ -605,7 +605,7 @@ private async Task OpenCircuit(CircuitStateController<int> controller, Outcome<i
605
605
606
606
await controller . OnHandledOutcomeAsync (
607
607
outcome ?? Outcome . FromResult ( 10 ) ,
608
- ResilienceContextPool . Shared . Get ( ) . Initialize < int > ( true ) ) ;
608
+ ResilienceContextPool . Shared . Get ( TestContext . Current . CancellationToken ) . Initialize < int > ( true ) ) ;
609
609
}
610
610
611
611
private void AdvanceTime ( TimeSpan timespan ) => _timeProvider . Advance ( timespan ) ;
0 commit comments