fix: Fix test testResolveAndApplyIntegerFlagError

spotify · Apr 30, 2024 · 7a1929a · 7a1929a
1 parent 3bdebd1
commit 7a1929a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Tests/ConfidenceProviderTests/ConfidenceFeatureProviderTest.swift b/Tests/ConfidenceProviderTests/ConfidenceFeatureProviderTest.swift
@@ -340,6 +340,7 @@ class ConfidenceFeatureProviderTest: XCTestCase {
     }
 
     func testResolveAndApplyIntegerFlagError() throws {
+        flagApplier = FlagApplierMock(expectedApplies: 2)
         let resolve: [String: MockedResolveClientURLProtocol.ResolvedTestFlag] = [
             "user1": .init(variant: "control", value: .structure(["size": .integer(3)]))
         ]

diff --git a/Tests/ConfidenceProviderTests/Helpers/FlagApplierMock.swift b/Tests/ConfidenceProviderTests/Helpers/FlagApplierMock.swift
@@ -7,6 +7,10 @@ class FlagApplierMock: FlagApplier {
     var applyCallCount = 0
     var applyExpectation = XCTestExpectation(description: "Flag Applied")
 
+    init(expectedApplies: Int = 1) {
+        applyExpectation.expectedFulfillmentCount = expectedApplies
+    }
+
     func apply(flagName: String, resolveToken: String) async {
         applyCallCount += 1
         applyExpectation.fulfill()