1/* 2 * 3 * Copyright 2015-2016 gRPC authors. 4 * 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 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 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. 16 * 17 */ 18 19#import "GRPCCall+Tests.h" 20 21#import "private/GRPCCore/GRPCHost.h" 22 23#import "GRPCCallOptions.h" 24 25@implementation GRPCCall (Tests) 26 27+ (void)useTestCertsPath:(NSString *)certsPath 28 testName:(NSString *)testName 29 forHost:(NSString *)host { 30 if (!host) { 31 [NSException raise:NSInvalidArgumentException format:@"host must be provided."]; 32 } 33 if (!certsPath) { 34 [NSException raise:NSInvalidArgumentException format:@"certpath be provided."]; 35 } 36 if (!testName) { 37 [NSException raise:NSInvalidArgumentException format:@"testname must be provided."]; 38 } 39 NSError *error = nil; 40 NSString *certs = [NSString stringWithContentsOfFile:certsPath 41 encoding:NSUTF8StringEncoding 42 error:&error]; 43 if (error != nil) { 44 [NSException raise:[error localizedDescription] format:@"failed to load certs"]; 45 } 46 47 GRPCHost *hostConfig = [GRPCHost hostWithAddress:host]; 48 [hostConfig setTLSPEMRootCerts:certs withPrivateKey:nil withCertChain:nil error:nil]; 49 hostConfig.hostNameOverride = testName; 50} 51 52+ (void)useInsecureConnectionsForHost:(NSString *)host { 53 GRPCHost *hostConfig = [GRPCHost hostWithAddress:host]; 54 hostConfig.transportType = GRPCTransportTypeInsecure; 55} 56 57+ (void)resetHostSettings { 58 [GRPCHost resetAllHostSettings]; 59} 60@end 61