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