Swift Playground中如何区分执行XCTest与PlaygroundPage场景？

当然可以实现这个需求！下面分享几个实用的方案，你可以根据自己的场景选择最合适的：

方案一：显式传递标记参数（最直观可靠） #

这是最简单也最不容易出错的方式——给你的MyClass添加一个初始化参数，用来标记当前是否处于测试模式：

import UIKit
import PlaygroundSupport

class MyClass: UIView {
    let isRunningTest: Bool
    
    // 带标记的初始化方法，默认非测试模式
    init(isRunningTest: Bool = false) {
        self.isRunningTest = isRunningTest
        super.init(frame: .zero)
        setup()
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    private func setup() {
        if isRunningTest {
            print("当前是测试套件执行模式")
            // 这里写测试专属逻辑
        } else {
            print("当前是LiveView加载模式")
            // 这里写LiveView专属逻辑
        }
    }
}

// 在测试中调用时，传入标记
// MyClassTest.defaultTestSuite.run() 内部实例化代码：
// let myClass = MyClass(isRunningTest: true)

// 在Playground中设置LiveView时，使用默认值即可
PlaygroundPage.current.liveView = MyClass()

这种方式完全由你控制逻辑，不依赖任何环境细节，稳定性最高。

方案二：通过环境变量标记 #

你可以在测试运行前设置一个自定义环境变量，然后在代码中读取这个变量来判断场景：

步骤1：在测试类中设置环境变量 #

import XCTest
@testable import YourPlaygroundModule

class MyClassTest: XCTestCase {
    override class func setUp() {
        super.setUp()
        // 设置测试环境标记
        ProcessInfo.processInfo.environment["IS_RUNNING_TEST"] = "true"
    }
    
    override class func tearDown() {
        // 测试结束后清理环境变量
        ProcessInfo.processInfo.environment.removeValue(forKey: "IS_RUNNING_TEST")
        super.tearDown()
    }
    
    func testExample() {
        let myClass = MyClass()
        // 执行你的测试逻辑
    }
}

步骤2：在MyClass中检查环境变量 #

class MyClass: UIView {
    private func setup() {
        let isRunningTest = ProcessInfo.processInfo.environment["IS_RUNNING_TEST"] == "true"
        if isRunningTest {
            print("测试模式")
        } else {
            print("LiveView模式")
        }
    }
}

方案三：检查运行时信息（适配性稍弱） #

如果不想修改初始化方法或设置环境变量，还可以通过运行时的Bundle信息或调用栈来判断：

方式A：检查Bundle标识符 #

测试目标的Bundle ID通常会带有Tests后缀，你可以通过Bundle信息判断：

private func isRunningTest() -> Bool {
    guard let bundleID = Bundle.main.bundleIdentifier else { return false }
    return bundleID.contains("Tests")
}

方式B：检查调用栈 #

通过解析调用栈，查看是否包含测试相关的方法名：

private func isRunningTest() -> Bool {
    let callStack = Thread.callStackSymbols
    // 检查调用栈中是否包含测试套件或XCTest相关的标识
    return callStack.contains { $0.contains("defaultTestSuite") || $0.contains("XCTest") }
}

不过这种方式依赖运行时的调用栈结构，可能会因为Swift版本或Playground更新而失效，不如前两种方案稳定。

内容的提问来源于stack exchange，提问作者user2924482