88 lines
2.9 KiB
Objective-C
88 lines
2.9 KiB
Objective-C
//
|
||
// CacheKeyFunctionTest.m
|
||
// TrustHttpDNSTests
|
||
//
|
||
// Created by xuyecan on 2024/6/12.
|
||
// Copyright © 2024 trustapp.com. All rights reserved.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import "TestBase.h"
|
||
|
||
@interface CacheKeyFunctionTest : TestBase
|
||
|
||
@end
|
||
|
||
static NSString *sdnsHost = @"sdns1.onlyforhttpdnstest.run.place";
|
||
|
||
@implementation CacheKeyFunctionTest
|
||
|
||
+ (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 setPersistentCacheIPEnabled:YES];
|
||
[self.httpdns setReuseExpiredIPEnabled:NO];
|
||
|
||
[self.httpdns setLogHandler:self];
|
||
|
||
self.currentTimeStamp = [[NSDate date] timeIntervalSince1970];
|
||
}
|
||
|
||
- (void)testSimpleSpecifyingCacheKeySituation {
|
||
dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
|
||
|
||
NSString *testHost = hostNameIpPrefixMap.allKeys.firstObject;
|
||
NSString *cacheKey = [NSString stringWithFormat:@"cacheKey-%@", testHost];
|
||
__block NSString *ipPrefix = hostNameIpPrefixMap[testHost];
|
||
|
||
// 使用正常解析到的ttl
|
||
[self.httpdns setTtlDelegate:nil];
|
||
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:testHost byIpType:HttpdnsQueryIPTypeIpv4 withSdnsParams:nil sdnsCacheKey:cacheKey];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:testHost]);
|
||
XCTAssertGreaterThan(result.ttl, 0);
|
||
// 同步接口，不复用过期ip的情况下，解析出的ip一定是未过期的
|
||
XCTAssertLessThan([[NSDate date] timeIntervalSince1970], result.lastUpdatedTimeInterval + result.ttl);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [testHost UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
dispatch_semaphore_signal(semaphore);
|
||
});
|
||
|
||
dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
|
||
|
||
[NSThread sleepForTimeInterval:3];
|
||
|
||
// 清空缓存
|
||
[self.httpdns.requestManager cleanAllHostMemoryCache];
|
||
|
||
// 从db再加载到缓存<E7BC93><E5AD98>?
|
||
[self.httpdns.requestManager syncLoadCacheFromDbToMemory];
|
||
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:testHost byIpType:HttpdnsQueryIPTypeIpv4];
|
||
// 没有使用cacheKey，所以这里应该是nil
|
||
XCTAssertNil(result);
|
||
|
||
result = [self.httpdns resolveHostSyncNonBlocking:testHost byIpType:HttpdnsQueryIPTypeIpv4 withSdnsParams:nil sdnsCacheKey:cacheKey];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:testHost]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}
|
||
|
||
@end
|