@@ -13,6 +13,7 @@ public class PollingBackgroundServiceTests
1313 private readonly Listener _listener = new ( ) ;
1414 private readonly FakeTimeProvider _timeProvider = new ( ) ;
1515 private readonly CorePollingSettings _settings = new ( ) ;
16+ private readonly CancellationToken _ct = TestContext . Current . CancellationToken ;
1617
1718 [ Fact ]
1819 public async Task WhenServiceStartsTheProducerIsInvoked ( )
@@ -21,7 +22,7 @@ public async Task WhenServiceStartsTheProducerIsInvoked()
2122 var sut = new PollingBackgroundService ( Key , _listener , producerSpy , _timeProvider , _settings , Logger ) ;
2223
2324 await sut . StartAsync ( CancellationToken . None ) ;
24- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run and block
25+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run and block
2526
2627 await producerSpy . Received ( 1 ) . ProducePendingAsync ( Arg . Any < CancellationToken > ( ) ) ;
2728 }
@@ -33,11 +34,11 @@ public async Task UntilPollingIntervalIsReachedTheProducerIsNotInvokedAgain()
3334 var sut = new PollingBackgroundService ( Key , _listener , producerSpy , _timeProvider , _settings , Logger ) ;
3435
3536 await sut . StartAsync ( CancellationToken . None ) ;
36- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run and block
37+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run and block
3738 producerSpy . ClearReceivedCalls ( ) ; // ignore startup call
3839
3940 _timeProvider . Advance ( _settings . PollingInterval - TimeSpan . FromMilliseconds ( 1 ) ) ;
40- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run again
41+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run again
4142
4243 await producerSpy . Received ( 0 ) . ProducePendingAsync ( Arg . Any < CancellationToken > ( ) ) ;
4344 }
@@ -49,11 +50,11 @@ public async Task WhenPollingIntervalIsReachedThenTheProducerIsInvokedAgain()
4950 var sut = new PollingBackgroundService ( Key , _listener , producerSpy , _timeProvider , _settings , Logger ) ;
5051
5152 await sut . StartAsync ( CancellationToken . None ) ;
52- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run and block
53+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run and block
5354 producerSpy . ClearReceivedCalls ( ) ; // ignore startup call
5455
5556 _timeProvider . Advance ( _settings . PollingInterval ) ;
56- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run again
57+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run again
5758
5859 await producerSpy . Received ( 1 ) . ProducePendingAsync ( Arg . Any < CancellationToken > ( ) ) ;
5960 }
@@ -66,11 +67,11 @@ public async Task WhenListenerIsTriggeredThenTheProducerIsInvokedAgain()
6667 var sut = new PollingBackgroundService ( Key , _listener , producerSpy , _timeProvider , _settings , Logger ) ;
6768
6869 await sut . StartAsync ( CancellationToken . None ) ;
69- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run and block
70+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run and block
7071 producerSpy . ClearReceivedCalls ( ) ; // ignore startup call
7172
7273 _listener . OnNewMessages ( ) ;
73- await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) ) ; // give it a bit to run again
74+ await Task . Delay ( TimeSpan . FromMilliseconds ( 100 ) , _ct ) ; // give it a bit to run again
7475
7576 await producerSpy . Received ( 1 ) . ProducePendingAsync ( Arg . Any < CancellationToken > ( ) ) ;
7677 }
0 commit comments