test(nehelper): fix flaky cert refresh test

ProtonVPN · Feb 5, 2024 · fb72441 · fb72441
1 parent 2bf285c
commit fb72441
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/libraries/NEHelper/Tests/NEHelperTests/ExtensionAPITests/CertificateRefreshTests.swift b/libraries/NEHelper/Tests/NEHelperTests/ExtensionAPITests/CertificateRefreshTests.swift
@@ -751,9 +751,8 @@ class CertificateRefreshTests: ExtensionAPIServiceTestCase {
  XCTFail("Shouldn't have tried to refresh; operation should have been cancelled")
  }
  expectations.managerStop.fulfill()
+ self.timerFactory.runAllScheduledWork()
  }
-
- timerFactory.runAllScheduledWork()
  wait(for: [expectations.certRefreshCancelled, expectations.managerStop], timeout: expectationTimeout, enforceOrder: true)
  }
 

diff --git a/libraries/NEHelper/Tests/NEHelperTests/ExtensionAPITests/ExtensionAPIServiceTestCase.swift b/libraries/NEHelper/Tests/NEHelperTests/ExtensionAPITests/ExtensionAPIServiceTestCase.swift
@@ -118,7 +118,7 @@ class ExtensionAPIServiceTestCase: XCTestCase, ExtensionAPIServiceDelegate {
  serverStatusCallback = failCallback
 
  keychain = MockAuthKeychain()
- try! await keychain.store(AuthCredentials(username: "johnny",
+ try! keychain.store(AuthCredentials(username: "johnny",
  accessToken: "12345",
  refreshToken: "54321",
  sessionId: "15213",