/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

@testable import Account
import FxA
import Shared
import SwiftyJSON

import XCTest

class FxAStateTests: XCTestCase {
    class func stateForLabel(_ label: FxAStateLabel) -> FxAState {
        let keyLength = UInt(KeyLength) // Ah, Swift.
        let now = Date.now()

        switch label {
        case .engagedBeforeVerified:
            return EngagedBeforeVerifiedState(
                knownUnverifiedAt: now + 1, lastNotifiedUserAt: now + 2,
                sessionToken: Data.randomOfLength(keyLength)!,
                keyFetchToken: Data.randomOfLength(keyLength)!,
                unwrapkB: Data.randomOfLength(keyLength)!)

        case .engagedAfterVerified:
            return EngagedAfterVerifiedState(
                sessionToken: Data.randomOfLength(keyLength)!,
                keyFetchToken: Data.randomOfLength(keyLength)!,
                unwrapkB: Data.randomOfLength(keyLength)!)

        case .cohabitingBeforeKeyPair:
            return CohabitingBeforeKeyPairState(sessionToken: Data.randomOfLength(keyLength)!,
                kA: Data.randomOfLength(keyLength)!, kB: Data.randomOfLength(keyLength)!)

        case .cohabitingAfterKeyPair:
            let keyPair = RSAKeyPair.generate(withModulusSize: 512)!
            return CohabitingAfterKeyPairState(sessionToken: Data.randomOfLength(keyLength)!,
                kA: Data.randomOfLength(keyLength)!, kB: Data.randomOfLength(keyLength)!,
                keyPair: keyPair, keyPairExpiresAt: now + 1)

        case .married:
            let keyPair = RSAKeyPair.generate(withModulusSize: 512)!
            return MarriedState(sessionToken: Data.randomOfLength(keyLength)!,
                kA: Data.randomOfLength(keyLength)!, kB: Data.randomOfLength(keyLength)!,
                keyPair: keyPair, keyPairExpiresAt: now + 1,
                certificate: "certificate", certificateExpiresAt: now + 2)

        case .separated:
            return SeparatedState()

        case .doghouse:
            return DoghouseState()
        }
    }

    func testSerialization() {
        // Journal of Negative Results: make sure we aren't *always* succeeding.
        // This Married state will have an earlier timestamp than the one generated after the loop.
        let state1 = FxAStateTests.stateForLabel(.married) as! MarriedState

        for stateLabel in FxAStateLabel.allValues {
            let stateFromLabel = FxAStateTests.stateForLabel(stateLabel)
            let d = stateFromLabel.asJSON()
            if let e = state(fromJSON:d)?.asJSON() {
                // We can't compare arbitrary Swift Dictionary instances directly, but the following appears to work.
                XCTAssertEqual(
                    NSDictionary(dictionary: d.dictionaryObject!),
                    NSDictionary(dictionary: e.dictionaryObject!))
            } else {
                XCTFail("Expected to create state.")
            }
        }

        // This Married state will have a later timestamp than the one generated before the loop.
        let state2 = FxAStateTests.stateForLabel(.married) as! MarriedState
        // We can't compare arbitrary Swift Dictionary instances directly, but the following appears to work.
        XCTAssertNotEqual(
            NSDictionary(dictionary: state1.asJSON().dictionaryObject!),
            NSDictionary(dictionary: state2.asJSON().dictionaryObject!))
    }
}