//
//  ScheduleCenterV6Test.m
//  TrustHttpDNSTests
//
//  Created by xuyecan on 2024/6/17.
//  Copyright © 2024 trustapp.com. All rights reserved.
//
#import <Foundation/Foundation.h>
#import <OCMock/OCMock.h>
#import "TestBase.h"
#import "HttpdnsHostObject.h"
#import "HttpdnsScheduleExecutor.h"
#import "HttpdnsScheduleCenter.h"
#import "HttpdnsService.h"
#import "HttpdnsService_Internal.h"
#import "HttpdnsUtil.h"


/**
 * 由于使用OCMock在连续的测试用例中重复Mock对象(即使每次都已经stopMocking)会有内存错乱的问题，
 * 目前还解决不了，所以这个类中的测试case，需要手动单独执�?
 */
@interface ScheduleCenterV6Test : TestBase

@end


@implementation ScheduleCenterV6Test

+ (void)setUp {
    [super setUp];
}

+ (void)tearDown {
    [super tearDown];
}

- (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:NO];

    [self.httpdns setLogHandler:self];

    self.currentTimeStamp = [[NSDate date] timeIntervalSince1970];
}

- (void)tearDown {
    [super tearDown];
}

- (void)testUpdateFailureWillMoveToNextUpdateServer {
    [self presetNetworkEnvAsIpv6];

    HttpdnsScheduleExecutor *realRequest = [HttpdnsScheduleExecutor new];
    id mockRequest = OCMPartialMock(realRequest);
    OCMStub([mockRequest fetchRegionConfigFromServer:[OCMArg any] error:(NSError * __autoreleasing *)[OCMArg anyPointer]])
        .andReturn(nil);

    id mockRequestClass = OCMClassMock([HttpdnsScheduleExecutor class]);
    OCMStub([mockRequestClass new]).andReturn(mockRequest);

    HttpdnsScheduleCenter *scheduleCenter = [[HttpdnsScheduleCenter alloc] initWithAccountId:100000];

    NSArray<NSString *> *updateServerHostList = [scheduleCenter currentUpdateServerV4HostList];

    int updateServerCount = (int)[updateServerHostList count];
    XCTAssertGreaterThan(updateServerCount, 0);

    // 指定已经重试2次，避免重试影响计算
    [scheduleCenter asyncUpdateRegionScheduleConfigAtRetry:2];
    [NSThread sleepForTimeInterval:0.1];

    NSString *activeUpdateHost = [scheduleCenter getActiveUpdateServerHost];

    // 因为可能是域名，所以只判断一定不是ipv4
    XCTAssertFalse([HttpdnsUtil isIPv4Address:activeUpdateHost]);

    OCMVerify([mockRequest fetchRegionConfigFromServer:[OCMArg any] error:(NSError * __autoreleasing *)[OCMArg anyPointer]]);

    [NSThread sleepForTimeInterval:3];
}

@end