TestUnit.swift

// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//

class TestUnit: XCTestCase {

    static var allTests: [(String, (TestUnit) -> () throws -> Void)] {
        return [
            ("test_equality", test_equality),
        ]
    }

    func test_equality() {
        let s1 = "a"
        let s2 = "ab"

        let u1 = Unit(symbol: s1)
        let u2 = Unit(symbol: s1)
        let u3 = Unit(symbol: s2)

        XCTAssertEqual(u1, u2)
        XCTAssertEqual(u2, u1)
        XCTAssertNotEqual(u1, u3)
        XCTAssertNotEqual(u3, u1)

        let uc1 = UnitConverterLinear(coefficient: 1, constant: 2)
        let uc2 = UnitConverterLinear(coefficient: 1, constant: 3)

        let d1 = Dimension(symbol: s1, converter: uc1)
        let d2 = Dimension(symbol: s1, converter: uc1)
        let d3 = Dimension(symbol: s2, converter: uc1)
        let d4 = Dimension(symbol: s1, converter: uc2)

        XCTAssertEqual(d1, d2)
        XCTAssertEqual(d2, d1)
        XCTAssertNotEqual(d1, d3)
        XCTAssertNotEqual(d3, d1)
        XCTAssertNotEqual(d1, d4)
        XCTAssertNotEqual(d4, d1)

        XCTAssertEqual(u1, d1)
        XCTAssertNotEqual(d1, u1)

        func testEquality<T: Dimension>(ofDimensionSubclass: T.Type) {
            let u0 = Unit(symbol: s1)
            let d1 = Dimension(symbol: s1, converter: uc1)

            let u1 = T(symbol: s1, converter: uc1)
            let u2 = T(symbol: s1, converter: uc1)
            let u3 = T(symbol: s1, converter: uc2)
            let u4 = T(symbol: s2, converter: uc1)

            XCTAssertEqual(u1, u2)
            XCTAssertEqual(u2, u1)
            XCTAssertNotEqual(u1, u3)
            XCTAssertNotEqual(u3, u1)
            XCTAssertNotEqual(u1, u4)
            XCTAssertNotEqual(u4, u1)

            XCTAssertEqual(u0, u1)
            XCTAssertNotEqual(u1, u0)

            XCTAssertEqual(d1, u1)
            XCTAssertNotEqual(u1, d1)
        }

        testEquality(ofDimensionSubclass: UnitAcceleration.self)
        testEquality(ofDimensionSubclass: UnitAngle.self)
        testEquality(ofDimensionSubclass: UnitArea.self)
        testEquality(ofDimensionSubclass: UnitConcentrationMass.self)
        testEquality(ofDimensionSubclass: UnitDispersion.self)
        testEquality(ofDimensionSubclass: UnitDuration.self)
        testEquality(ofDimensionSubclass: UnitElectricCharge.self)
        testEquality(ofDimensionSubclass: UnitElectricCurrent.self)
        testEquality(ofDimensionSubclass: UnitElectricPotentialDifference.self)
        testEquality(ofDimensionSubclass: UnitElectricResistance.self)
        testEquality(ofDimensionSubclass: UnitEnergy.self)
        testEquality(ofDimensionSubclass: UnitFrequency.self)
        testEquality(ofDimensionSubclass: UnitFuelEfficiency.self)
        testEquality(ofDimensionSubclass: UnitIlluminance.self)
        testEquality(ofDimensionSubclass: UnitLength.self)
        testEquality(ofDimensionSubclass: UnitMass.self)
        testEquality(ofDimensionSubclass: UnitPower.self)
        testEquality(ofDimensionSubclass: UnitPressure.self)
        testEquality(ofDimensionSubclass: UnitSpeed.self)
        testEquality(ofDimensionSubclass: UnitTemperature.self)
        testEquality(ofDimensionSubclass: UnitVolume.self)
    }

}

class TestDimension: XCTestCase {
    static var allTests: [(String, (TestDimension) -> () throws -> Void)] {
        return [
            ("test_encodeDecode", test_encodeDecode),
        ]
    }

    func test_encodeDecode() {
        let original = Dimension(symbol: "symbol", converter: UnitConverterLinear(coefficient: 1.0))

        let encodedData = NSMutableData()
        let archiver = NSKeyedArchiver(forWritingWith: encodedData)
        original.encode(with: archiver)
        archiver.finishEncoding()

        let unarchiver = NSKeyedUnarchiver(forReadingWith: encodedData as Data)
        let decoded = Dimension(coder: unarchiver)

        XCTAssertNotNil(decoded)
        XCTAssertEqual(original, decoded)
    }
}