HawkHelperTests.swift

/* 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 Alamofire
import Foundation
import FxA

import XCTest

class HawkHelperTests: XCTestCase {
    func testSpecSignatureExample() {
        let input = "hawk.1.header\n" +
        "1353832234\n" +
        "j4h3g2\n" +
        "GET\n" +
        "/resource/1?b=1&a=2\n" +
        "example.com\n" +
        "8000\n" +
        "\n" +
        "some-app-ext-data\n"

        let expected = HawkHelper.getSignatureFor(input.utf8EncodedData, key: "werxhqb98rpaxn39848xrunpaw3489ruxnpa98w4rxn".utf8EncodedData)
        XCTAssertEqual("6R4rV5iE+NPoym+WwjeHzjAGXUtLNIxmo1vpMofpLAE=", expected)
    }

    func testSpecRequestString() {
        let timestamp = Int64(1353832234)
        let nonce = "j4h3g2"
        let extra = "some-app-ext-data"

        let req = Alamofire.request(URL(string: "http://example.com:8000/resource/1?b=1&a=2")!)
        let expected = "hawk.1.header\n" +
            "1353832234\n" +
            "j4h3g2\n" +
            "GET\n" +
            "/resource/1?b=1&a=2\n" +
            "example.com\n" +
            "8000\n" +
            "\n" +
            "some-app-ext-data\n"
        XCTAssertEqual(HawkHelper.getRequestStringFor(req.request!, timestampString: String(timestamp), nonce: nonce, hash: "", extra: extra).components(separatedBy: "\n"),
            expected.components(separatedBy: "\n"))
    }

    func testSpecWithoutPayloadExample() {
        let helper = HawkHelper(id: "dh37fgj492je",
            key: "werxhqb98rpaxn39848xrunpaw3489ruxnpa98w4rxn".utf8EncodedData)
        let req = Alamofire.request(URL(string: "http://example.com:8000/resource/1?b=1&a=2")!)
        let timestamp = Int64(1353832234)
        let nonce = "j4h3g2"
        let extra = "some-app-ext-data"
        let value = helper.getAuthorizationValueFor(req.request!, at: timestamp, nonce: nonce, extra: extra)
        let expected = "Hawk id=\"dh37fgj492je\", ts=\"1353832234\", nonce=\"j4h3g2\", ext=\"some-app-ext-data\", mac=\"6R4rV5iE+NPoym+WwjeHzjAGXUtLNIxmo1vpMofpLAE=\""
        XCTAssertEqual(value, expected)
    }

    func testSpecWithPayloadExample() {
        let helper = HawkHelper(id: "dh37fgj492je",
            key: "werxhqb98rpaxn39848xrunpaw3489ruxnpa98w4rxn".utf8EncodedData)
        let body = "Thank you for flying Hawk"

        let req = Alamofire.request("http://example.com:8000/resource/1?b=1&a=2", method: .post, parameters: [:], encoding: URF8BodyEncoding(body: body))
        let timestamp = Int64(1353832234)
        let nonce = "j4h3g2"
        let extra = "some-app-ext-data"
        let value = helper.getAuthorizationValueFor(req.request!, at: timestamp, nonce: nonce, extra: extra)
        let expected = "Hawk id=\"dh37fgj492je\", ts=\"1353832234\", nonce=\"j4h3g2\", hash=\"Yi9LfIIFRtBEPt74PVmbTF/xVAwPn7ub15ePICfgnuY=\", ext=\"some-app-ext-data\", mac=\"aSe1DERmZuRl3pI36/9BdZmnErTw3sNzOOAUlfeKjVw=\""
        XCTAssertEqual(value, expected)
    }

    func testGetBaseContentType() {
        XCTAssertEqual("text/plain", HawkHelper.getBaseContentTypeFor("text/plain"))
        XCTAssertEqual("text/plain", HawkHelper.getBaseContentTypeFor("text/plain;one"))
        XCTAssertEqual("text/plain", HawkHelper.getBaseContentTypeFor("text/plain;one;two"))
        XCTAssertEqual("text/html", HawkHelper.getBaseContentTypeFor("text/html;charset=UTF-8"))
        XCTAssertEqual("text/html", HawkHelper.getBaseContentTypeFor("text/html; charset=UTF-8"))
        XCTAssertEqual("text/html", HawkHelper.getBaseContentTypeFor("text/html ;charset=UTF-8"))
    }
}

struct URF8BodyEncoding: ParameterEncoding {
    var body: String
    func encode(_ urlRequest: URLRequestConvertible, with parameters: Parameters?) throws -> URLRequest {
        var mutableRequest = urlRequest.urlRequest
        mutableRequest?.httpBody = body.utf8EncodedData
        return mutableRequest!
    }
}