166 lines
5.4 KiB
Objective-C
166 lines
5.4 KiB
Objective-C
//
|
||
// HttpdnsHostObjectTest.m
|
||
// TrustHttpDNSTests
|
||
//
|
||
// Created by xuyecan on 2025/3/14.
|
||
// Copyright © 2025 trustapp.com. All rights reserved.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import "../Testbase/TestBase.h"
|
||
#import "HttpdnsHostObject.h"
|
||
#import "HttpdnsIpObject.h"
|
||
#import <OCMock/OCMock.h>
|
||
|
||
@interface HttpdnsHostObjectTest : TestBase
|
||
|
||
@end
|
||
|
||
@implementation HttpdnsHostObjectTest
|
||
|
||
- (void)setUp {
|
||
[super setUp];
|
||
}
|
||
|
||
- (void)tearDown {
|
||
[super tearDown];
|
||
}
|
||
|
||
#pragma mark - 基本属性测<E680A7><E6B58B>?
|
||
|
||
- (void)testHostObjectProperties {
|
||
// 创建一个HttpdnsHostObject实例
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
|
||
// 设置基本属<E69CAC><E5B19E>?
|
||
hostObject.host = @"example.com";
|
||
hostObject.ttl = 60;
|
||
hostObject.queryTimes = 1;
|
||
hostObject.clientIP = @"192.168.1.1";
|
||
|
||
// 验证属性<E5B19E><E680A7>?
|
||
XCTAssertEqualObjects(hostObject.host, @"example.com", @"host属性应该被正确设置");
|
||
XCTAssertEqual(hostObject.ttl, 60, @"ttl属性应该被正确设置");
|
||
XCTAssertEqual(hostObject.queryTimes, 1, @"queryTimes属性应该被正确设置");
|
||
XCTAssertEqualObjects(hostObject.clientIP, @"192.168.1.1", @"clientIP属性应该被正确设置");
|
||
}
|
||
|
||
#pragma mark - IP对象测试
|
||
|
||
- (void)testIpObjectProperties {
|
||
// 创建一个HttpdnsIpObject实例
|
||
HttpdnsIpObject *ipObject = [[HttpdnsIpObject alloc] init];
|
||
|
||
// 设置基本属<E69CAC><E5B19E>?
|
||
ipObject.ip = @"1.2.3.4";
|
||
ipObject.ttl = 300;
|
||
ipObject.priority = 10;
|
||
ipObject.detectRT = 50; // 测试新添加的detectRT属<54><E5B19E>?
|
||
|
||
// 验证属性<E5B19E><E680A7>?
|
||
XCTAssertEqualObjects(ipObject.ip, @"1.2.3.4", @"ip属性应该被正确设置");
|
||
XCTAssertEqual(ipObject.ttl, 300, @"ttl属性应该被正确设置");
|
||
XCTAssertEqual(ipObject.priority, 10, @"priority属性应该被正确设置");
|
||
XCTAssertEqual(ipObject.detectRT, 50, @"detectRT属性应该被正确设置");
|
||
}
|
||
|
||
- (void)testIpObjectDetectRTMethods {
|
||
// 创建一个HttpdnsIpObject实例
|
||
HttpdnsIpObject *ipObject = [[HttpdnsIpObject alloc] init];
|
||
|
||
// 测试默认<E9BB98><E8AEA4>?
|
||
XCTAssertEqual(ipObject.detectRT, -1, @"detectRT的默认值应该是-1");
|
||
|
||
// 测试设置检测时<E6B58B><E697B6>?
|
||
[ipObject setDetectRT:100];
|
||
XCTAssertEqual(ipObject.detectRT, 100, @"detectRT应该被正确设置为100");
|
||
|
||
// 测试设置为负<E4B8BA><E8B49F>?
|
||
[ipObject setDetectRT:-5];
|
||
XCTAssertEqual(ipObject.detectRT, -1, @"设置负值时detectRT应该被设置为-1");
|
||
|
||
// 测试设置<E8AEBE><E7BDAE>?
|
||
[ipObject setDetectRT:0];
|
||
XCTAssertEqual(ipObject.detectRT, 0, @"detectRT应该被正确设置为0");
|
||
}
|
||
|
||
#pragma mark - 主机对象IP管理测试
|
||
|
||
- (void)testHostObjectIpManagement {
|
||
// 创建一个HttpdnsHostObject实例
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
hostObject.host = @"example.com";
|
||
|
||
// 创建IP对象
|
||
HttpdnsIpObject *ipv4Object = [[HttpdnsIpObject alloc] init];
|
||
ipv4Object.ip = @"1.2.3.4";
|
||
ipv4Object.ttl = 300;
|
||
ipv4Object.detectRT = 50;
|
||
|
||
HttpdnsIpObject *ipv6Object = [[HttpdnsIpObject alloc] init];
|
||
ipv6Object.ip = @"2001:db8::1";
|
||
ipv6Object.ttl = 600;
|
||
ipv6Object.detectRT = 80;
|
||
|
||
// 添加IP对象到主机对<E69CBA><E5AFB9>?
|
||
[hostObject addIpv4:ipv4Object];
|
||
[hostObject addIpv6:ipv6Object];
|
||
|
||
// 验证IP对象是否被正确添<E7A1AE><E6B7BB>?
|
||
XCTAssertEqual(hostObject.ipv4List.count, 1, @"应该<EFBFBD><EFBFBD>?个IPv4对象");
|
||
XCTAssertEqual(hostObject.ipv6List.count, 1, @"应该<EFBFBD><EFBFBD>?个IPv6对象");
|
||
|
||
// 验证IP对象的属<E79A84><E5B19E>?
|
||
HttpdnsIpObject *retrievedIpv4 = hostObject.ipv4List.firstObject;
|
||
XCTAssertEqualObjects(retrievedIpv4.ip, @"1.2.3.4", @"IPv4地址应该正确");
|
||
XCTAssertEqual(retrievedIpv4.detectRT, 50, @"IPv4的detectRT应该正确");
|
||
|
||
HttpdnsIpObject *retrievedIpv6 = hostObject.ipv6List.firstObject;
|
||
XCTAssertEqualObjects(retrievedIpv6.ip, @"2001:db8::1", @"IPv6地址应该正确");
|
||
XCTAssertEqual(retrievedIpv6.detectRT, 80, @"IPv6的detectRT应该正确");
|
||
}
|
||
|
||
#pragma mark - IP排序测试
|
||
|
||
- (void)testIpSortingByDetectRT {
|
||
// 创建一个HttpdnsHostObject实例
|
||
HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
|
||
hostObject.host = @"example.com";
|
||
|
||
// 创建多个IP对象，具有不同的检测时<E6B58B><E697B6>?
|
||
HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
|
||
ip1.ip = @"1.1.1.1";
|
||
ip1.detectRT = 100;
|
||
|
||
HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
|
||
ip2.ip = @"2.2.2.2";
|
||
ip2.detectRT = 50;
|
||
|
||
HttpdnsIpObject *ip3 = [[HttpdnsIpObject alloc] init];
|
||
ip3.ip = @"3.3.3.3";
|
||
ip3.detectRT = 200;
|
||
|
||
HttpdnsIpObject *ip4 = [[HttpdnsIpObject alloc] init];
|
||
ip4.ip = @"4.4.4.4";
|
||
ip4.detectRT = -1; // 未检<E69CAA><E6A380>?
|
||
|
||
// 添加IP对象到主机对象（顺序不重要）
|
||
[hostObject addIpv4:ip1];
|
||
[hostObject addIpv4:ip2];
|
||
[hostObject addIpv4:ip3];
|
||
[hostObject addIpv4:ip4];
|
||
|
||
// 获取排序后的IP列表
|
||
NSArray<HttpdnsIpObject *> *sortedIps = [hostObject sortedIpv4List];
|
||
|
||
// 验证排序结果
|
||
// 预期顺序：ip2(50ms) -> ip1(100ms) -> ip3(200ms) -> ip4(-1ms)
|
||
XCTAssertEqual(sortedIps.count, 4, @"应该<EFBFBD><EFBFBD>?个IP对象");
|
||
XCTAssertEqualObjects(sortedIps[0].ip, @"2.2.2.2", @"检测时间最短的IP应该排在第一<EFBFBD><EFBFBD>?);
|
||
XCTAssertEqualObjects(sortedIps[1].ip, @"1.1.1.1", @"检测时间第二短的IP应该排在第二<EFBFBD><EFBFBD>?);
|
||
XCTAssertEqualObjects(sortedIps[2].ip, @"3.3.3.3", @"检测时间第三短的IP应该排在第三<EFBFBD><EFBFBD>?);
|
||
XCTAssertEqualObjects(sortedIps[3].ip, @"4.4.4.4", @"未检测的IP应该排在最<EFBFBD><EFBFBD>?);
|
||
}
|
||
|
||
@end
|