//
// Copyright 2010-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License").
// You may not use this file except in compliance with the License.
// A copy of the License is located at
//
// http://aws.amazon.com/apache2.0
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language governing
// permissions and limitations under the License.
//

#import <UIKit/UIKit.h>
#import <XCTest/XCTest.h>
#import "AWSCore.h"

@interface AWSLogger()

- (NSString *)logLevelLabel:(AWSLogLevel)logLevel;

@end

@interface AWSUtilityTests : XCTestCase

@end

@implementation AWSUtilityTests

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

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

- (void)testCategoryNSString {
    NSString *inputString = @"U29tZSBpbnB1dCBzdHJpbmc="; // Base64 encoded "Some input string"
    XCTAssertTrue([inputString aws_isBase64Data]);

    inputString = @"Some input string";
    XCTAssertFalse([inputString aws_isBase64Data]);

    inputString = nil;
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"ab";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"0123456789012345678901234567890123456789012345678901234567890123456789";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"0123456789-";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"0123_456789";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"0123-.456789";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"0123.-456789";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);

    inputString = @"abcDefg";
    XCTAssertFalse([inputString aws_isVirtualHostedStyleCompliant]);
}

- (void)testCategoryNSJSONSerialization {
    NSError *error = nil;
    NSData *data = [NSJSONSerialization aws_dataWithJSONObject:@"test"
                                                       options:kNilOptions
                                                         error:&error];
    NSString *outputString = [[NSString alloc] initWithData:data
                                                   encoding:NSUTF8StringEncoding];
    XCTAssertEqualObjects(outputString, @"\"test\"");

    data = [NSJSONSerialization aws_dataWithJSONObject:nil
                                               options:kNilOptions
                                                 error:&error];
    XCTAssertNil(data);
}

- (void)testCategoryNSDictionaryRemoveNullValues {
    NSDictionary *testDictionary = @{
                                     @"key1":@"value1",
                                     @"key2":[NSNull null],
                                     @"key3":@{
                                             @"subkey1":@"subkeyvalue1",
                                             @"subkey2":[NSNull null]
                                             },
                                     @"key4":@{
                                             @"onlyKey":[NSNull null]
                                             }
                                     };
    NSDictionary *expectedResultDic = @{
                                        @"key1":@"value1",
                                        @"key3":@{
                                                @"subkey1":@"subkeyvalue1",
                                                },
                                        @"key4":@{
                                                }
                                        };
    NSDictionary *resultDic = [testDictionary aws_removeNullValues];
    XCTAssertEqualObjects(expectedResultDic, resultDic);
    
    
}

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wdeprecated-declarations"

- (void)testLogger {
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelUnknown], @"?");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelNone], @"?");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelError], @"Error");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelWarn], @"Warn");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelInfo], @"Info");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelDebug], @"Debug");
    XCTAssertEqualObjects([[AWSLogger defaultLogger] logLevelLabel:AWSLogLevelVerbose], @"Verbose");
}

#pragma clang diagnostic pop

@end