mirror of https://github.com/grpc/grpc.git
Muxi Yan
5 years ago
parent
1bfdbc1f6c
commit
35a1f0a963
6 changed files with 264 additions and 1 deletions
@ -0,0 +1,232 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import <GRPCClient/GRPCCall.h> |
||||
#import <GRPCClient/GRPCInterceptor.h> |
||||
#import <GRPCClient/GRPCTransport.h> |
||||
#import <XCTest/XCTest.h> |
||||
|
||||
#define TEST_TIMEOUT (8.0) |
||||
|
||||
static NSString *const kRemoteHost = @"grpc-test.sandbox.googleapis.com:443"; |
||||
|
||||
static const GRPCTransportID kFakeTransportID = "io.grpc.transport.unittest.fake"; |
||||
|
||||
@class GRPCFakeTransportFactory; |
||||
dispatch_once_t initFakeTransportFactory; |
||||
static GRPCFakeTransportFactory *fakeTransportFactory; |
||||
|
||||
@interface GRPCFakeTransportFactory : NSObject<GRPCTransportFactory> |
||||
|
||||
@property(atomic) GRPCTransport *nextTransportInstance; |
||||
- (void)setTransportInterceptorFactories:(NSArray<id<GRPCInterceptorFactory>> *)factories; |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCFakeTransportFactory { |
||||
NSArray<id<GRPCInterceptorFactory>> *_interceptorFactories; |
||||
} |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&initFakeTransportFactory, ^{ |
||||
fakeTransportFactory = [[GRPCFakeTransportFactory alloc] init]; |
||||
}); |
||||
return fakeTransportFactory; |
||||
} |
||||
|
||||
+ (void)load { |
||||
[[GRPCTransportRegistry sharedInstance] registerTransportWithID:kFakeTransportID |
||||
factory:[self sharedInstance]]; |
||||
} |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager { |
||||
return _nextTransportInstance; |
||||
} |
||||
|
||||
- (void)setTransportInterceptorFactories:(NSArray<id<GRPCInterceptorFactory>> *)factories { |
||||
_interceptorFactories = [NSArray arrayWithArray:factories]; |
||||
} |
||||
|
||||
- (NSArray<id<GRPCInterceptorFactory>> *)transportInterceptorFactories { |
||||
return _interceptorFactories; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@interface DummyInterceptor : GRPCInterceptor |
||||
|
||||
@property(atomic) BOOL hit; |
||||
|
||||
@end |
||||
|
||||
@implementation DummyInterceptor { |
||||
GRPCInterceptorManager *_manager; |
||||
} |
||||
|
||||
- (instancetype)initWithInterceptorManager:(GRPCInterceptorManager *)interceptorManager |
||||
dispatchQueue:(dispatch_queue_t)dispatchQueue { |
||||
if (dispatchQueue == nil) { |
||||
dispatchQueue = dispatch_queue_create(NULL, DISPATCH_QUEUE_SERIAL); |
||||
} |
||||
if ((self = [super initWithInterceptorManager:interceptorManager dispatchQueue:dispatchQueue])) { |
||||
_manager = interceptorManager; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)startWithRequestOptions:(GRPCRequestOptions *)requestOptions |
||||
callOptions:(GRPCCallOptions *)callOptions { |
||||
self.hit = YES; |
||||
[_manager |
||||
forwardPreviousInterceptorCloseWithTrailingMetadata:nil |
||||
error:[NSError |
||||
errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeCancelled |
||||
userInfo:@{ |
||||
NSLocalizedDescriptionKey : |
||||
@"Canceled." |
||||
}]]; |
||||
[_manager shutDown]; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@interface DummyInterceptorFactory : NSObject<GRPCInterceptorFactory> |
||||
|
||||
+ (instancetype)sharedInstance; |
||||
|
||||
@end |
||||
|
||||
static DummyInterceptorFactory *dummyInterceptorFactory; |
||||
static dispatch_once_t initDummyInterceptorFactory; |
||||
|
||||
@implementation DummyInterceptorFactory |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&initDummyInterceptorFactory, ^{ |
||||
dummyInterceptorFactory = [[DummyInterceptorFactory alloc] init]; |
||||
}); |
||||
return dummyInterceptorFactory; |
||||
} |
||||
|
||||
- (GRPCInterceptor *)createInterceptorWithManager:(GRPCInterceptorManager *)interceptorManager { |
||||
return [[DummyInterceptor alloc] |
||||
initWithInterceptorManager:interceptorManager |
||||
dispatchQueue:dispatch_queue_create(NULL, DISPATCH_QUEUE_SERIAL)]; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@interface TestsBlockCallbacks : NSObject<GRPCResponseHandler> |
||||
|
||||
- (instancetype)initWithInitialMetadataCallback:(void (^)(NSDictionary *))initialMetadataCallback |
||||
dataCallback:(void (^)(id))dataCallback |
||||
closeCallback:(void (^)(NSDictionary *, NSError *))closeCallback |
||||
writeMessageCallback:(void (^)(void))writeMessageCallback; |
||||
|
||||
@end |
||||
|
||||
@implementation TestsBlockCallbacks { |
||||
void (^_initialMetadataCallback)(NSDictionary *); |
||||
void (^_dataCallback)(id); |
||||
void (^_closeCallback)(NSDictionary *, NSError *); |
||||
void (^_writeMessageCallback)(void); |
||||
dispatch_queue_t _dispatchQueue; |
||||
} |
||||
|
||||
- (instancetype)initWithInitialMetadataCallback:(void (^)(NSDictionary *))initialMetadataCallback |
||||
dataCallback:(void (^)(id))dataCallback |
||||
closeCallback:(void (^)(NSDictionary *, NSError *))closeCallback |
||||
writeMessageCallback:(void (^)(void))writeMessageCallback { |
||||
if ((self = [super init])) { |
||||
_initialMetadataCallback = initialMetadataCallback; |
||||
_dataCallback = dataCallback; |
||||
_closeCallback = closeCallback; |
||||
_writeMessageCallback = writeMessageCallback; |
||||
_dispatchQueue = dispatch_queue_create(nil, DISPATCH_QUEUE_SERIAL); |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)didReceiveInitialMetadata:(NSDictionary *)initialMetadata { |
||||
if (_initialMetadataCallback) { |
||||
_initialMetadataCallback(initialMetadata); |
||||
} |
||||
} |
||||
|
||||
- (void)didReceiveProtoMessage:(id)message { |
||||
if (_dataCallback) { |
||||
_dataCallback(message); |
||||
} |
||||
} |
||||
|
||||
- (void)didCloseWithTrailingMetadata:(NSDictionary *)trailingMetadata error:(NSError *)error { |
||||
if (_closeCallback) { |
||||
_closeCallback(trailingMetadata, error); |
||||
} |
||||
} |
||||
|
||||
- (void)didWriteMessage { |
||||
if (_writeMessageCallback) { |
||||
_writeMessageCallback(); |
||||
} |
||||
} |
||||
|
||||
- (dispatch_queue_t)dispatchQueue { |
||||
return _dispatchQueue; |
||||
} |
||||
@end |
||||
|
||||
@interface TransportTests : XCTestCase |
||||
|
||||
@end |
||||
|
||||
@implementation TransportTests |
||||
|
||||
- (void)testTransportInterceptors { |
||||
__weak XCTestExpectation *expectComplete = |
||||
[self expectationWithDescription:@"Expect call complete"]; |
||||
[GRPCFakeTransportFactory sharedInstance].nextTransportInstance = nil; |
||||
|
||||
[[GRPCFakeTransportFactory sharedInstance] |
||||
setTransportInterceptorFactories:@[ [DummyInterceptorFactory sharedInstance] ]]; |
||||
GRPCRequestOptions *requestOptions = |
||||
[[GRPCRequestOptions alloc] initWithHost:kRemoteHost |
||||
path:@"/UnaryCall" |
||||
safety:GRPCCallSafetyDefault]; |
||||
GRPCMutableCallOptions *callOptions = [[GRPCMutableCallOptions alloc] init]; |
||||
callOptions.transport = kFakeTransportID; |
||||
GRPCCall2 *call = [[GRPCCall2 alloc] |
||||
initWithRequestOptions:requestOptions |
||||
responseHandler:[[TestsBlockCallbacks alloc] |
||||
initWithInitialMetadataCallback:nil |
||||
dataCallback:nil |
||||
closeCallback:^(NSDictionary *trailingMetadata, |
||||
NSError *error) { |
||||
XCTAssertNotNil(error); |
||||
XCTAssertEqual(error.code, |
||||
GRPCErrorCodeCancelled); |
||||
[expectComplete fulfill]; |
||||
} |
||||
writeMessageCallback:nil] |
||||
callOptions:callOptions]; |
||||
[call start]; |
||||
[self waitForExpectationsWithTimeout:TEST_TIMEOUT handler:nil]; |
||||
} |
||||
|
||||
@end |
||||
Loading…
Reference in new issue