Signal-iOS/SignalServiceKit/tests/Storage/TestModelTests.swift

//
//  Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//

import Foundation
import XCTest
@testable import SignalServiceKit

class TestModelTests: SSKBaseTestSwift {

    func testTestModel() {

        let writeModel = TestModel()

        self.write { transaction in
            writeModel.anyInsert(transaction: transaction)

            let readModel = TestModel.anyFetch(uniqueId: writeModel.uniqueId, transaction: transaction)!
            XCTAssertFalse(writeModel === readModel)
            XCTAssertEqual(0, readModel.doubleValue)
            XCTAssertEqual(0, readModel.floatValue)
            XCTAssertEqual(0, readModel.uint64Value)
            XCTAssertEqual(0, readModel.int64Value)
            XCTAssertEqual(0, readModel.nsuIntegerValue)
            XCTAssertEqual(0, readModel.nsIntegerValue)
            XCTAssertNil(readModel.nsNumberValueUsingInt64)
            XCTAssertNil(readModel.nsNumberValueUsingUInt64)
            XCTAssertNil(readModel.dateValue)
        }

        // MARK: - Happy path

        self.write { transaction in

            let now = Date(timeIntervalSince1970: 123456)
            let nsNumberUsingInt64: Int64 = -1
            let nsNumberUsingUInt64: UInt64 = 1

            writeModel.anyUpdate(transaction: transaction) { model in
                model.doubleValue = 1
                model.floatValue = 1
                model.uint64Value = 1
                model.int64Value = 1
                model.nsuIntegerValue = 1
                model.nsIntegerValue = 1
                model.nsNumberValueUsingInt64 = NSNumber(value: nsNumberUsingInt64)
                model.nsNumberValueUsingUInt64 = NSNumber(value: nsNumberUsingUInt64)
                model.dateValue = now
            }

            let readModel = TestModel.anyFetch(uniqueId: writeModel.uniqueId, transaction: transaction)!
            XCTAssertFalse(writeModel === readModel)
            XCTAssertEqual(1, readModel.doubleValue)
            XCTAssertEqual(1, readModel.floatValue)
            XCTAssertEqual(1, readModel.uint64Value)
            XCTAssertEqual(1, readModel.int64Value)
            XCTAssertEqual(1, readModel.nsuIntegerValue)
            XCTAssertEqual(1, readModel.nsIntegerValue)
            XCTAssertEqual(NSNumber(value: nsNumberUsingInt64), readModel.nsNumberValueUsingInt64)
            XCTAssertEqual(NSNumber(value: nsNumberUsingUInt64), readModel.nsNumberValueUsingUInt64)
            XCTAssertEqual(now, readModel.dateValue)
        }

        // MARK: - Max values

        self.write { transaction in

            // SQLite doesn't support numeric values larger than Int64.max;
            // UInt64.max for example is not valid.
            let uint64MaxUnsafe: UInt64 = UInt64.max
            let uint64MaxSafe: UInt64 = UInt64(Int64.max)
            let uint64Max: UInt64 = uint64MaxSafe
            XCTAssertLessThanOrEqual(uint64MaxSafe, uint64MaxUnsafe)

            let int64Max: Int64 = Int64.max

            let nsUintMaxUnsafe: UInt = NSUIntegerMaxValue()
            let nsUintMaxSafe: UInt = NSUIntegerMaxValue() < Int64.max ? NSUIntegerMaxValue() : UInt(Int64.max)
            let nsUintMax: UInt = nsUintMaxSafe
            XCTAssertLessThanOrEqual(nsUintMaxSafe, nsUintMaxUnsafe)

            let nsIntMax: Int = NSIntegerMax
            let nsNumberUsingInt64 = int64Max

            let nsNumberUsingUInt64Unsafe = uint64MaxUnsafe
            let nsNumberUsingUInt64Safe = uint64MaxSafe
            let nsNumberUsingUInt64 = nsNumberUsingUInt64Safe
            XCTAssertLessThanOrEqual(nsNumberUsingUInt64Safe, nsNumberUsingUInt64Unsafe)

            let dateMax = Date(timeIntervalSince1970: Double.greatestFiniteMagnitude)

            writeModel.anyUpdate(transaction: transaction) { model in
                model.doubleValue = Double.greatestFiniteMagnitude
                model.floatValue = Float.greatestFiniteMagnitude
                model.uint64Value = uint64Max
                model.int64Value = int64Max
                model.nsuIntegerValue = nsUintMax
                model.nsIntegerValue = nsIntMax
                model.nsNumberValueUsingInt64 = NSNumber(value: nsNumberUsingInt64)
                model.nsNumberValueUsingUInt64 = NSNumber(value: nsNumberUsingUInt64)
                model.dateValue = dateMax
            }

            let readModel = TestModel.anyFetch(uniqueId: writeModel.uniqueId, transaction: transaction)!
            XCTAssertFalse(writeModel === readModel)
            XCTAssertEqual(Double.greatestFiniteMagnitude, readModel.doubleValue)
            XCTAssertEqual(Float.greatestFiniteMagnitude, readModel.floatValue)
            XCTAssertEqual(uint64Max, readModel.uint64Value)
            XCTAssertEqual(int64Max, readModel.int64Value)
            XCTAssertEqual(nsUintMax, readModel.nsuIntegerValue)
            XCTAssertEqual(nsIntMax, readModel.nsIntegerValue)
            XCTAssertEqual(NSNumber(value: nsNumberUsingInt64), readModel.nsNumberValueUsingInt64)
            XCTAssertEqual(NSNumber(value: nsNumberUsingUInt64), readModel.nsNumberValueUsingUInt64)
            XCTAssertEqual(dateMax, readModel.dateValue)
        }

        // MARK: - Min values

        self.write { transaction in

            let dateMin = Date(timeIntervalSince1970: Double.leastNormalMagnitude)

            writeModel.anyUpdate(transaction: transaction) { model in
                model.doubleValue = Double.leastNormalMagnitude
                model.floatValue = Float.leastNormalMagnitude
                model.uint64Value = UInt64.min
                model.int64Value = Int64.min
                model.nsuIntegerValue = UInt.min
                model.nsIntegerValue = Int.min
                model.dateValue = dateMin
            }

            let readModel = TestModel.anyFetch(uniqueId: writeModel.uniqueId, transaction: transaction)!
            XCTAssertFalse(writeModel === readModel)
            XCTAssertEqual(Double.leastNormalMagnitude, readModel.doubleValue)
            XCTAssertEqual(Float.leastNormalMagnitude, readModel.floatValue)
            XCTAssertEqual(UInt64.min, readModel.uint64Value)
            XCTAssertEqual(Int64.min, readModel.int64Value)
            XCTAssertEqual(UInt.min, readModel.nsuIntegerValue)
            XCTAssertEqual(Int.min, readModel.nsIntegerValue)
            XCTAssertEqual(dateMin, readModel.dateValue)
        }

        // MARK: - Date rounding/precision

        self.write { transaction in

            let interval: TimeInterval = 123456.123456
            let dateValue = Date(timeIntervalSince1970: interval)

            writeModel.anyUpdate(transaction: transaction) { model in
                model.dateValue = dateValue
            }

            let readModel = TestModel.anyFetch(uniqueId: writeModel.uniqueId, transaction: transaction)!
            XCTAssertFalse(writeModel === readModel)
            // This date value will NOT be truncated during the roundtrip.
            // NSDate is backed by NSTimeInterval (seconds as double) and we
            // serialiuze using TimeInterval/timeIntervalSince1970.
            XCTAssertEqual(dateValue, readModel.dateValue)
        }
    }
}