AutoCat/AutoCatTests/Extensions/VehicleDto+Presets.swift

//
//  VehicleDto+Presets.swift
//  AutoCatTests
//
//  Created by Selim Mustafaev on 13.07.2024.
//  Copyright © 2024 Selim Mustafaev. All rights reserved.
//

import AutoCatCore

extension VehicleDto {
    
    static let validNumber: String = "А123АА761"
    static let validNumber2: String = "А456АА761"
    
    static var normal: VehicleDto {
        var vehicle = VehicleDto()
        vehicle.number = validNumber
        vehicle.brand = VehicleBrandDto()
        return vehicle
    }
    
    static var normal2: VehicleDto {
        var vehicle = VehicleDto()
        vehicle.number = validNumber2
        vehicle.brand = VehicleBrandDto()
        return vehicle
    }
    
    static var unrecognizedVehicle: VehicleDto {
        var vehicle = VehicleDto()
        vehicle.number = validNumber
        return vehicle
    }
    
    static var outdatedVehicle: VehicleDto {
        var vehicle = VehicleDto()
        vehicle.number = validNumber
        vehicle.currentNumber = validNumber2
        return vehicle
    }
    
    static var needSyncVehicle: VehicleDto {
        var vehicle = VehicleDto()
        vehicle.brand = VehicleBrandDto()
        vehicle.number = validNumber
        vehicle.synchronized = false
        return vehicle
    }
}

// Fluent

extension VehicleDto {
    
    func addNote(text: String) -> Self {
        
        var vehicle = self
        vehicle.notes.append(VehicleNoteDto(text: text, user: ""))
        return vehicle
    }
    
    func addNote(text: String, id: String) -> Self {
        
        var note = VehicleNoteDto(text: text, user: "")
        note.id = id
        
        var vehicle = self
        vehicle.notes.append(note)
        return vehicle
    }
    
    func addEvent(_ event: VehicleEventDto) -> Self {
        
        var vehicle = self
        vehicle.events.append(event)
        return vehicle
    }
}