Testing Timer Based Features in Swift

[fespinoza]

• Source: https://medium.com/@isahkaren16/testing-timer-based-features-in-swift-5ecaa4bb0c73

class MockTimer: Timer {
    var block: ((Timer) -> Void)!

    static var currentTimer: MockTimer!

    override func fire() {
        block(self)
    }

    override open class func scheduledTimer(
        withTimeInterval interval: TimeInterval, repeats: Bool, block: @escaping (Timer) -> Void
    ) -> Timer {
        let mockTimer = MockTimer()
        mockTimer.block = block
        MockTimer.currentTimer = mockTimer
        return mockTimer
    }
}