阿里sdk

2026-02-20 17:56:24 +08:00
parent 39524692e5
commit f3af234308
524 changed files with 58345 additions and 0 deletions
--- a/EdgeHttpDNS/sdk/ios/AlicloudHttpDNSTests/Testbase/TestBase.m
+++ b/EdgeHttpDNS/sdk/ios/AlicloudHttpDNSTests/Testbase/TestBase.m
@@ -0,0 +1,133 @@
+//
+//  TestBase.m
+//  AlicloudHttpDNS
+//
+//  Created by ElonChan（地风） on 2017/4/14.
+//  Copyright © 2017年 alibaba-inc.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