65 lines
1.4 KiB
Swift
65 lines
1.4 KiB
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
|
||
}
|
||
}
|
||
|
||
// 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
|
||
}
|
||
}
|