@@ -15,34 +15,37 @@ public partial class AdbSocketTests
1515 [ Fact ]
1616 public async Task SendSyncDATARequestAsyncTest ( ) =>
1717 await RunTestAsync (
18- socket => socket . SendSyncRequestAsync ( SyncCommand . DATA , 2 , default ) ,
19- [ ( byte ) 'D' , ( byte ) 'A' , ( byte ) 'T' , ( byte ) 'A' , 2 , 0 , 0 , 0 ] ) ;
18+ ( socket , ctx ) => socket . SendSyncRequestAsync ( SyncCommand . DATA , 2 , ctx ) ,
19+ [ ( byte ) 'D' , ( byte ) 'A' , ( byte ) 'T' , ( byte ) 'A' , 2 , 0 , 0 , 0 ] ,
20+ TestContext . Current . CancellationToken ) ;
2021
2122 /// <summary>
2223 /// Tests the <see cref="AdbSocket.SendSyncRequestAsync(SyncCommand, string, UnixFileStatus, CancellationToken)"/> method.
2324 /// </summary>
2425 [ Fact ]
2526 public async Task SendSyncSENDRequestAsyncTest ( ) =>
2627 await RunTestAsync (
27- socket => socket . SendSyncRequestAsync ( SyncCommand . SEND , "/test" , UnixFileStatus . GroupMask | UnixFileStatus . StickyBit | UnixFileStatus . UserExecute | UnixFileStatus . OtherExecute , default ) ,
28- [ ( byte ) 'S' , ( byte ) 'E' , ( byte ) 'N' , ( byte ) 'D' , 9 , 0 , 0 , 0 , ( byte ) '/' , ( byte ) 't' , ( byte ) 'e' , ( byte ) 's' , ( byte ) 't' , ( byte ) ',' , ( byte ) '6' , ( byte ) '3' , ( byte ) '3' ] ) ;
28+ ( socket , ctx ) => socket . SendSyncRequestAsync ( SyncCommand . SEND , "/test" , UnixFileStatus . GroupMask | UnixFileStatus . StickyBit | UnixFileStatus . UserExecute | UnixFileStatus . OtherExecute , ctx ) ,
29+ [ ( byte ) 'S' , ( byte ) 'E' , ( byte ) 'N' , ( byte ) 'D' , 9 , 0 , 0 , 0 , ( byte ) '/' , ( byte ) 't' , ( byte ) 'e' , ( byte ) 's' , ( byte ) 't' , ( byte ) ',' , ( byte ) '6' , ( byte ) '3' , ( byte ) '3' ] ,
30+ TestContext . Current . CancellationToken ) ;
2931
3032 /// <summary>
3133 /// Tests the <see cref="AdbSocket.SendSyncRequestAsync(SyncCommand, string, CancellationToken)"/> method.
3234 /// </summary>
3335 [ Fact ]
3436 public async Task SendSyncDENTRequestAsyncTest ( ) =>
3537 await RunTestAsync (
36- socket => socket . SendSyncRequestAsync ( SyncCommand . DENT , "/data" , default ) ,
37- [ ( byte ) 'D' , ( byte ) 'E' , ( byte ) 'N' , ( byte ) 'T' , 5 , 0 , 0 , 0 , ( byte ) '/' , ( byte ) 'd' , ( byte ) 'a' , ( byte ) 't' , ( byte ) 'a' ] ) ;
38+ ( socket , ctx ) => socket . SendSyncRequestAsync ( SyncCommand . DENT , "/data" , ctx ) ,
39+ [ ( byte ) 'D' , ( byte ) 'E' , ( byte ) 'N' , ( byte ) 'T' , 5 , 0 , 0 , 0 , ( byte ) '/' , ( byte ) 'd' , ( byte ) 'a' , ( byte ) 't' , ( byte ) 'a' ] ,
40+ TestContext . Current . CancellationToken ) ;
3841
3942 /// <summary>
4043 /// Tests the <see cref="AdbSocket.SendSyncRequestAsync(SyncCommand, string, CancellationToken)"/> method.
4144 /// </summary>
4245 [ Fact ]
4346 public async Task SendSyncNullRequestAsyncTest ( ) =>
4447 _ = await Assert . ThrowsAsync < ArgumentNullException > ( ( ) =>
45- RunTestAsync ( socket => socket . SendSyncRequestAsync ( SyncCommand . DATA , null , default ) , [ ] ) ) ;
48+ RunTestAsync ( ( socket , ctx ) => socket . SendSyncRequestAsync ( SyncCommand . DATA , null , ctx ) , [ ] , TestContext . Current . CancellationToken ) ) ;
4649
4750 /// <summary>
4851 /// Tests the <see cref="AdbSocket.ReadSyncResponseAsync(CancellationToken)"/> method.
@@ -60,7 +63,7 @@ public async Task ReadSyncResponseAsync()
6063
6164 tcpSocket . InputStream . Position = 0 ;
6265
63- Assert . Equal ( SyncCommand . DENT , await socket . ReadSyncResponseAsync ( ) ) ;
66+ Assert . Equal ( SyncCommand . DENT , await socket . ReadSyncResponseAsync ( TestContext . Current . CancellationToken ) ) ;
6467 }
6568
6669 /// <summary>
@@ -81,7 +84,7 @@ public async Task ReadStringAsyncTest()
8184
8285 tcpSocket . InputStream . Position = 0 ;
8386
84- Assert . Equal ( "Hello" , await socket . ReadStringAsync ( ) ) ;
87+ Assert . Equal ( "Hello" , await socket . ReadStringAsync ( TestContext . Current . CancellationToken ) ) ;
8588 }
8689
8790 /// <summary>
@@ -103,7 +106,7 @@ public async Task ReadFailStringAsyncTest()
103106
104107 tcpSocket . InputStream . Position = 0 ;
105108
106- Assert . Equal ( "Hello" , await socket . ReadStringAsync ( ) ) ;
109+ Assert . Equal ( "Hello" , await socket . ReadStringAsync ( TestContext . Current . CancellationToken ) ) ;
107110 }
108111
109112 /// <summary>
@@ -124,7 +127,7 @@ public async Task ReadSyncStringAsyncTest()
124127
125128 tcpSocket . InputStream . Position = 0 ;
126129
127- Assert . Equal ( "Hello" , await socket . ReadSyncStringAsync ( ) ) ;
130+ Assert . Equal ( "Hello" , await socket . ReadSyncStringAsync ( TestContext . Current . CancellationToken ) ) ;
128131 }
129132
130133 /// <summary>
@@ -143,7 +146,7 @@ public async Task ReadAdbOkayResponseAsyncTest()
143146
144147 tcpSocket . InputStream . Position = 0 ;
145148
146- AdbResponse response = await socket . ReadAdbResponseAsync ( ) ;
149+ AdbResponse response = await socket . ReadAdbResponseAsync ( TestContext . Current . CancellationToken ) ;
147150 Assert . True ( response . IOSuccess ) ;
148151 Assert . Equal ( string . Empty , response . Message ) ;
149152 Assert . True ( response . Okay ) ;
@@ -168,7 +171,7 @@ public async Task ReadAdbFailResponseAsyncTest()
168171
169172 tcpSocket . InputStream . Position = 0 ;
170173
171- _ = await Assert . ThrowsAsync < AdbException > ( ( ) => socket . ReadAdbResponseAsync ( ) ) ;
174+ _ = await Assert . ThrowsAsync < AdbException > ( ( ) => socket . ReadAdbResponseAsync ( TestContext . Current . CancellationToken ) ) ;
172175 }
173176
174177 /// <summary>
@@ -187,13 +190,13 @@ public async Task ReadAsyncTest()
187190 data [ i ] = ( byte ) i ;
188191 }
189192
190- await tcpSocket . InputStream . WriteAsync ( data ) ;
193+ await tcpSocket . InputStream . WriteAsync ( data , TestContext . Current . CancellationToken ) ;
191194 tcpSocket . InputStream . Position = 0 ;
192195
193196 // Buffer has a capacity of 101, but we'll only want to read 100 bytes
194197 byte [ ] received = new byte [ 101 ] ;
195198
196- await socket . ReadAsync ( received , 100 ) ;
199+ await socket . ReadAsync ( received , 100 , TestContext . Current . CancellationToken ) ;
197200
198201 for ( int i = 0 ; i < 100 ; i ++ )
199202 {
@@ -219,13 +222,13 @@ public async Task ReadAsyncMemoryTest()
219222 data [ i ] = ( byte ) i ;
220223 }
221224
222- await tcpSocket . InputStream . WriteAsync ( data ) ;
225+ await tcpSocket . InputStream . WriteAsync ( data , TestContext . Current . CancellationToken ) ;
223226 tcpSocket . InputStream . Position = 0 ;
224227
225228 // Buffer has a capacity of 101, but we'll only want to read 100 bytes
226229 byte [ ] received = new byte [ 101 ] ;
227230
228- await socket . ReadAsync ( received . AsMemory ( 0 , 100 ) ) ;
231+ await socket . ReadAsync ( received . AsMemory ( 0 , 100 ) , TestContext . Current . CancellationToken ) ;
229232
230233 for ( int i = 0 ; i < 100 ; i ++ )
231234 {
@@ -241,16 +244,17 @@ public async Task ReadAsyncMemoryTest()
241244 [ Fact ]
242245 public async Task SendAdbRequestAsyncTest ( ) =>
243246 await RunTestAsync (
244- socket => socket . SendAdbRequestAsync ( "Test" , default ) ,
245- "0004Test"u8 . ToArray ( ) ) ;
247+ ( socket , ctx ) => socket . SendAdbRequestAsync ( "Test" , ctx ) ,
248+ "0004Test"u8 . ToArray ( ) ,
249+ TestContext . Current . CancellationToken ) ;
246250
247- private static async Task RunTestAsync ( Func < IAdbSocket , Task > test , byte [ ] expectedDataSent )
251+ private static async Task RunTestAsync ( Func < IAdbSocket , CancellationToken , Task > test , byte [ ] expectedDataSent , CancellationToken cancellationToken = default )
248252 {
249253 using DummyTcpSocket tcpSocket = new ( ) ;
250254 using AdbSocket socket = new ( tcpSocket ) ;
251255
252256 // Run the test.
253- await test ( socket ) ;
257+ await test ( socket , cancellationToken ) ;
254258
255259 // Validate the data that was sent over the wire.
256260 Assert . Equal ( expectedDataSent , tcpSocket . GetBytesSent ( ) ) ;
0 commit comments