139 lines
4.5 KiB
Swift
139 lines
4.5 KiB
Swift
//
|
|
// SnapshotHelper.swift
|
|
// Example
|
|
//
|
|
// Created by Felix Krause on 10/8/15.
|
|
// Copyright © 2015 Felix Krause. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import XCTest
|
|
|
|
var deviceLanguage = ""
|
|
var locale = ""
|
|
|
|
@available(*, deprecated, message: "use setupSnapshot: instead")
|
|
func setLanguage(_ app: XCUIApplication) {
|
|
setupSnapshot(app)
|
|
}
|
|
|
|
func setupSnapshot(_ app: XCUIApplication) {
|
|
Snapshot.setupSnapshot(app)
|
|
}
|
|
|
|
func snapshot(_ name: String, waitForLoadingIndicator: Bool = true) {
|
|
Snapshot.snapshot(name, waitForLoadingIndicator: waitForLoadingIndicator)
|
|
}
|
|
|
|
open class Snapshot: NSObject {
|
|
|
|
open class func setupSnapshot(_ app: XCUIApplication) {
|
|
setLanguage(app)
|
|
setLocale(app)
|
|
setLaunchArguments(app)
|
|
}
|
|
|
|
class func setLanguage(_ app: XCUIApplication) {
|
|
guard let prefix = pathPrefix() else {
|
|
return
|
|
}
|
|
|
|
let path = prefix.appendingPathComponent("language.txt")
|
|
|
|
do {
|
|
let trimCharacterSet = CharacterSet.whitespacesAndNewlines
|
|
deviceLanguage = try NSString(contentsOfFile: path, encoding: String.Encoding.utf8.rawValue).trimmingCharacters(in: trimCharacterSet) as String
|
|
app.launchArguments += ["-AppleLanguages", "(\(deviceLanguage))"]
|
|
} catch {
|
|
print("Couldn't detect/set language...")
|
|
}
|
|
}
|
|
|
|
class func setLocale(_ app: XCUIApplication) {
|
|
guard let prefix = pathPrefix() else {
|
|
return
|
|
}
|
|
|
|
let path = prefix.appendingPathComponent("locale.txt")
|
|
|
|
do {
|
|
let trimCharacterSet = CharacterSet.whitespacesAndNewlines
|
|
locale = try NSString(contentsOfFile: path, encoding: String.Encoding.utf8.rawValue).trimmingCharacters(in: trimCharacterSet) as String
|
|
} catch {
|
|
print("Couldn't detect/set locale...")
|
|
}
|
|
if locale.isEmpty {
|
|
locale = Locale(identifier: deviceLanguage).identifier
|
|
}
|
|
app.launchArguments += ["-AppleLocale", "\"\(locale)\""]
|
|
}
|
|
|
|
class func setLaunchArguments(_ app: XCUIApplication) {
|
|
guard let prefix = pathPrefix() else {
|
|
return
|
|
}
|
|
|
|
let path = prefix.appendingPathComponent("snapshot-launch_arguments.txt")
|
|
app.launchArguments += ["-FASTLANE_SNAPSHOT", "YES", "-ui_testing"]
|
|
|
|
do {
|
|
let launchArguments = try NSString(contentsOfFile: path, encoding: String.Encoding.utf8.rawValue) as String
|
|
let regex = try NSRegularExpression(pattern: "(\\\".+?\\\"|\\S+)", options: [])
|
|
let matches = regex.matches(in: launchArguments, options: [], range: NSRange(location:0, length:launchArguments.characters.count))
|
|
let results = matches.map { result -> String in
|
|
(launchArguments as NSString).substring(with: result.range)
|
|
}
|
|
app.launchArguments += results
|
|
} catch {
|
|
print("Couldn't detect/set launch_arguments...")
|
|
}
|
|
}
|
|
|
|
open class func snapshot(_ name: String, waitForLoadingIndicator: Bool = true) {
|
|
if waitForLoadingIndicator {
|
|
waitForLoadingIndicatorToDisappear()
|
|
}
|
|
|
|
print("snapshot: \(name)") // more information about this, check out https://github.com/fastlane/fastlane/tree/master/snapshot#how-does-it-work
|
|
|
|
sleep(1) // Waiting for the animation to be finished (kind of)
|
|
|
|
#if os(tvOS)
|
|
XCUIApplication().childrenMatchingType(.Browser).count
|
|
#else
|
|
XCUIDevice.shared.orientation = .unknown
|
|
#endif
|
|
}
|
|
|
|
class func waitForLoadingIndicatorToDisappear() {
|
|
#if os(tvOS)
|
|
return
|
|
#endif
|
|
|
|
let query = XCUIApplication().statusBars.children(matching: .other).element(boundBy: 1).children(matching: .other)
|
|
|
|
while (0..<query.count).map({ query.element(boundBy: $0) }).contains(where: { $0.isLoadingIndicator }) {
|
|
sleep(1)
|
|
print("Waiting for loading indicator to disappear...")
|
|
}
|
|
}
|
|
|
|
class func pathPrefix() -> NSString? {
|
|
if let path = ProcessInfo().environment["SIMULATOR_HOST_HOME"] as NSString? {
|
|
return path.appendingPathComponent("Library/Caches/tools.fastlane") as NSString?
|
|
}
|
|
print("Couldn't find Snapshot configuration files at ~/Library/Caches/tools.fastlane")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
extension XCUIElement {
|
|
var isLoadingIndicator: Bool {
|
|
return self.frame.size == CGSize(width: 10, height: 20)
|
|
}
|
|
}
|
|
|
|
// Please don't remove the lines below
|
|
// They are used to detect outdated configuration files
|
|
// SnapshotHelperVersion [1.2]
|