302 lines
13 KiB
Objective-C
302 lines
13 KiB
Objective-C
//
|
||
// PresetCacheAndRetrieveTest.m
|
||
// TrustHttpDNSTests
|
||
//
|
||
// Created by xuyecan on 2024/5/26.
|
||
// Copyright ﾂｩ 2024 trustapp.com. All rights reserved.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import <OCMock/OCMock.h>
|
||
#import "TestBase.h"
|
||
#import "HttpdnsHostObject.h"
|
||
#import "HttpdnsService.h"
|
||
#import "HttpdnsService_Internal.h"
|
||
|
||
|
||
/**
|
||
* 逕ｱ莠惹ｽｿ逕ｨOCMock蝨ｨ霑樒ｻｭ逧<EFBDAD>ｵ玖ｯ慕畑萓倶ｸｭ驥榊､庚ock蟇ｹ雎｡(蜊ｳ菴ｿ豈乗ｬ｡驛ｽ蟾ｲ扈峻topMocking)莨壽怏蜀<E6808F>ｭ倬漠荵ｱ逧<EFBDB1>琉鬚假ｼ<E58187>
|
||
* 逶ｮ蜑崎ｿ倩ｧ｣蜀ｳ荳堺ｺ<E5A0BA>ｼ梧園莉･霑吩ｸｪ邀ｻ荳ｭ逧<EFBDAD>ｵ玖ｯ苗ase<73>碁怙隕∵焔蜉ｨ蜊慕峡謇ｧ陦?
|
||
*/
|
||
@interface PresetCacheAndRetrieveTest : TestBase
|
||
|
||
@end
|
||
|
||
|
||
@implementation PresetCacheAndRetrieveTest
|
||
|
||
+ (void)setUp {
|
||
[super setUp];
|
||
|
||
HttpDnsService *httpdns = [[HttpDnsService alloc] initWithAccountID:100000];
|
||
[httpdns setLogEnabled:YES];
|
||
}
|
||
|
||
+ (void)tearDown {
|
||
[super tearDown];
|
||
}
|
||
|
||
- (void)setUp {
|
||
[super setUp];
|
||
|
||
self.httpdns = [HttpDnsService sharedInstance];
|
||
self.currentTimeStamp = [[NSDate date] timeIntervalSince1970];
|
||
}
|
||
|
||
- (void)tearDown {
|
||
[super tearDown];
|
||
}
|
||
|
||
// 鄂醍ｻ懈ュ蜀ｵ荳ｺipv4荳狗噪郛灘ｭ俶ｵ玖ｯ<E78E96>
|
||
- (void)testSimplyRetrieveCachedResultUnderIpv4Only {
|
||
[self presetNetworkEnvAsIpv4];
|
||
[self.httpdns cleanAllHostCache];
|
||
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv4AndIpv6HostObject];
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv4AndIpv6Host cacheKey:ipv4AndIpv6Host underQueryIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺipv4<76>梧響蛻ｰipv4扈捺棡
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺipv6<76>梧響蛻ｰipv6扈捺棡
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv6];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// both
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺauto<74>悟宵諡ｿ蛻ｰipv4扈捺棡
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeAuto];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 0);
|
||
}
|
||
|
||
// 鄂醍ｻ懆ｯｷ豎ゆｸｺipv6荳狗噪郛灘ｭ俶ｵ玖ｯ<E78E96>
|
||
- (void)testSimplyRetrieveCachedResultUnderIpv6Only {
|
||
[self presetNetworkEnvAsIpv6];
|
||
[self.httpdns cleanAllHostCache];
|
||
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv4AndIpv6HostObject];
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv4AndIpv6Host cacheKey:ipv4AndIpv6Host underQueryIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺipv4<76>梧響蛻ｰipv4扈捺棡
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺipv6<76>梧響蛻ｰipv6扈捺棡
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv6];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// both
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// 隸ｷ豎らｱｻ蝙倶ｸｺauto<74>梧ｳｨ諢擾ｼ梧<EFBDBC>莉ｬ隶､荳ｺipv6only蜿ｪ蟄伜惠逅<E683A0>ｮｺ荳奇ｼ梧ｯ泌ｦょｮ樣ｪ悟ｮ､邇ｯ蠅<EFBDAF>
|
||
// 蝗<>豁､<E8B181>景pv4逧<34>慍蝮譏ｯ荳螳壻ｼ壼悉隗｣譫千噪<E58D83>径uto逧<6F>ｽ懃畑蝨ｨ莠趣ｼ悟ｦよ棡蜿醍鴫鄂醍ｻ懆ｿ俶髪謖（pv6<76>碁ぅ蟆ｱ螟夊執蜿吠pv6逧<36>ｻ捺<EFBDBB>?
|
||
// 蝗<>豁､<E8B181>瑚ｿ咎㈹蠕怜芦逧<E88AA6>ｹ滓弍ipv4+ipv6
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeAuto];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
}
|
||
|
||
// 鄂醍ｻ懈ュ蜀ｵ荳ｺipv4蜥景pv6荳狗噪郛灘ｭ俶ｵ玖ｯ<E78E96>
|
||
- (void)testSimplyRetrieveCachedResultUnderDualStack {
|
||
[self presetNetworkEnvAsIpv4AndIpv6];
|
||
[self.httpdns cleanAllHostCache];
|
||
|
||
// 蟄伜<E89F84>ipv4蜥景pv6逧<36>慍蝮
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv4AndIpv6HostObject];
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv4AndIpv6Host cacheKey:ipv4AndIpv6Host underQueryIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
// 蜿ｪ隸ｷ豎Ｊpv4
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
|
||
// 蜿ｪ隸ｷ豎Ｊpv6
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeIpv6];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// 隸ｷ豎Ｊpv4蜥景pv6
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
|
||
// 閾ｪ蜉ｨ蛻､譁ｭ邀ｻ蝙<EFBDBB>
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeAuto];
|
||
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4AndIpv6Host]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
XCTAssertTrue([result.ipv6s count] == 2);
|
||
XCTAssertTrue([result.ipv6s[0] isEqualToString:ipv61]);
|
||
}
|
||
|
||
// ttl縲〕astLookupTime<6D>景pv4蜥景pv6譏ｯ蛻<EFBDAF>ｼ螟<C280>炊逧?
|
||
- (void)testTTLAndLastLookUpTime {
|
||
[self presetNetworkEnvAsIpv4AndIpv6];
|
||
[self.httpdns cleanAllHostCache];
|
||
|
||
// 蟄伜<E89F84>ipv4蜥景pv6逧<36>慍蝮
|
||
HttpdnsHostObject *hostObject1 = [self constructSimpleIpv4AndIpv6HostObject];
|
||
hostObject1.v4ttl = 200;
|
||
hostObject1.v6ttl = 300;
|
||
|
||
int64_t currentTimestamp = [[NSDate new] timeIntervalSince1970];
|
||
|
||
hostObject1.lastIPv4LookupTime = currentTimestamp - 1;
|
||
hostObject1.lastIPv6LookupTime = currentTimestamp - 2;
|
||
|
||
// 隨ｬ荳谺｡隶ｾ鄂ｮ郛灘ｭ?
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject1 host:ipv4AndIpv6Host cacheKey:ipv4AndIpv6Host underQueryIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
// auto蝨ｨ蠖灘燕邇ｯ蠅<EFBDAF>ｸ句叉隸ｷ豎Ｊpv4蜥景pv6
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeAuto];
|
||
XCTAssertEqual(result.ttl, hostObject1.v4ttl);
|
||
XCTAssertEqual(result.lastUpdatedTimeInterval, hostObject1.lastIPv4LookupTime);
|
||
XCTAssertEqual(result.v6ttl, hostObject1.v6ttl);
|
||
XCTAssertEqual(result.v6LastUpdatedTimeInterval, hostObject1.lastIPv6LookupTime);
|
||
|
||
HttpdnsHostObject *hostObject2 = [self constructSimpleIpv4HostObject];
|
||
hostObject2.hostName = ipv4AndIpv6Host;
|
||
hostObject2.v4ttl = 600;
|
||
hostObject2.lastIPv4LookupTime = currentTimestamp - 10;
|
||
|
||
// 蜊慕峡蝨ｨ郛灘ｭ俶峩譁ｰipv4蝨ｰ蝮逧<C280>嶌蜈ｳ菫｡諱?
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject2 host:ipv4AndIpv6Host cacheKey:ipv4AndIpv6Host underQueryIpType:HttpdnsQueryIPTypeIpv4];
|
||
|
||
// v4逧<34>ｿ｡諱ｯ蜿醍函蜿伜喧<E4BC9C>計6逧<36>ｿ｡諱ｯ菫晄戟荳榊<E88DB3>?
|
||
result = [self.httpdns resolveHostSyncNonBlocking:ipv4AndIpv6Host byIpType:HttpdnsQueryIPTypeAuto];
|
||
XCTAssertEqual(result.ttl, hostObject2.v4ttl);
|
||
XCTAssertEqual(result.lastUpdatedTimeInterval, hostObject2.lastIPv4LookupTime);
|
||
XCTAssertEqual(result.v6ttl, hostObject1.v6ttl);
|
||
XCTAssertEqual(result.v6LastUpdatedTimeInterval, hostObject1.lastIPv6LookupTime);
|
||
}
|
||
|
||
// 蜿ｪ郛灘ｭ亙pv4蜊墓<E89C8A>育噪蝨ｰ蝮<E89DAE>梧潔隸ｷ豎ょ曙譬育ｱｻ蝙句ｭ伜<EFBDAD><E4BC9C>梧ｭ､譌ｶ莨壽<E88EA8><E5A3BD>ｮｰ隸･蝓溷錐豐｡譛永pv6蝨ｰ蝮
|
||
// 謖蛾｢<E89BBE>悄<EFBFBD>御ｼ壼愛譁ｭ隸･蝓溷錐豐｡譛永pv6蝨ｰ蝮<E89DAE>悟屏豁､荳堺ｼ夊ｿ泌屓ipv6蝨ｰ蝮<E89DAE>御ｹ滉ｸ堺ｼ壼書隸ｷ豎?
|
||
- (void)testMergeNoIpv6ResultAndGetBoth {
|
||
[self presetNetworkEnvAsIpv4AndIpv6];
|
||
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv4HostObject];
|
||
|
||
// 蜿梧<E89CBF>井ｸ玖ｧ｣譫千ｻ捺棡莉<E6A3A1>怏ipv4<76>悟粋蟷ｶ譌ｶ莨壽<E88EA8><E5A3BD>ｮｰ隸･host譌<74>ipv6
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv4OnlyHost cacheKey:ipv4OnlyHost underQueryIpType:HttpdnsQueryIPTypeBoth];
|
||
|
||
[self shouldNotHaveCallNetworkRequestWhenResolving:^{
|
||
dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
|
||
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:ipv4OnlyHost byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4OnlyHost]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
|
||
result = [self.httpdns resolveHostSync:ipv4OnlyHost byIpType:HttpdnsQueryIPTypeAuto];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:ipv4OnlyHost]);
|
||
XCTAssertTrue([result.ips count] == 2);
|
||
XCTAssertTrue([result.ips[0] isEqualToString:ipv41]);
|
||
|
||
dispatch_semaphore_signal(semaphore);
|
||
});
|
||
|
||
dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
|
||
}];
|
||
}
|
||
|
||
// 郛灘ｭ亙pv4蜊墓<E89C8A>育噪蝨ｰ蝮<E89DAE>御ｽ<E5BEA1>弍隸ｷ豎Ｊpv4邀ｻ蝙句ｭ伜<EFBDAD><E4BC9C>梧ｭ､譌ｶ荳堺ｼ壽遠譬<E981A0>ｮｰ豐｡譛永pv6
|
||
// 莠取弍<E58F96>瑚ｯｻ蜿匁慮<E58C81>御ｼ壼ｰ晁ｯ募書隸ｷ豎り執蜿吠pv6蝨ｰ蝮
|
||
- (void)testMergeOnlyIpv4ResultAndGetBoth {
|
||
[self presetNetworkEnvAsIpv4AndIpv6];
|
||
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv4HostObject];
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv4OnlyHost cacheKey:ipv4OnlyHost underQueryIpType:HttpdnsQueryIPTypeIpv4];
|
||
|
||
dispatch_semaphore_t sema = dispatch_semaphore_create(0);
|
||
|
||
// 菴ｿ逕ｨ蜷梧ｭ･謗･蜿｣<E89CBF>瑚ｦ∝<EFBDA6>謐｢蛻ｰ蠑よｭ･郤ｿ遞具ｼ悟凄蛻吝<E89BBB>驛ｨ莨夊<E88EA8>蟾ｱ蛻<EFBDB1>
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
[self shouldHaveCalledRequestWhenResolving:^{
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:ipv4OnlyHost byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNil(result);
|
||
dispatch_semaphore_signal(sema);
|
||
}];
|
||
});
|
||
|
||
dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
|
||
}
|
||
|
||
// 郛灘ｭ亙pv6蜊墓<E89C8A>育噪蝨ｰ蝮<E89DAE>御ｽ<E5BEA1>弍隸ｷ豎Ｊpv6邀ｻ蝙句ｭ伜<EFBDAD><E4BC9C>梧ｭ､譌ｶ荳堺ｼ壽遠譬<E981A0>ｮｰ豐｡譛永pv4
|
||
// 莠取弍<E58F96>瑚ｯｻ蜿匁慮<E58C81>御ｼ壼ｰ晁ｯ募書隸ｷ豎り執蜿吠pv4蝨ｰ蝮
|
||
- (void)testMergeOnlyIpv6ResultAndGetBoth {
|
||
[self presetNetworkEnvAsIpv4AndIpv6];
|
||
|
||
HttpdnsHostObject *hostObject = [self constructSimpleIpv6HostObject];
|
||
[self.httpdns.requestManager mergeLookupResultToManager:hostObject host:ipv6OnlyHost cacheKey:ipv6OnlyHost underQueryIpType:HttpdnsQueryIPTypeIpv6];
|
||
|
||
dispatch_semaphore_t sema = dispatch_semaphore_create(0);
|
||
|
||
// 菴ｿ逕ｨ蜷梧ｭ･謗･蜿｣<E89CBF>瑚ｦ∝<EFBDA6>謐｢蛻ｰ蠑よｭ･郤ｿ遞具ｼ悟凄蛻吝<E89BBB>驛ｨ莨夊<E88EA8>蟾ｱ蛻<EFBDB1>
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
[self shouldHaveCalledRequestWhenResolving:^{
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:ipv6OnlyHost byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNil(result);
|
||
dispatch_semaphore_signal(sema);
|
||
}];
|
||
});
|
||
|
||
dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
|
||
}
|
||
|
||
@end
|