From f9dc1d7a72d524b0f13f3cca0cee36f38e2a25b5 Mon Sep 17 00:00:00 2001 From: Daniel Saidi Date: Wed, 5 Aug 2020 14:19:15 +0200 Subject: [PATCH] Improve async trigger --- Tests/SwiftKitTests/AsyncTrigger.swift | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/Tests/SwiftKitTests/AsyncTrigger.swift b/Tests/SwiftKitTests/AsyncTrigger.swift index 3b57738..23f4f6b 100644 --- a/Tests/SwiftKitTests/AsyncTrigger.swift +++ b/Tests/SwiftKitTests/AsyncTrigger.swift @@ -8,12 +8,19 @@ import Foundation -struct AsyncTrigger { +class AsyncTrigger { - var hasTriggered: Bool { triggerCount > 0 } - private(set) var triggerCount = 0 + public init() {} - mutating func trigger() { - triggerCount += 1 + private var counter = 0 + + public var hasTriggered: Bool { counter > 0 } + + public func hasTriggered(numberOfTimes: Int) -> Bool { + counter == numberOfTimes + } + + public func trigger() { + counter += 1 } }