//
//  SecEncodeTransformTests.swift
//  Base32
//
//  Created by 野村 憲男 on 1/25/15.
//  Copyright (c) 2015 Norio Nomura. All rights reserved.
//

import Foundation
import XCTest
import Security
import Base32

let vectors: [(String, String, String)] = [
    ("", "", ""),
    ("f", "MY======", "CO======"),
    ("fo", "MZXQ====", "CPNG===="),
    ("foo", "MZXW6===", "CPNMU==="),
    ("foob", "MZXW6YQ=", "CPNMUOG="),
    ("fooba", "MZXW6YTB", "CPNMUOJ1"),
    ("foobar", "MZXW6YTBOI======", "CPNMUOJ1E8======"),
]

let convertedVectors = vectors.map {($0.dataUsingUTF8StringEncoding, $1, $2)}

class SecEncodeTransformTests: XCTestCase {
    
    
    override func setUp() {
        super.setUp()
        // Put setup code here. This method is called before the invocation of each test method in the class.
    }
    
    override func tearDown() {
        // Put teardown code here. This method is called after the invocation of each test method in the class.
        super.tearDown()
    }
    
    // MARK: https://tools.ietf.org/html/rfc4648

    // MARK: Using SecEncodeTransform
    func test_RFC4648_Encode_UsingSecEncodeTransform() {
        var results = Array<String>(count: count(convertedVectors), repeatedValue: "")
        let vectorsAndIndices = Zip2(convertedVectors, indices(results))
        self.measureBlock{
            for _ in 0...100 {
                for ((test, _, _), index) in vectorsAndIndices {
                    results[index] = TTTBase32EncodedStringFromData(test)
                }
            }
        }
        for ((test, expect, _), result) in Zip2(convertedVectors, results) {
            XCTAssertEqual(result, expect, "TTTBase32EncodedStringFromData for \(test)")
        }
    }
    
    func test_RFC4648_Decode_UsingSecEncodeTransform() {
        var results = Array<NSData>(count: count(convertedVectors), repeatedValue: NSData())
        let vectorsAndIndices = Zip2(convertedVectors, indices(results))
        self.measureBlock{
            for _ in 0...100 {
                for ((_, test, _), index) in vectorsAndIndices {
                    results[index] = TTTDataFromBase32EncodedString(test)
                }
            }
        }
        for ((expect, test, _), result) in Zip2(convertedVectors, results) {
            XCTAssertEqual(result, expect, "TTTDataFromBase32EncodedString for \(test)")
        }
    }
    
    // MARK: Using Base32
    func test_RFC4648_Encode_UsingBase32() {
        var results = Array<String>(count: count(convertedVectors), repeatedValue: "")
        let vectorsAndIndices = Zip2(convertedVectors, indices(results))
        self.measureBlock{
            for _ in 0...100 {
                for ((test, _, _), index) in vectorsAndIndices {
                    results[index] = base32Encode(test)
                }
            }
        }
        for ((test, expect, _), result) in Zip2(convertedVectors, results) {
            XCTAssertEqual(result, expect, "base32Encode for \(test)")
        }
    }
    
    func test_RFC4648_Decode_UsingBase32() {
        var results = Array<NSData>(count: count(convertedVectors), repeatedValue: NSData())
        let vectorsAndIndices = Zip2(convertedVectors, indices(results))
        self.measureBlock{
            for _ in 0...100 {
                for ((_, test, _), index) in vectorsAndIndices {
                    results[index] = base32DecodeToData(test)!
                }
            }
        }
        for ((expect, test, _), result) in Zip2(convertedVectors, results) {
            XCTAssertEqual(result, expect, "base32Decode for \(test)")
        }
    }
}