stripe-ios/StripeIdentity/StripeIdentityTests/Helpers/IdentityMockData.swift

//
//  IdentityMockData.swift
//  StripeIdentityTests
//
//  Created by Mel Ludowise on 10/27/21.
//  Copyright © 2021 Stripe, Inc. All rights reserved.
//

import Foundation
import StripeCoreTestUtils
import UIKit

@testable import StripeIdentity

// Dummy class to determine this bundle
private class ClassForBundle {}

enum VerificationPageMock: String, MockData {
    typealias ResponseType = StripeAPI.VerificationPage
    var bundle: Bundle { return Bundle(for: ClassForBundle.self) }

    case response200 = "VerificationPage_200"
    case response200TestMode = "VerificationPage_200_testMode"
    case response200Submitted = "VerificationPage_200_submitted"
    case requireLiveCapture = "VerificationPage_require_live_capture"
    case noSelfie = "VerificationPage_no_selfie"
    case typeDocumentRequireIdNumber = "VerificationPage_type_doc_require_idNumber"
    case typeDocumentRequireAddress = "VerificationPage_type_doc_require_address"
    case typeDocumentRequireIdNumberAndAddress = "VerificationPage_type_doc_require_idNumber_and_address"
    case typeIdNumber = "VerificationPage_type_idNumber"
    case typeAddress = "VerificationPage_type_address"
}

enum VerificationPageDataMock: String, MockData {
    typealias ResponseType = StripeAPI.VerificationPageData
    var bundle: Bundle { return Bundle(for: ClassForBundle.self) }

    case response200 = "VerificationPageData_200"
    case noErrors = "VerificationPageData_no_errors"
    case noErrorsNeedback = "VerificationPageData_no_errors_needback"
    case submitted = "VerificationPageData_submitted"

    static func noErrorsWithMissings(
        with missingRequirements: Set<StripeAPI.VerificationPageFieldType>
    ) throws -> ResponseType {
        let noErrorsResponse = try noErrors.make()
        return .init(
            id: noErrorsResponse.id,
            requirements: .init(
                errors: noErrorsResponse.requirements.errors,
                missing: missingRequirements
            ),
            status: noErrorsResponse.status,
            submitted: noErrorsResponse.submitted
        )
    }
}

enum CapturedImageMock: String {
    var bundle: Bundle { return Bundle(for: ClassForBundle.self) }

    case frontDriversLicense = "front_drivers_license"
    case backDriversLicense = "back_drivers_license"

    var url: URL {
        return bundle.url(forResource: rawValue, withExtension: "jpg")!
    }

    var image: UIImage {
        return UIImage(named: "\(rawValue).jpg", in: bundle, compatibleWith: nil)!
    }
}

enum VerificationPageDataUpdateMock {
    static let `default` = StripeAPI.VerificationPageDataUpdate(
        clearData: nil,
        collectedData: .init(
            biometricConsent: false,
            face: .init(
                bestHighResImage: "best_high_res_id",
                bestLowResImage: "best_low_res_id",
                firstHighResImage: "first_high_res_id",
                firstLowResImage: "first_low_res_id",
                lastHighResImage: "last_high_res_id",
                lastLowResImage: "last_low_res_id",
                bestFaceScore: .init(0.9),
                faceScoreVariance: .init(0.2),
                numFrames: 8,
                bestBrightnessValue: nil,
                bestCameraLensModel: nil,
                bestExposureDuration: nil,
                bestExposureIso: nil,
                bestFocalLength: nil,
                bestIsVirtualCamera: nil,
                trainingConsent: true
            ),
            idDocumentBack: .init(
                backScore: .init(1),
                brightnessValue: nil,
                cameraLensModel: nil,
                exposureDuration: nil,
                exposureIso: nil,
                focalLength: nil,
                frontCardScore: .init(0),
                highResImage: "back_user_upload_id",
                invalidScore: .init(0),
                iosBarcodeDecoded: nil,
                iosBarcodeSymbology: nil,
                iosTimeToFindBarcode: nil,
                isVirtualCamera: nil,
                lowResImage: "back_full_frame_id",
                passportScore: .init(0),
                uploadMethod: .autoCapture
            ),
            idDocumentFront: .init(
                backScore: .init(0),
                brightnessValue: nil,
                cameraLensModel: nil,
                exposureDuration: nil,
                exposureIso: nil,
                focalLength: nil,
                frontCardScore: .init(1),
                highResImage: "front_user_upload_id",
                invalidScore: .init(0),
                iosBarcodeDecoded: nil,
                iosBarcodeSymbology: nil,
                iosTimeToFindBarcode: nil,
                isVirtualCamera: nil,
                lowResImage: "front_full_frame_id",
                passportScore: .init(0),
                uploadMethod: .autoCapture
            ),
            idDocumentType: .drivingLicense
        )
    )

    static let frontOnly = StripeAPI.VerificationPageDataUpdate(
        clearData: nil,
        collectedData: .init(
            biometricConsent: false,
            face: nil,
            idDocumentBack: nil,
            idDocumentFront: .init(
                backScore: .init(0),
                brightnessValue: nil,
                cameraLensModel: nil,
                exposureDuration: nil,
                exposureIso: nil,
                focalLength: nil,
                frontCardScore: .init(1),
                highResImage: "front_user_upload_id",
                invalidScore: .init(0),
                iosBarcodeDecoded: nil,
                iosBarcodeSymbology: nil,
                iosTimeToFindBarcode: nil,
                isVirtualCamera: nil,
                lowResImage: "front_full_frame_id",
                passportScore: .init(0),
                uploadMethod: .autoCapture
            ),
            idDocumentType: .drivingLicense
        )
    )
}