Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2015 gRPC authors. |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Jorge Canizales | 9195233 | 2015-07-31 17:50:49 -0700 | [diff] [blame] | 19 | #import "InteropTests.h" |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 20 | |
Jorge Canizales | 9195233 | 2015-07-31 17:50:49 -0700 | [diff] [blame] | 21 | #include <grpc/status.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 22 | |
Makarand Dharmapurikar | de733bd | 2016-05-25 10:38:56 -0700 | [diff] [blame] | 23 | #import <Cronet/Cronet.h> |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 24 | #import <GRPCClient/GRPCCall+ChannelArg.h> |
Muxi Yan | 8e6aec5 | 2017-02-10 09:40:18 -0800 | [diff] [blame] | 25 | #import <GRPCClient/GRPCCall+Tests.h> |
| 26 | #import <GRPCClient/internal_testing/GRPCCall+InternalTests.h> |
Makarand Dharmapurikar | 9b5f640 | 2016-05-20 15:57:34 -0700 | [diff] [blame] | 27 | #import <GRPCClient/GRPCCall+Cronet.h> |
Jorge Canizales | c42a38e | 2015-06-21 14:44:25 -0700 | [diff] [blame] | 28 | #import <ProtoRPC/ProtoRPC.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 29 | #import <RemoteTest/Messages.pbobjc.h> |
| 30 | #import <RemoteTest/Test.pbobjc.h> |
| 31 | #import <RemoteTest/Test.pbrpc.h> |
Jorge Canizales | c42a38e | 2015-06-21 14:44:25 -0700 | [diff] [blame] | 32 | #import <RxLibrary/GRXBufferedPipe.h> |
| 33 | #import <RxLibrary/GRXWriter+Immediate.h> |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 34 | #import <grpc/support/log.h> |
| 35 | #import <grpc/grpc.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 36 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 37 | #define TEST_TIMEOUT 32 |
| 38 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 39 | // Convenience constructors for the generated proto messages: |
| 40 | |
| 41 | @interface RMTStreamingOutputCallRequest (Constructors) |
| 42 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize |
| 43 | requestedResponseSize:(NSNumber *)responseSize; |
| 44 | @end |
| 45 | |
| 46 | @implementation RMTStreamingOutputCallRequest (Constructors) |
| 47 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize |
| 48 | requestedResponseSize:(NSNumber *)responseSize { |
| 49 | RMTStreamingOutputCallRequest *request = [self message]; |
| 50 | RMTResponseParameters *parameters = [RMTResponseParameters message]; |
Yuchen Zeng | afb62f8 | 2016-06-24 19:11:08 -0700 | [diff] [blame] | 51 | parameters.size = responseSize.intValue; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 52 | [request.responseParametersArray addObject:parameters]; |
| 53 | request.payload.body = [NSMutableData dataWithLength:payloadSize.unsignedIntegerValue]; |
| 54 | return request; |
| 55 | } |
| 56 | @end |
| 57 | |
| 58 | @interface RMTStreamingOutputCallResponse (Constructors) |
| 59 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize; |
| 60 | @end |
| 61 | |
| 62 | @implementation RMTStreamingOutputCallResponse (Constructors) |
| 63 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize { |
| 64 | RMTStreamingOutputCallResponse * response = [self message]; |
| 65 | response.payload.type = RMTPayloadType_Compressable; |
| 66 | response.payload.body = [NSMutableData dataWithLength:payloadSize.unsignedIntegerValue]; |
| 67 | return response; |
| 68 | } |
| 69 | @end |
| 70 | |
Jorge Canizales | b2bd067 | 2015-08-01 23:19:11 -0700 | [diff] [blame] | 71 | #pragma mark Tests |
| 72 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 73 | @implementation InteropTests { |
| 74 | RMTTestService *_service; |
| 75 | } |
| 76 | |
Jorge Canizales | 945f72c | 2015-07-31 17:40:59 -0700 | [diff] [blame] | 77 | + (NSString *)host { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 78 | return nil; |
Jorge Canizales | 945f72c | 2015-07-31 17:40:59 -0700 | [diff] [blame] | 79 | } |
Jorge Canizales | ce8ee61 | 2015-07-24 20:52:49 -0700 | [diff] [blame] | 80 | |
Muxi Yan | 88a41bf | 2017-03-08 14:54:11 -0800 | [diff] [blame] | 81 | // This number indicates how many bytes of overhead does Protocol Buffers encoding add onto the |
| 82 | // message. The number varies as different message.proto is used on different servers. The actual |
| 83 | // number for each interop server is overridden in corresponding derived test classes. |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 84 | - (int32_t)encodingOverhead { |
| 85 | return 0; |
| 86 | } |
| 87 | |
Muxi Yan | da2f0cf | 2017-04-09 15:34:28 -0700 | [diff] [blame] | 88 | + (void)setUp { |
Muxi Yan | 4b947d3 | 2017-11-30 09:36:29 -0800 | [diff] [blame^] | 89 | NSLog(@"InteropTest Started, class: %@", [[self class] description]); |
Muxi Yan | da2f0cf | 2017-04-09 15:34:28 -0700 | [diff] [blame] | 90 | #ifdef GRPC_COMPILE_WITH_CRONET |
| 91 | // Cronet setup |
| 92 | [Cronet setHttp2Enabled:YES]; |
| 93 | [Cronet start]; |
| 94 | [GRPCCall useCronetWithEngine:[Cronet getGlobalEngine]]; |
| 95 | #endif |
| 96 | } |
| 97 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 98 | - (void)setUp { |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 99 | self.continueAfterFailure = NO; |
| 100 | |
Jorge Canizales | 32fde7a | 2016-07-29 17:31:07 -0700 | [diff] [blame] | 101 | [GRPCCall resetHostSettings]; |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 102 | |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 103 | _service = self.class.host ? [RMTTestService serviceWithHost:self.class.host] : nil; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 104 | } |
| 105 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 106 | - (void)testEmptyUnaryRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 107 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 108 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"EmptyUnary"]; |
| 109 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 110 | GPBEmpty *request = [GPBEmpty message]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 111 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 112 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 113 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 114 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 115 | id expectedResponse = [GPBEmpty message]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 116 | XCTAssertEqualObjects(response, expectedResponse); |
| 117 | |
| 118 | [expectation fulfill]; |
| 119 | }]; |
| 120 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 121 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 122 | } |
| 123 | |
| 124 | - (void)testLargeUnaryRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 125 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 126 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"LargeUnary"]; |
| 127 | |
| 128 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
| 129 | request.responseType = RMTPayloadType_Compressable; |
| 130 | request.responseSize = 314159; |
| 131 | request.payload.body = [NSMutableData dataWithLength:271828]; |
| 132 | |
| 133 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 134 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 135 | |
| 136 | RMTSimpleResponse *expectedResponse = [RMTSimpleResponse message]; |
| 137 | expectedResponse.payload.type = RMTPayloadType_Compressable; |
| 138 | expectedResponse.payload.body = [NSMutableData dataWithLength:314159]; |
| 139 | XCTAssertEqualObjects(response, expectedResponse); |
| 140 | |
| 141 | [expectation fulfill]; |
| 142 | }]; |
| 143 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 144 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 145 | } |
| 146 | |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 147 | - (void)testPacketCoalescing { |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 148 | XCTAssertNotNil(self.class.host); |
| 149 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"LargeUnary"]; |
| 150 | |
| 151 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
| 152 | request.responseType = RMTPayloadType_Compressable; |
| 153 | request.responseSize = 10; |
| 154 | request.payload.body = [NSMutableData dataWithLength:10]; |
| 155 | |
Muxi Yan | 40d7c62 | 2017-02-08 14:41:45 -0800 | [diff] [blame] | 156 | [GRPCCall enableOpBatchLog:YES]; |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 157 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 158 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 159 | |
| 160 | RMTSimpleResponse *expectedResponse = [RMTSimpleResponse message]; |
| 161 | expectedResponse.payload.type = RMTPayloadType_Compressable; |
| 162 | expectedResponse.payload.body = [NSMutableData dataWithLength:10]; |
| 163 | XCTAssertEqualObjects(response, expectedResponse); |
| 164 | |
Muxi Yan | 1fb35b5 | 2017-03-07 17:42:25 -0800 | [diff] [blame] | 165 | // The test is a success if there is a batch of exactly 3 ops (SEND_INITIAL_METADATA, |
| 166 | // SEND_MESSAGE, SEND_CLOSE_FROM_CLIENT). Without packet coalescing each batch of ops contains |
| 167 | // only one op. |
Muxi Yan | 40d7c62 | 2017-02-08 14:41:45 -0800 | [diff] [blame] | 168 | NSArray *opBatches = [GRPCCall obtainAndCleanOpBatchLog]; |
Muxi Yan | 1fb35b5 | 2017-03-07 17:42:25 -0800 | [diff] [blame] | 169 | const NSInteger kExpectedOpBatchSize = 3; |
Muxi Yan | 40d7c62 | 2017-02-08 14:41:45 -0800 | [diff] [blame] | 170 | for (NSObject *o in opBatches) { |
| 171 | if ([o isKindOfClass:[NSArray class]]) { |
| 172 | NSArray *batch = (NSArray *)o; |
Muxi Yan | 1fb35b5 | 2017-03-07 17:42:25 -0800 | [diff] [blame] | 173 | if ([batch count] == kExpectedOpBatchSize) { |
Muxi Yan | 40d7c62 | 2017-02-08 14:41:45 -0800 | [diff] [blame] | 174 | [expectation fulfill]; |
| 175 | break; |
| 176 | } |
| 177 | } |
| 178 | } |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 179 | }]; |
| 180 | |
| 181 | [self waitForExpectationsWithTimeout:16 handler:nil]; |
Muxi Yan | 40d7c62 | 2017-02-08 14:41:45 -0800 | [diff] [blame] | 182 | [GRPCCall enableOpBatchLog:NO]; |
Muxi Yan | a40ccd8 | 2016-11-05 21:39:44 -0700 | [diff] [blame] | 183 | } |
| 184 | |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 185 | - (void)test4MBResponsesAreAccepted { |
| 186 | XCTAssertNotNil(self.class.host); |
| 187 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"MaxResponseSize"]; |
| 188 | |
| 189 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 190 | const int32_t kPayloadSize = 4 * 1024 * 1024 - self.encodingOverhead; // 4MB - encoding overhead |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 191 | request.responseSize = kPayloadSize; |
| 192 | |
| 193 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 194 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 195 | XCTAssertEqual(response.payload.body.length, kPayloadSize); |
| 196 | [expectation fulfill]; |
| 197 | }]; |
| 198 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 199 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 200 | } |
| 201 | |
| 202 | - (void)testResponsesOverMaxSizeFailWithActionableMessage { |
| 203 | XCTAssertNotNil(self.class.host); |
| 204 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ResponseOverMaxSize"]; |
| 205 | |
| 206 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 207 | const int32_t kPayloadSize = 4 * 1024 * 1024 - self.encodingOverhead + 1; // 1B over max size |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 208 | request.responseSize = kPayloadSize; |
| 209 | |
| 210 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
Jorge Canizales | ec0743e | 2016-07-29 01:50:06 -0700 | [diff] [blame] | 211 | // TODO(jcanizales): Catch the error and rethrow it with an actionable message: |
| 212 | // - Use +[GRPCCall setResponseSizeLimit:forHost:] to set a higher limit. |
| 213 | // - If you're developing the server, consider using response streaming, or let clients filter |
| 214 | // responses by setting a google.protobuf.FieldMask in the request: |
| 215 | // https://github.com/google/protobuf/blob/master/src/google/protobuf/field_mask.proto |
Mark D. Roth | a4d9ee2 | 2016-08-31 13:17:26 -0700 | [diff] [blame] | 216 | XCTAssertEqualObjects(error.localizedDescription, @"Received message larger than max (4194305 vs. 4194304)"); |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 217 | [expectation fulfill]; |
| 218 | }]; |
| 219 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 220 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 221 | } |
| 222 | |
| 223 | - (void)testResponsesOver4MBAreAcceptedIfOptedIn { |
| 224 | XCTAssertNotNil(self.class.host); |
| 225 | __weak XCTestExpectation *expectation = |
| 226 | [self expectationWithDescription:@"HigherResponseSizeLimit"]; |
| 227 | |
| 228 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
| 229 | const size_t kPayloadSize = 5 * 1024 * 1024; // 5MB |
| 230 | request.responseSize = kPayloadSize; |
| 231 | |
| 232 | [GRPCCall setResponseSizeLimit:6 * 1024 * 1024 forHost:self.class.host]; |
| 233 | |
| 234 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 235 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 236 | XCTAssertEqual(response.payload.body.length, kPayloadSize); |
| 237 | [expectation fulfill]; |
| 238 | }]; |
| 239 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 240 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 241 | } |
| 242 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 243 | - (void)testClientStreamingRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 244 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 245 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ClientStreaming"]; |
| 246 | |
| 247 | RMTStreamingInputCallRequest *request1 = [RMTStreamingInputCallRequest message]; |
| 248 | request1.payload.body = [NSMutableData dataWithLength:27182]; |
| 249 | |
| 250 | RMTStreamingInputCallRequest *request2 = [RMTStreamingInputCallRequest message]; |
| 251 | request2.payload.body = [NSMutableData dataWithLength:8]; |
| 252 | |
| 253 | RMTStreamingInputCallRequest *request3 = [RMTStreamingInputCallRequest message]; |
| 254 | request3.payload.body = [NSMutableData dataWithLength:1828]; |
| 255 | |
| 256 | RMTStreamingInputCallRequest *request4 = [RMTStreamingInputCallRequest message]; |
| 257 | request4.payload.body = [NSMutableData dataWithLength:45904]; |
| 258 | |
Jorge Canizales | a8c5d96 | 2015-07-16 18:55:31 -0700 | [diff] [blame] | 259 | GRXWriter *writer = [GRXWriter writerWithContainer:@[request1, request2, request3, request4]]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 260 | |
| 261 | [_service streamingInputCallWithRequestsWriter:writer |
| 262 | handler:^(RMTStreamingInputCallResponse *response, |
| 263 | NSError *error) { |
| 264 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 265 | |
| 266 | RMTStreamingInputCallResponse *expectedResponse = [RMTStreamingInputCallResponse message]; |
| 267 | expectedResponse.aggregatedPayloadSize = 74922; |
| 268 | XCTAssertEqualObjects(response, expectedResponse); |
| 269 | |
| 270 | [expectation fulfill]; |
| 271 | }]; |
| 272 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 273 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 274 | } |
| 275 | |
| 276 | - (void)testServerStreamingRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 277 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 278 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ServerStreaming"]; |
| 279 | |
| 280 | NSArray *expectedSizes = @[@31415, @9, @2653, @58979]; |
| 281 | |
| 282 | RMTStreamingOutputCallRequest *request = [RMTStreamingOutputCallRequest message]; |
| 283 | for (NSNumber *size in expectedSizes) { |
| 284 | RMTResponseParameters *parameters = [RMTResponseParameters message]; |
Yuchen Zeng | afb62f8 | 2016-06-24 19:11:08 -0700 | [diff] [blame] | 285 | parameters.size = [size intValue]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 286 | [request.responseParametersArray addObject:parameters]; |
| 287 | } |
| 288 | |
| 289 | __block int index = 0; |
| 290 | [_service streamingOutputCallWithRequest:request |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 291 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 292 | RMTStreamingOutputCallResponse *response, |
| 293 | NSError *error){ |
| 294 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 295 | XCTAssertTrue(done || response, @"Event handler called without an event."); |
| 296 | |
| 297 | if (response) { |
| 298 | XCTAssertLessThan(index, 4, @"More than 4 responses received."); |
| 299 | id expected = [RMTStreamingOutputCallResponse messageWithPayloadSize:expectedSizes[index]]; |
| 300 | XCTAssertEqualObjects(response, expected); |
| 301 | index += 1; |
| 302 | } |
| 303 | |
| 304 | if (done) { |
| 305 | XCTAssertEqual(index, 4, @"Received %i responses instead of 4.", index); |
| 306 | [expectation fulfill]; |
| 307 | } |
| 308 | }]; |
| 309 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 310 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 311 | } |
| 312 | |
| 313 | - (void)testPingPongRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 314 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 315 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"PingPong"]; |
| 316 | |
| 317 | NSArray *requests = @[@27182, @8, @1828, @45904]; |
| 318 | NSArray *responses = @[@31415, @9, @2653, @58979]; |
| 319 | |
| 320 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
| 321 | |
| 322 | __block int index = 0; |
| 323 | |
| 324 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:requests[index] |
| 325 | requestedResponseSize:responses[index]]; |
| 326 | [requestsBuffer writeValue:request]; |
| 327 | |
| 328 | [_service fullDuplexCallWithRequestsWriter:requestsBuffer |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 329 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 330 | RMTStreamingOutputCallResponse *response, |
| 331 | NSError *error) { |
| 332 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 333 | XCTAssertTrue(done || response, @"Event handler called without an event."); |
| 334 | |
| 335 | if (response) { |
| 336 | XCTAssertLessThan(index, 4, @"More than 4 responses received."); |
| 337 | id expected = [RMTStreamingOutputCallResponse messageWithPayloadSize:responses[index]]; |
| 338 | XCTAssertEqualObjects(response, expected); |
| 339 | index += 1; |
| 340 | if (index < 4) { |
| 341 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:requests[index] |
| 342 | requestedResponseSize:responses[index]]; |
| 343 | [requestsBuffer writeValue:request]; |
| 344 | } else { |
| 345 | [requestsBuffer writesFinishedWithError:nil]; |
| 346 | } |
| 347 | } |
| 348 | |
| 349 | if (done) { |
| 350 | XCTAssertEqual(index, 4, @"Received %i responses instead of 4.", index); |
| 351 | [expectation fulfill]; |
| 352 | } |
| 353 | }]; |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 354 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 355 | } |
| 356 | |
| 357 | - (void)testEmptyStreamRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 358 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 359 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"EmptyStream"]; |
| 360 | [_service fullDuplexCallWithRequestsWriter:[GRXWriter emptyWriter] |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 361 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 362 | RMTStreamingOutputCallResponse *response, |
| 363 | NSError *error) { |
| 364 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 365 | XCTAssert(done, @"Unexpected response: %@", response); |
| 366 | [expectation fulfill]; |
| 367 | }]; |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 368 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 369 | } |
| 370 | |
| 371 | - (void)testCancelAfterBeginRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 372 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 373 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"CancelAfterBegin"]; |
| 374 | |
| 375 | // A buffered pipe to which we never write any value acts as a writer that just hangs. |
| 376 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
| 377 | |
Yuchen Zeng | 3d399cb | 2016-06-21 11:57:46 -0700 | [diff] [blame] | 378 | GRPCProtoCall *call = |
| 379 | [_service RPCToStreamingInputCallWithRequestsWriter:requestsBuffer |
| 380 | handler:^(RMTStreamingInputCallResponse *response, |
| 381 | NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 382 | XCTAssertEqual(error.code, GRPC_STATUS_CANCELLED); |
| 383 | [expectation fulfill]; |
| 384 | }]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 385 | XCTAssertEqual(call.state, GRXWriterStateNotStarted); |
| 386 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 387 | [call start]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 388 | XCTAssertEqual(call.state, GRXWriterStateStarted); |
| 389 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 390 | [call cancel]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 391 | XCTAssertEqual(call.state, GRXWriterStateFinished); |
| 392 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 393 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 394 | } |
| 395 | |
| 396 | - (void)testCancelAfterFirstResponseRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 397 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 398 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"CancelAfterFirstResponse"]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 399 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 400 | // A buffered pipe to which we write a single value but never close |
| 401 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 402 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 403 | __block BOOL receivedResponse = NO; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 404 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 405 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:@21782 |
| 406 | requestedResponseSize:@31415]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 407 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 408 | [requestsBuffer writeValue:request]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 409 | |
Yuchen Zeng | dbe2b9e | 2016-06-15 20:23:04 -0700 | [diff] [blame] | 410 | __block GRPCProtoCall *call = |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 411 | [_service RPCToFullDuplexCallWithRequestsWriter:requestsBuffer |
| 412 | eventHandler:^(BOOL done, |
| 413 | RMTStreamingOutputCallResponse *response, |
| 414 | NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 415 | if (receivedResponse) { |
| 416 | XCTAssert(done, @"Unexpected extra response %@", response); |
| 417 | XCTAssertEqual(error.code, GRPC_STATUS_CANCELLED); |
| 418 | [expectation fulfill]; |
| 419 | } else { |
| 420 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 421 | XCTAssertFalse(done, @"Finished without response"); |
| 422 | XCTAssertNotNil(response); |
| 423 | receivedResponse = YES; |
| 424 | [call cancel]; |
| 425 | } |
| 426 | }]; |
| 427 | [call start]; |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 428 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 429 | } |
| 430 | |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 431 | - (void)testRPCAfterClosingOpenConnections { |
| 432 | XCTAssertNotNil(self.class.host); |
| 433 | __weak XCTestExpectation *expectation = |
| 434 | [self expectationWithDescription:@"RPC after closing connection"]; |
| 435 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 436 | GPBEmpty *request = [GPBEmpty message]; |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 437 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 438 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 439 | XCTAssertNil(error, @"First RPC finished with unexpected error: %@", error); |
| 440 | |
Jorge Canizales | 00ca8ea | 2016-06-24 16:02:56 -0700 | [diff] [blame] | 441 | #pragma clang diagnostic push |
| 442 | #pragma clang diagnostic ignored "-Wdeprecated-declarations" |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 443 | [GRPCCall closeOpenConnections]; |
Jorge Canizales | 00ca8ea | 2016-06-24 16:02:56 -0700 | [diff] [blame] | 444 | #pragma clang diagnostic pop |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 445 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 446 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 447 | XCTAssertNil(error, @"Second RPC finished with unexpected error: %@", error); |
| 448 | [expectation fulfill]; |
| 449 | }]; |
| 450 | }]; |
| 451 | |
Muxi Yan | b1341a3 | 2016-12-05 16:38:02 -0800 | [diff] [blame] | 452 | [self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 453 | } |
| 454 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 455 | @end |