134 lines
4.8 KiB
Objective-C
134 lines
4.8 KiB
Objective-C
//
|
||
// TestBase.m
|
||
// TrustHttpDNS
|
||
//
|
||
// Created by ElonChan（地风） on 2017/4/14.
|
||
// Copyright © 2017�trustapp.com. All rights reserved.
|
||
//
|
||
|
||
#import "TestBase.h"
|
||
#import <mach/mach.h>
|
||
#import "HttpdnsIpStackDetector.h"
|
||
|
||
NSDictionary<NSString *, NSString *> *hostNameIpPrefixMap;
|
||
|
||
@implementation TestBase
|
||
|
||
+ (void)setUp {
|
||
hostNameIpPrefixMap = @{
|
||
@"v4host1.onlyforhttpdnstest.run.place": @"0.0.1",
|
||
@"v4host2.onlyforhttpdnstest.run.place": @"0.0.2",
|
||
@"v4host3.onlyforhttpdnstest.run.place": @"0.0.3",
|
||
@"v4host4.onlyforhttpdnstest.run.place": @"0.0.4",
|
||
@"v4host5.onlyforhttpdnstest.run.place": @"0.0.5"
|
||
};
|
||
}
|
||
|
||
- (void)setUp {
|
||
[super setUp];
|
||
}
|
||
|
||
- (void)tearDown {
|
||
[super tearDown];
|
||
}
|
||
|
||
- (void)log:(NSString *)logStr {
|
||
mach_port_t threadID = mach_thread_self();
|
||
NSString *threadIDString = [NSString stringWithFormat:@"%x", threadID];
|
||
printf("%ld-%s %s\n", (long)[[NSDate date] timeIntervalSince1970], [threadIDString UTF8String], [logStr UTF8String]);
|
||
}
|
||
|
||
- (HttpdnsHostObject *)constructSimpleIpv4HostObject {
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
hostObject.hostName = ipv4OnlyHost;
|
||
hostObject.v4ttl = 60;
|
||
HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
|
||
[ip1 setIp:ipv41];
|
||
HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
|
||
[ip2 setIp:ipv42];
|
||
hostObject.v4Ips = @[ip1, ip2];
|
||
hostObject.lastIPv4LookupTime = self.currentTimeStamp;
|
||
return hostObject;
|
||
}
|
||
|
||
- (HttpdnsHostObject *)constructSimpleIpv6HostObject {
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
hostObject.hostName = ipv4OnlyHost;
|
||
hostObject.v6ttl = 60;
|
||
HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
|
||
[ip1 setIp:@"2001:4860:4860::8888"];
|
||
HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
|
||
[ip2 setIp:@"2001:4860:4860::8844"];
|
||
hostObject.v6Ips = @[ip1, ip2];
|
||
hostObject.lastIPv6LookupTime = self.currentTimeStamp;
|
||
return hostObject;
|
||
}
|
||
|
||
- (HttpdnsHostObject *)constructSimpleIpv4AndIpv6HostObject {
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
hostObject.hostName = ipv4AndIpv6Host;
|
||
hostObject.v4ttl = 60;
|
||
HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
|
||
[ip1 setIp:ipv41];
|
||
HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
|
||
[ip2 setIp:ipv42];
|
||
hostObject.v4Ips = @[ip1, ip2];
|
||
hostObject.lastIPv4LookupTime = self.currentTimeStamp;
|
||
|
||
hostObject.v6ttl = 60;
|
||
HttpdnsIpObject *ip3 = [[HttpdnsIpObject alloc] init];
|
||
[ip3 setIp:ipv61];
|
||
HttpdnsIpObject *ip4 = [[HttpdnsIpObject alloc] init];
|
||
[ip4 setIp:ipv62];
|
||
hostObject.v6Ips = @[ip3, ip4];
|
||
hostObject.lastIPv6LookupTime = self.currentTimeStamp;
|
||
return hostObject;
|
||
}
|
||
|
||
- (void)presetNetworkEnvAsIpv4 {
|
||
HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
|
||
OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpv4Only);
|
||
OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(NO);
|
||
|
||
id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
|
||
OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
|
||
}
|
||
|
||
- (void)presetNetworkEnvAsIpv6 {
|
||
HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
|
||
OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpv6Only);
|
||
OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(YES);
|
||
|
||
id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
|
||
OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
|
||
}
|
||
|
||
- (void)presetNetworkEnvAsIpv4AndIpv6 {
|
||
HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
|
||
OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpDual);
|
||
OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(NO);
|
||
|
||
id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
|
||
OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
|
||
}
|
||
|
||
- (void)shouldNotHaveCallNetworkRequestWhenResolving:(void (^)(void))resolvingBlock {
|
||
HttpDnsService *httpdns = [HttpDnsService sharedInstance];
|
||
HttpdnsRequestManager *requestManager = httpdns.requestManager;
|
||
HttpdnsRequestManager *mockScheduler = OCMPartialMock(requestManager);
|
||
OCMReject([mockScheduler executeRequest:[OCMArg any] retryCount:0]);
|
||
resolvingBlock();
|
||
OCMVerifyAll(mockScheduler);
|
||
}
|
||
|
||
- (void)shouldHaveCalledRequestWhenResolving:(void (^)(void))resolvingBlock {
|
||
HttpDnsService *httpdns = [HttpDnsService sharedInstance];
|
||
HttpdnsRequestManager *requestManager = httpdns.requestManager;
|
||
HttpdnsRequestManager *mockScheduler = OCMPartialMock(requestManager);
|
||
OCMExpect([mockScheduler executeRequest:[OCMArg any] retryCount:0]).andReturn(nil);
|
||
resolvingBlock();
|
||
OCMVerifyAll(mockScheduler);
|
||
}
|
||
|
||
@end
|