XCTContext.runActivities(setUp:tearDown:blocks:)

XCTContext+Extension.swift

//
//  XCTContext+Extension.swift
//  TestEachTests
//
//  Created by ky0me22 on 2022/04/07.
//

import XCTest

extension XCTContext {
    typealias VoidClosure = () -> Void
    typealias ThrowClosure = () throws -> Void

    private static func runBlock(block: ThrowClosure) rethrows -> Void {
        do {
            try block()
        } catch {
            throw error
        }
    }

    class func runActivities(
        _ dummyClosure: ThrowClosure = {},
        setUp: VoidClosure? = nil,
        tearDown: VoidClosure? = nil,
        _ blocks: ThrowClosure...
    ) rethrows {
        try XCTContext.runBlock {
            for block in blocks {
                do {
                    setUp?()
                    try block()
                    tearDown?()
                } catch {
                    throw error
                }
            }
        }
    }
}