102 lines
3.4 KiB
Objective-C
102 lines
3.4 KiB
Objective-C
//
|
||
// EnableReuseExpiredIpTest.m
|
||
// AlicloudHttpDNSTests
|
||
//
|
||
// Created by xuyecan on 2024/5/28.
|
||
// Copyright © 2024 alibaba-inc.com. All rights reserved.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import "TestBase.h"
|
||
|
||
@interface EnableReuseExpiredIpTest : TestBase <HttpdnsTTLDelegate>
|
||
|
||
@end
|
||
|
||
static int ttlForTest = 3;
|
||
|
||
@implementation EnableReuseExpiredIpTest
|
||
|
||
+ (void)setUp {
|
||
[super setUp];
|
||
}
|
||
|
||
- (void)setUp {
|
||
[super setUp];
|
||
|
||
static dispatch_once_t onceToken;
|
||
dispatch_once(&onceToken, ^{
|
||
self.httpdns = [[HttpDnsService alloc] initWithAccountID:100000];
|
||
});
|
||
|
||
[self.httpdns setLogEnabled:YES];
|
||
|
||
[self.httpdns setReuseExpiredIPEnabled:YES];
|
||
|
||
[self.httpdns setTtlDelegate:self];
|
||
[self.httpdns setLogHandler:self];
|
||
|
||
self.currentTimeStamp = [[NSDate date] timeIntervalSince1970];
|
||
}
|
||
|
||
- (int64_t)httpdnsHost:(NSString *)host ipType:(AlicloudHttpDNS_IPType)ipType ttl:(int64_t)ttl {
|
||
// 在测试中域名快速过期
|
||
return ttlForTest;
|
||
}
|
||
|
||
- (void)testReuseExpiredIp {
|
||
NSString *host = hostNameIpPrefixMap.allKeys.firstObject;
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
|
||
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
// 清空缓存
|
||
[self.httpdns.requestManager cleanAllHostMemoryCache];
|
||
|
||
// 首次解析
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
XCTAssertGreaterThan(result.ttl, 0);
|
||
XCTAssertLessThanOrEqual(result.ttl, ttlForTest);
|
||
XCTAssertLessThan([[NSDate date] timeIntervalSince1970], result.lastUpdatedTimeInterval + result.ttl);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
|
||
// 等待过期
|
||
[NSThread sleepForTimeInterval:ttlForTest + 1];
|
||
|
||
// 重复解析
|
||
HttpdnsResult *result2 = [self.httpdns resolveHostSync:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
XCTAssertNotNil(result2);
|
||
XCTAssertTrue([result2.host isEqualToString:host]);
|
||
XCTAssertGreaterThan(result2.ttl, 0);
|
||
XCTAssertLessThanOrEqual(result2.ttl, ttlForTest);
|
||
// 因为运行复用过期解析结果，因此这里获得的一定是已经过期的
|
||
XCTAssertGreaterThan([[NSDate date] timeIntervalSince1970], result2.lastUpdatedTimeInterval + result2.ttl);
|
||
NSString *firstIp2 = [result2 firstIpv4Address];
|
||
XCTAssertTrue([firstIp2 hasPrefix:ipPrefix]);
|
||
|
||
// 等待第二次解析触发的请求完成
|
||
[NSThread sleepForTimeInterval:1];
|
||
|
||
// 再次使用nonblocking方法解析，此时应该已经拿到有效结果
|
||
HttpdnsResult *result3 = [self.httpdns resolveHostSyncNonBlocking:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
XCTAssertNotNil(result3);
|
||
XCTAssertTrue([result3.host isEqualToString:host]);
|
||
XCTAssertGreaterThan(result3.ttl, 0);
|
||
XCTAssertLessThanOrEqual(result3.ttl, ttlForTest);
|
||
// 有效结果必定未过期
|
||
XCTAssertLessThan([[NSDate date] timeIntervalSince1970], result3.lastUpdatedTimeInterval + result3.ttl);
|
||
NSString *firstIp3 = [result3 firstIpv4Address];
|
||
XCTAssertTrue([firstIp3 hasPrefix:ipPrefix]);
|
||
|
||
dispatch_semaphore_signal(semaphore);
|
||
});
|
||
|
||
dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
|
||
}
|
||
|
||
@end
|