85 lines
2.2 KiB
Swift
85 lines
2.2 KiB
Swift
import App
|
|
import CSVParser
|
|
import DatabaseClient
|
|
import Dependencies
|
|
import Fluent
|
|
import FluentSQLiteDriver
|
|
import Foundation
|
|
import ManualDCore
|
|
import NIO
|
|
import Vapor
|
|
|
|
// Helper to create an in-memory database used for testing.
|
|
func withDatabase(
|
|
setupDependencies: (inout DependencyValues) -> Void = { _ in },
|
|
operation: () async throws -> Void
|
|
) async throws {
|
|
let app = try await Application.make(.testing)
|
|
do {
|
|
try await configure(app, in: .live())
|
|
let database = app.db
|
|
try await app.autoMigrate()
|
|
|
|
try await withDependencies {
|
|
$0.uuid = .incrementing
|
|
$0.date = .init { Date() }
|
|
$0.database = .live(database: database)
|
|
setupDependencies(&$0)
|
|
} operation: {
|
|
try await operation()
|
|
}
|
|
|
|
try await app.autoRevert()
|
|
try await app.asyncShutdown()
|
|
|
|
} catch {
|
|
try? await app.autoRevert()
|
|
try await app.asyncShutdown()
|
|
throw error
|
|
}
|
|
|
|
}
|
|
|
|
/// Set's up the database and a test user for running tests that require a
|
|
/// a user.
|
|
func withTestUser(
|
|
setupDependencies: (inout DependencyValues) -> Void = { _ in },
|
|
operation: (User) async throws -> Void
|
|
) async throws {
|
|
try await withDatabase(setupDependencies: setupDependencies) {
|
|
@Dependency(\.database.users) var users
|
|
let user = try await users.create(
|
|
.init(email: "testy@example.com", password: "super-secret", confirmPassword: "super-secret")
|
|
)
|
|
try await operation(user)
|
|
}
|
|
}
|
|
|
|
/// Set's up the database and a test user for running tests that require a
|
|
/// a user.
|
|
func withTestUserAndProject(
|
|
setupDependencies: (inout DependencyValues) -> Void = { _ in },
|
|
operation: (User, Project) async throws -> Void
|
|
) async throws {
|
|
try await withDatabase(setupDependencies: setupDependencies) {
|
|
@Dependency(\.database) var database
|
|
let user = try await database.users.create(
|
|
.init(email: "testy@example.com", password: "super-secret", confirmPassword: "super-secret")
|
|
)
|
|
let project = try await database.projects.create(user.id, .mock)
|
|
try await operation(user, project)
|
|
}
|
|
}
|
|
|
|
extension Project.Create {
|
|
|
|
static let mock = Self(
|
|
name: "Testy McTestface",
|
|
streetAddress: "1234 Sesame St",
|
|
city: "Nowhere",
|
|
state: "MN",
|
|
zipCode: "55555",
|
|
sensibleHeatRatio: 0.83
|
|
)
|
|
}
|