Add preferredScreenCaptureFormat to XCScheme.TestAcion (#792)

tuist · Sep 8, 2023 · 3797181 · 3797181
1 parent 30df77b
commit 3797181
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/Fixtures/iOS/Project.xcodeproj/xcshareddata/xcschemes/iOS.xcscheme b/Fixtures/iOS/Project.xcodeproj/xcshareddata/xcschemes/iOS.xcscheme
@@ -46,6 +46,7 @@
       selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
       selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
       shouldUseLaunchSchemeArgsEnv = "YES"
+      preferredScreenCaptureFormat = "screenshots"
       codeCoverageEnabled = "YES"
       onlyGenerateCoverageForSpecifiedTargets = "YES">
       <PreActions>

diff --git a/Sources/XcodeProj/Extensions/AEXML+XcodeFormat.swift b/Sources/XcodeProj/Extensions/AEXML+XcodeFormat.swift
@@ -38,6 +38,7 @@ let attributesOrder: [String: [String]] = [
         "shouldUseLaunchSchemeArgsEnv",
         "disableMainThreadChecker",
         "region",
+        "preferredScreenCaptureFormat",
         "codeCoverageEnabled",
         "onlyGenerateCoverageForSpecifiedTargets",
     ],

diff --git a/Sources/XcodeProj/Scheme/XCScheme+TestAction.swift b/Sources/XcodeProj/Scheme/XCScheme+TestAction.swift
@@ -7,6 +7,10 @@ extension XCScheme {
         public enum AttachmentLifetime: String {
             case keepAlways, keepNever
         }
+
+        public enum ScreenCaptureFormat: String {
+            case screenshots, screenRecording
+        }
 
         // MARK: - Static
 
@@ -35,6 +39,7 @@ extension XCScheme {
         public var language: String?
         public var region: String?
         public var systemAttachmentLifetime: AttachmentLifetime?
+        public var preferredScreenCaptureFormat: ScreenCaptureFormat?
         public var userAttachmentLifetime: AttachmentLifetime?
         public var customLLDBInitFile: String?
 
@@ -63,6 +68,7 @@ extension XCScheme {
                     language: String? = nil,
                     region: String? = nil,
                     systemAttachmentLifetime: AttachmentLifetime? = nil,
+                    preferredScreenCaptureFormat: ScreenCaptureFormat? = nil,
                     userAttachmentLifetime: AttachmentLifetime? = nil,
                     customLLDBInitFile: String? = nil) {
             self.buildConfiguration = buildConfiguration
@@ -86,6 +92,7 @@ extension XCScheme {
             self.language = language
             self.region = region
             self.systemAttachmentLifetime = systemAttachmentLifetime
+            self.preferredScreenCaptureFormat = preferredScreenCaptureFormat
             self.userAttachmentLifetime = userAttachmentLifetime
             self.customLLDBInitFile = customLLDBInitFile
             super.init(preActions, postActions)
@@ -137,6 +144,8 @@ extension XCScheme {
 
             systemAttachmentLifetime = element.attributes["systemAttachmentLifetime"]
                 .flatMap(AttachmentLifetime.init(rawValue:))
+            preferredScreenCaptureFormat = element.attributes["preferredScreenCaptureFormat"]
+                .flatMap(ScreenCaptureFormat.init(rawValue:))
             userAttachmentLifetime = element.attributes["userAttachmentLifetime"]
                 .flatMap(AttachmentLifetime.init(rawValue:))
             customLLDBInitFile = element.attributes["customLLDBInitFile"]
@@ -177,6 +186,14 @@ extension XCScheme {
                 attributes["disableMainThreadChecker"] = disableMainThreadChecker.xmlString
             }
             attributes["systemAttachmentLifetime"] = systemAttachmentLifetime?.rawValue
+
+            switch preferredScreenCaptureFormat {
+            case .screenshots:
+                attributes["preferredScreenCaptureFormat"] = preferredScreenCaptureFormat?.rawValue
+            case .none, .screenRecording:
+                break
+            }
+
             if case .keepAlways? = userAttachmentLifetime {
                 attributes["userAttachmentLifetime"] = userAttachmentLifetime?.rawValue
             }
@@ -253,6 +270,7 @@ extension XCScheme {
                 language == rhs.language &&
                 region == rhs.region &&
                 systemAttachmentLifetime == rhs.systemAttachmentLifetime &&
+                preferredScreenCaptureFormat == rhs.preferredScreenCaptureFormat &&
                 userAttachmentLifetime == rhs.userAttachmentLifetime &&
                 codeCoverageTargets == rhs.codeCoverageTargets &&
                 customLLDBInitFile == rhs.customLLDBInitFile

diff --git a/Tests/XcodeProjTests/Scheme/XCSchemeTests.swift b/Tests/XcodeProjTests/Scheme/XCSchemeTests.swift
@@ -343,6 +343,7 @@ final class XCSchemeIntegrationTests: XCTestCase {
         XCTAssertEqual(scheme.testAction?.selectedDebuggerIdentifier, "Xcode.DebuggerFoundation.Debugger.LLDB")
         XCTAssertEqual(scheme.testAction?.selectedDebuggerIdentifier, "Xcode.DebuggerFoundation.Debugger.LLDB")
         XCTAssertEqual(scheme.testAction?.shouldUseLaunchSchemeArgsEnv, true)
+        XCTAssertEqual(scheme.testAction?.preferredScreenCaptureFormat, .screenshots)
         XCTAssertEqual(scheme.testAction?.codeCoverageEnabled, true)
         XCTAssertEqual(scheme.testAction?.onlyGenerateCoverageForSpecifiedTargets, true)
         XCTAssertEqual(scheme.testAction?.testables.first?.skipped, false)