//
//  TestBase.m
//  AlicloudHttpDNS
//
//  Created by ElonChan（地风） on 2017/4/14.
//  Copyright © 2017年 alibaba-inc.com. All rights reserved.
//

#import "TestBase.h"
#import <mach/mach.h>
#import "HttpdnsIpStackDetector.h"

NSDictionary<NSString *, NSString *> *hostNameIpPrefixMap;

@implementation TestBase

+ (void)setUp {
    hostNameIpPrefixMap = @{
        @"v4host1.onlyforhttpdnstest.run.place": @"0.0.1",
        @"v4host2.onlyforhttpdnstest.run.place": @"0.0.2",
        @"v4host3.onlyforhttpdnstest.run.place": @"0.0.3",
        @"v4host4.onlyforhttpdnstest.run.place": @"0.0.4",
        @"v4host5.onlyforhttpdnstest.run.place": @"0.0.5"
    };
}

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

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

- (void)log:(NSString *)logStr {
    mach_port_t threadID = mach_thread_self();
    NSString *threadIDString = [NSString stringWithFormat:@"%x", threadID];
    printf("%ld-%s %s\n", (long)[[NSDate date] timeIntervalSince1970], [threadIDString UTF8String], [logStr UTF8String]);
}

- (HttpdnsHostObject *)constructSimpleIpv4HostObject {
    HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
    hostObject.hostName = ipv4OnlyHost;
    hostObject.v4ttl = 60;
    HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
    [ip1 setIp:ipv41];
    HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
    [ip2 setIp:ipv42];
    hostObject.v4Ips = @[ip1, ip2];
    hostObject.lastIPv4LookupTime = self.currentTimeStamp;
    return hostObject;
}

- (HttpdnsHostObject *)constructSimpleIpv6HostObject {
    HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
    hostObject.hostName = ipv4OnlyHost;
    hostObject.v6ttl = 60;
    HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
    [ip1 setIp:@"2001:4860:4860::8888"];
    HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
    [ip2 setIp:@"2001:4860:4860::8844"];
    hostObject.v6Ips = @[ip1, ip2];
    hostObject.lastIPv6LookupTime = self.currentTimeStamp;
    return hostObject;
}

- (HttpdnsHostObject *)constructSimpleIpv4AndIpv6HostObject {
    HttpdnsHostObject *hostObject = [[HttpdnsHostObject alloc] init];
    hostObject.hostName = ipv4AndIpv6Host;
    hostObject.v4ttl = 60;
    HttpdnsIpObject *ip1 = [[HttpdnsIpObject alloc] init];
    [ip1 setIp:ipv41];
    HttpdnsIpObject *ip2 = [[HttpdnsIpObject alloc] init];
    [ip2 setIp:ipv42];
    hostObject.v4Ips = @[ip1, ip2];
    hostObject.lastIPv4LookupTime = self.currentTimeStamp;

    hostObject.v6ttl = 60;
    HttpdnsIpObject *ip3 = [[HttpdnsIpObject alloc] init];
    [ip3 setIp:ipv61];
    HttpdnsIpObject *ip4 = [[HttpdnsIpObject alloc] init];
    [ip4 setIp:ipv62];
    hostObject.v6Ips = @[ip3, ip4];
    hostObject.lastIPv6LookupTime = self.currentTimeStamp;
    return hostObject;
}

- (void)presetNetworkEnvAsIpv4 {
    HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
    OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpv4Only);
    OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(NO);

    id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
    OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
}

- (void)presetNetworkEnvAsIpv6 {
    HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
    OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpv6Only);
    OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(YES);

    id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
    OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
}

- (void)presetNetworkEnvAsIpv4AndIpv6 {
    HttpdnsIpStackDetector *mockIpv6Adapter = OCMPartialMock([HttpdnsIpStackDetector sharedInstance]);
    OCMStub([mockIpv6Adapter currentIpStack]).andReturn(kHttpdnsIpDual);
    OCMStub([mockIpv6Adapter isIpv6OnlyNetwork]).andReturn(NO);

    id mockAdapterClass = OCMClassMock([HttpdnsIpStackDetector class]);
    OCMStub([mockAdapterClass sharedInstance]).andReturn(mockIpv6Adapter);
}

- (void)shouldNotHaveCallNetworkRequestWhenResolving:(void (^)(void))resolvingBlock {
    HttpDnsService *httpdns = [HttpDnsService sharedInstance];
    HttpdnsRequestManager *requestManager = httpdns.requestManager;
    HttpdnsRequestManager *mockScheduler = OCMPartialMock(requestManager);
    OCMReject([mockScheduler executeRequest:[OCMArg any] retryCount:0]);
    resolvingBlock();
    OCMVerifyAll(mockScheduler);
}

- (void)shouldHaveCalledRequestWhenResolving:(void (^)(void))resolvingBlock {
    HttpDnsService *httpdns = [HttpDnsService sharedInstance];
    HttpdnsRequestManager *requestManager = httpdns.requestManager;
    HttpdnsRequestManager *mockScheduler = OCMPartialMock(requestManager);
    OCMExpect([mockScheduler executeRequest:[OCMArg any] retryCount:0]).andReturn(nil);
    resolvingBlock();
    OCMVerifyAll(mockScheduler);
}

@end