@@ -32,35 +32,40 @@ - (void)testSingleSave {
3232 BNCCallbackMap *map = [BNCCallbackMap new ];
3333
3434 // block variable callback will update
35- __block NSString *status = @" no callback" ;
35+ __block BOOL successResult = NO ;
36+ __block NSString *statusMessageResult = @" no callback" ;
3637
3738 // store a request and callback block
3839 BNCServerRequest *request = [BNCServerRequest new ];
39- [map storeRequest: request withCompletion: ^(NSString * _Nonnull statusMessage) {
40- status = statusMessage;
40+ [map storeRequest: request withCompletion: ^(BOOL success, NSString * _Nonnull statusMessage) {
41+ successResult = success;
42+ statusMessageResult = statusMessage;
4143 }];
4244
4345 // confirm there's one entry
4446 XCTAssert ([map containsRequest: request] != NO );
4547 XCTAssert (map.callbacks .count == 1 );
4648
4749 // call callback
48- [map callCompletionForRequest: request withStatusMessage :@" callback" ];
50+ [map callCompletionForRequest: request withSuccessStatus: YES message :@" callback" ];
4951
5052 // check if variable was updated
51- XCTAssert ([@" callback" isEqualToString: status]);
53+ XCTAssertTrue (successResult);
54+ XCTAssert ([@" callback" isEqualToString: statusMessageResult]);
5255}
5356
5457- (void )testDeletedRequest {
5558 BNCCallbackMap *map = [BNCCallbackMap new ];
5659
5760 // block variable callback will update
58- __block NSString *status = @" no callback" ;
61+ __block BOOL successResult = NO ;
62+ __block NSString *statusMessageResult = @" no callback" ;
5963
6064 // store a request and callback block
6165 BNCServerRequest *request = [BNCServerRequest new ];
62- [map storeRequest: request withCompletion: ^(NSString * _Nonnull statusMessage) {
63- status = statusMessage;
66+ [map storeRequest: request withCompletion: ^(BOOL success, NSString * _Nonnull statusMessage) {
67+ successResult = success;
68+ statusMessageResult = statusMessage;
6469 }];
6570
6671 // confirm there's one entry
@@ -70,10 +75,11 @@ - (void)testDeletedRequest {
7075 // confirm a new request results in no callback
7176 request = [BNCServerRequest new ];
7277 XCTAssert ([map containsRequest: request] == NO );
73- [map callCompletionForRequest: request withStatusMessage :@" callback" ];
78+ [map callCompletionForRequest: request withSuccessStatus: YES message :@" callback" ];
7479
7580 // check if variable was updated
76- XCTAssert ([@" no callback" isEqualToString: status]);
81+ XCTAssertFalse (successResult);
82+ XCTAssert ([@" no callback" isEqualToString: statusMessageResult]);
7783}
7884
7985- (void )testSeveralBlocks {
@@ -82,21 +88,21 @@ - (void)testSeveralBlocks {
8288 __block int count = 0 ;
8389
8490 BNCServerRequest *request = [BNCServerRequest new ];
85- [map storeRequest: request withCompletion: ^(NSString * _Nonnull statusMessage) {
91+ [map storeRequest: request withCompletion: ^(BOOL success, NSString * _Nonnull statusMessage) {
8692 count++;
8793 }];
8894
8995 for (int i=0 ; i<100 ; i++) {
9096 BNCServerRequest *tmp = [BNCServerRequest new ];
91- [map storeRequest: tmp withCompletion: ^(NSString * _Nonnull statusMessage) {
97+ [map storeRequest: tmp withCompletion: ^(BOOL success, NSString * _Nonnull statusMessage) {
9298 count++;
9399 }];
94100 }
95101
96102 // confirm there's less than 100 entries. By not retaining the tmp request, they should be getting ARC'd
97103 XCTAssert (map.callbacks .count < 100 );
98104
99- [map callCompletionForRequest: request withStatusMessage :@" callback" ];
105+ [map callCompletionForRequest: request withSuccessStatus: YES message :@" callback" ];
100106 XCTAssert (count == 1 );
101107}
102108
0 commit comments