import XCTest

extension XCTActivity {
    func addScreenshot(
        named name: String,
        lifetime: XCTAttachment.Lifetime = .keepAlways
    ) {
        let screenshot = XCUIScreen.main.screenshot()
        let attachment = XCTAttachment(screenshot: screenshot)
        attachment.name = name
        attachment.lifetime = lifetime
        add(attachment)
    }
}