@@ -435,16 +435,18 @@ final class CallbackConformance: XCTestCase {
435
435
proto. domain = " connect-crosstest "
436
436
}
437
437
let expectation = self . expectation ( description: " Stream completes " )
438
- var responseCount = 0
438
+ let responseCount = Locked ( 0 )
439
439
let sizes = [ 31_415 , 9 , 2_653 , 58_979 ]
440
440
let stream = client. failStreamingOutputCall { result in
441
441
switch result {
442
442
case . headers:
443
443
break
444
444
445
445
case . message( let output) :
446
- XCTAssertEqual ( output. payload. body. count, sizes [ responseCount] )
447
- responseCount += 1
446
+ responseCount. perform { responseCount in
447
+ XCTAssertEqual ( output. payload. body. count, sizes [ responseCount] )
448
+ responseCount += 1
449
+ }
448
450
449
451
case . complete( _, let error, _) :
450
452
guard let connectError = error as? ConnectError else {
@@ -468,7 +470,7 @@ final class CallbackConformance: XCTestCase {
468
470
} )
469
471
470
472
XCTAssertEqual ( XCTWaiter ( ) . wait ( for: [ expectation] , timeout: kTimeout) , . completed)
471
- XCTAssertEqual ( responseCount, 4 )
473
+ XCTAssertEqual ( responseCount. value , 4 )
472
474
}
473
475
}
474
476
@@ -480,9 +482,9 @@ final class CallbackConformance: XCTestCase {
480
482
let cancelable = client. emptyCall (
481
483
request: SwiftProtobuf . Google_Protobuf_Empty ( )
482
484
) { response in
483
- XCTAssertEqual ( response. code, . canceled)
484
- XCTAssertEqual ( response. error? . code, . canceled)
485
- expectation. fulfill ( )
485
+ XCTAssertEqual ( response. code, . canceled)
486
+ XCTAssertEqual ( response. error? . code, . canceled)
487
+ expectation. fulfill ( )
486
488
}
487
489
cancelable. cancel ( )
488
490
XCTAssertEqual ( XCTWaiter ( ) . wait ( for: [ expectation] , timeout: kTimeout) , . completed)
0 commit comments