@@ -13,44 +13,6 @@ namespace TurboHTTP.Tests.Transport;
1313/// </summary>
1414public sealed class GenerationCounterSpec
1515{
16- [ Fact ( Timeout = 5000 ) ]
17- public void Stale_gen_should_cause_batch_to_be_discarded ( )
18- {
19- var currentGen = 5 ;
20- var batchGen = 3 ;
21-
22- var processed = false ;
23- var discarded = false ;
24-
25- if ( batchGen == currentGen )
26- {
27- processed = true ;
28- }
29- else
30- {
31- discarded = true ;
32- }
33-
34- Assert . False ( processed ) ;
35- Assert . True ( discarded ) ;
36- }
37-
38- [ Fact ( Timeout = 5000 ) ]
39- public void Current_gen_should_cause_batch_to_be_processed ( )
40- {
41- var currentGen = 5 ;
42- var batchGen = 5 ;
43-
44- var processed = false ;
45-
46- if ( batchGen == currentGen )
47- {
48- processed = true ;
49- }
50-
51- Assert . True ( processed ) ;
52- }
53-
5416 [ Fact ( Timeout = 5000 ) ]
5517 public async Task Pump_should_drain_and_exit_when_cts_canceled ( )
5618 {
@@ -95,7 +57,7 @@ public async Task Pump_should_drain_and_exit_when_cts_canceled()
9557 // Actor cancels after first item is processed
9658 cts . Cancel ( ) ;
9759 }
98- } ) ;
60+ } , TestContext . Current . CancellationToken ) ;
9961
10062 await pumpTask ;
10163
@@ -138,7 +100,7 @@ public async Task Pump_should_process_all_items_when_not_canceled()
138100 Interlocked . Increment ( ref processedCount ) ;
139101 }
140102 }
141- } ) ;
103+ } , TestContext . Current . CancellationToken ) ;
142104
143105 await pumpTask ;
144106
@@ -183,7 +145,7 @@ await Task.Run(() =>
183145 Interlocked . Increment ( ref discardedCount ) ;
184146 }
185147 }
186- } ) ;
148+ } , TestContext . Current . CancellationToken ) ;
187149
188150 Assert . True ( processedCount > 0 , "At least some batches should match current gen" ) ;
189151 Assert . True ( discardedCount > 0 , "At least some batches should be stale" ) ;
0 commit comments