David Jennes
88 lines
2.4 KiB
Swift
88 lines
2.4 KiB
Swift
//
|
|
// SwiftGenKit UnitTests
|
|
// Copyright © 2022 SwiftGen
|
|
// MIT Licence
|
|
//
|
|
|
|
import PathKit
|
|
import SwiftGenCLI
|
|
import SwiftGenKit
|
|
import XCTest
|
|
|
|
// MARK: - Safe access to fixtures
|
|
|
|
public class Fixtures {
|
|
public enum Directory: String {
|
|
case colors = "Colors"
|
|
case coreData = "CoreData"
|
|
case files = "Files"
|
|
case fonts = "Fonts"
|
|
case interfaceBuilder = "IB"
|
|
case interfaceBuilderiOS = "IB-iOS"
|
|
case interfaceBuilderMacOS = "IB-macOS"
|
|
case json = "JSON"
|
|
case plist = "Plist"
|
|
case plistBad = "Plist/bad"
|
|
case plistGood = "Plist/good"
|
|
case strings = "Strings"
|
|
case xcassets = "XCAssets"
|
|
case yaml = "YAML"
|
|
case yamlBad = "YAML/bad"
|
|
case yamlGood = "YAML/good"
|
|
}
|
|
|
|
private static let testBundle = Bundle.module
|
|
private init() {}
|
|
|
|
public static func resourceDirectory(sub: Directory? = nil) -> Path {
|
|
guard let resources = testBundle.path(forResource: "Resources", ofType: nil) else {
|
|
fatalError("Unable to find resource directory URL")
|
|
}
|
|
|
|
if let dir = sub {
|
|
return Path(resources) + dir.rawValue
|
|
} else {
|
|
return Path(resources)
|
|
}
|
|
}
|
|
|
|
public static func resource(for name: String, sub: Directory) -> Path {
|
|
path(for: name, subDirectory: "Resources/\(sub.rawValue)")
|
|
}
|
|
|
|
public static func config(for name: String) -> Path {
|
|
path(for: "\(name).yml", subDirectory: "Configs")
|
|
}
|
|
|
|
public static func context(for name: String, sub: Directory) -> [String: Any] {
|
|
let path = self.path(for: name, subDirectory: "StencilContexts/\(sub.rawValue)")
|
|
|
|
guard let yaml = try? YAML.read(path: path),
|
|
let result = yaml as? [String: Any] else {
|
|
fatalError("Unable to load fixture content")
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
static func template(for name: String, sub: Directory) -> Path {
|
|
// Directly load from SwiftGenFramework bundle
|
|
return Path.bundledTemplates + sub.rawValue.lowercased() + name
|
|
}
|
|
|
|
static func output(template: String, variation: String, sub: Directory) -> String {
|
|
do {
|
|
return try path(for: variation, subDirectory: "Generated/\(sub.rawValue)/\(template)").read()
|
|
} catch let error {
|
|
fatalError("Unable to load fixture content: \(error)")
|
|
}
|
|
}
|
|
|
|
private static func path(for name: String, subDirectory: String? = nil) -> Path {
|
|
guard let path = testBundle.path(forResource: name, ofType: "", inDirectory: subDirectory) else {
|
|
fatalError("Unable to find fixture \"\(name)\"")
|
|
}
|
|
return Path(path)
|
|
}
|
|
}
|