#import <XCTest/XCTest.h>
#import "DnsManager.h"
#import "TestUtil.h"
#import "Util.h"
#import "IpAddress.h"
#import "ThreadManager.h"
#import "CancelTokenSource.h"
#import <netdb.h>

#define infrastructureTestHostName @"relay.whispersystems.org"
#define reliableHostName @"example.com"
#define invalidHostname @"∆©˙∆¨¥©©˜¨¥©˜†¥µ¬¬¨˙µ†¥∫®∂®†"
#define nonExistentHostname [NSString stringWithFormat:@"%@kfurmtludehntlgihmvnduyebntiinvbudydepqowudyfnrkt.com", \
    [[CryptoTools generateSecureRandomData:10] encodedAsBase64]]

@interface DnsManagerTest : XCTestCase

@end

@implementation DnsManagerTest

-(void) testQueryAddresses_Sequential {
    Future* f1 = [DnsManager asyncQueryAddressesForDomainName:reliableHostName
                                              unlessCancelled:nil];
    testChurnUntil([f1 hasSucceeded], 5.0);
    test([(NSArray*)[f1 forceGetResult] count] > 0);

    Future* f2 = [DnsManager asyncQueryAddressesForDomainName:invalidHostname
                                              unlessCancelled:nil];
    testChurnUntil([f2 hasFailed], 5.0);

    Future* f3 = [DnsManager asyncQueryAddressesForDomainName:nonExistentHostname
                                              unlessCancelled:nil];
    testChurnUntil([f3 hasFailed], 5.0);

    Future* f4 = [DnsManager asyncQueryAddressesForDomainName:infrastructureTestHostName
                                              unlessCancelled:nil];
    testChurnUntil([f4 hasSucceeded], 5.0);
    test([f4 hasSucceeded] && [(NSArray*)[f4 forceGetResult] count] > 0);
    
}

-(void) testQueryAddresses_Concurrent {
    Future* f1 = [DnsManager asyncQueryAddressesForDomainName:reliableHostName
                                                 unlessCancelled:nil];
    Future* f2 = [DnsManager asyncQueryAddressesForDomainName:invalidHostname
                                              unlessCancelled:nil];
    Future* f3 = [DnsManager asyncQueryAddressesForDomainName:nonExistentHostname
                                              unlessCancelled:nil];
    Future* f4 = [DnsManager asyncQueryAddressesForDomainName:infrastructureTestHostName
                                              unlessCancelled:nil];
    
    testChurnUntil([f1 hasSucceeded] && [f2 hasFailed] && [f3 hasFailed] && [f4 hasSucceeded], 5.0);
    test([f1 hasSucceeded] && [(NSArray*)[f1 forceGetResult] count] > 0);
    test([f4 hasSucceeded] && [(NSArray*)[f4 forceGetResult] count] > 0);
}

-(void) testQueryAddresses_Cancel {
    CancelTokenSource* c = [CancelTokenSource cancelTokenSource];
    Future* f1 = [DnsManager asyncQueryAddressesForDomainName:reliableHostName
                                              unlessCancelled:[c getToken]];
    Future* f2 = [DnsManager asyncQueryAddressesForDomainName:invalidHostname
                                              unlessCancelled:[c getToken]];
    Future* f3 = [DnsManager asyncQueryAddressesForDomainName:nonExistentHostname
                                              unlessCancelled:[c getToken]];
    Future* f4 = [DnsManager asyncQueryAddressesForDomainName:infrastructureTestHostName
                                              unlessCancelled:[c getToken]];
    [c cancel];
    
    testChurnUntil(![f1 isIncomplete] && [f2 hasFailed] && [f3 hasFailed] && ![f4 isIncomplete], 5.0);
    test([f1 hasSucceeded] || [[f1 forceGetFailure] conformsToProtocol:@protocol(CancelToken)]);
    test([f2 hasFailed]);
    test([f3 hasFailed]);
    test([f4 hasSucceeded] || [[f4 forceGetFailure] conformsToProtocol:@protocol(CancelToken)]);
}

-(void)testQueryAddresses_FastCancel {
    CancelTokenSource* c = [CancelTokenSource cancelTokenSource];
    Future* f = [DnsManager asyncQueryAddressesForDomainName:reliableHostName
                                             unlessCancelled:[c getToken]];
    [c cancel];
    test(![f isIncomplete]);
}

@end