//
//  PGWhitelistTests.m
//  PhoneGapLib
//
//  Created by Shazron Abdullah on 11-09-01.
//  Copyright 2011 Nitobi Software Inc. All rights reserved.
//

#import "PGWhitelistTests.h"
#import "PGWhitelist.h"

@implementation PGWhitelistTests

- (void)setUp
{
    [super setUp];
    
    // setup code here
}

- (void)tearDown
{
    // Tear-down code here.

    [super tearDown];
}

- (void) testAllowedSchemes
{
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"*.phonegap.com",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist schemeIsAllowed:@"http"], nil);
    STAssertTrue([whitelist schemeIsAllowed:@"https"], nil);
    STAssertTrue([whitelist schemeIsAllowed:@"ftp"], nil);
    STAssertTrue([whitelist schemeIsAllowed:@"ftps"], nil);
    STAssertFalse([whitelist schemeIsAllowed:@"gopher"], nil);
    
    [whitelist release];

}

- (void) testSubdomainWildcard
{
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                     @"*.phonegap.com",
                     nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://build.phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://sub1.sub0.build.phonegap.com"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com.ca"]], nil);
    
    [whitelist release];
}

- (void) testWildcardInTLD
{
    // NOTE: if the user chooses to do this (a wildcard in the TLD, not a wildcard as the TLD), we allow it because we assume they know what they are doing! We don't replace it with known TLDs
    // This might be applicable for custom TLDs on a local network DNS
    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"phonegap.c*m",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.corporateroom"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.foo"]], nil);
    
    [whitelist release];
}

- (void) testTLDWildcard
{    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"phonegap.*",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    NSString* hostname = @"phonegap";
    
    NSArray* knownTLDs = [NSArray arrayWithObjects:
                          @"aero", @"asia", @"arpa", @"biz", @"cat",
                          @"com", @"coop", @"edu", @"gov", @"info",
                          @"int", @"jobs", @"mil", @"mobi", @"museum",
                          @"name", @"net", @"org", @"pro", @"tel",
                          @"travel", @"xxx",
                          nil];
    
    // 26*26 combos
    NSMutableArray* twoCharCountryCodes = [NSMutableArray arrayWithCapacity:(26*26)];
    for (char c0 = 'a'; c0 <= 'z'; ++c0)
    {
        for (char c1 = 'a'; c1 <= 'z'; ++c1)
        {
            [twoCharCountryCodes addObject:[NSString stringWithFormat:@"%c%c", c0, c1]];
        }
    }

    NSMutableArray* shouldPass = [NSMutableArray arrayWithCapacity:[knownTLDs count]+[twoCharCountryCodes count]];

    NSEnumerator* knownTLDEnumerator = [knownTLDs objectEnumerator];
    NSString* tld = nil;
    
    while (tld = [knownTLDEnumerator nextObject])
    {
        [shouldPass addObject:[NSURL URLWithString:[NSString stringWithFormat:@"http://%@.%@", hostname, tld]]];
    }

    NSEnumerator* twoCharCountryCodesEnumerator = [twoCharCountryCodes objectEnumerator];
    NSString* cc = nil;
    
    while (cc = [twoCharCountryCodesEnumerator nextObject])
    {
        [shouldPass addObject:[NSURL URLWithString:[NSString stringWithFormat:@"http://%@.%@", hostname, cc]]];
    }
    
    NSEnumerator* shouldPassEnumerator = [shouldPass objectEnumerator];
    NSURL* url = nil;

    while (url = [shouldPassEnumerator nextObject])
    {
        STAssertTrue([whitelist URLIsAllowed:url], @"Url tested :%@", [url description]);
    }
    
    STAssertFalse(([whitelist URLIsAllowed:[NSURL URLWithString:[NSString stringWithFormat:@"http://%@.%@", hostname, @"faketld"]]]), nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://unknownhostname.faketld"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://unknownhostname.com"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.phonegap.com"]], nil);
    
    [whitelist release];
}

- (void) testCatchallWildcardOnly
{    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"*",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://build.phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://MyDangerousSite.org"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com.SuspiciousSite.com"]], nil);
    
    [whitelist release];
}

- (void) testWildcardInHostname
{    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"www.*phone*gap.com",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
        
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.phoneMACgap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.MACphonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.MACphoneMACgap.com"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com"]], nil);
    
    [whitelist release];
}

- (void) testExactMatch
{    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"www.phonegap.com",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.phonegap.com"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://build.phonegap.com"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com"]], nil);
    
    [whitelist release];
}

- (void) testWildcardMix
{    
    NSArray* allowedHosts = [NSArray arrayWithObjects: 
                             @"*.phone*gap.*",
                             nil];
    
    PGWhitelist* whitelist = [[PGWhitelist alloc] initWithArray:allowedHosts];
    
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://www.phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phonegap.com"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phoneMACgap.ca"]], nil);
    STAssertTrue([whitelist URLIsAllowed:[NSURL URLWithString:@"http://phoneMACgap.museum"]], nil);
    STAssertFalse([whitelist URLIsAllowed:[NSURL URLWithString:@"http://blahMACgap.museum"]], nil);
    
    [whitelist release];
}




@end