From d6d403ce690d5acad1046429fe59b08ec5bc8c6d Mon Sep 17 00:00:00 2001 From: Michael Kirk Date: Mon, 21 Aug 2017 11:24:24 -0400 Subject: [PATCH] fix some tests // FREEBIE --- Signal/test/Models/AccountManagerTest.swift | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/Signal/test/Models/AccountManagerTest.swift b/Signal/test/Models/AccountManagerTest.swift index 9f260b7e7..aa50d5e3b 100644 --- a/Signal/test/Models/AccountManagerTest.swift +++ b/Signal/test/Models/AccountManagerTest.swift @@ -43,9 +43,10 @@ class TokenObtainingTSAccountManager: VerifyingTSAccountManager { class AccountManagerTest: XCTestCase { let tsAccountManager = FailingTSAccountManager(networkManager: TSNetworkManager.shared(), storageManager: TSStorageManager.shared()) + var preferences = PropertyListPreferences() func testRegisterWhenEmptyCode() { - let accountManager = AccountManager(textSecureAccountManager: tsAccountManager) + let accountManager = AccountManager(textSecureAccountManager: tsAccountManager, preferences: self.preferences) let expectation = self.expectation(description: "should fail") @@ -66,7 +67,7 @@ class AccountManagerTest: XCTestCase { } func testRegisterWhenVerificationFails() { - let accountManager = AccountManager(textSecureAccountManager: tsAccountManager) + let accountManager = AccountManager(textSecureAccountManager: tsAccountManager, preferences: self.preferences) let expectation = self.expectation(description: "should fail") @@ -86,9 +87,11 @@ class AccountManagerTest: XCTestCase { } func testSuccessfulRegistration() { + Environment.setCurrent(Release.releaseEnvironment()) + let tsAccountManager = TokenObtainingTSAccountManager(networkManager: TSNetworkManager.shared(), storageManager: TSStorageManager.shared()) - let accountManager = AccountManager(textSecureAccountManager: tsAccountManager) + let accountManager = AccountManager(textSecureAccountManager: tsAccountManager, preferences: self.preferences) let expectation = self.expectation(description: "should succeed") @@ -104,7 +107,7 @@ class AccountManagerTest: XCTestCase { } func testUpdatePushTokens() { - let accountManager = AccountManager(textSecureAccountManager: tsAccountManager) + let accountManager = AccountManager(textSecureAccountManager: tsAccountManager, preferences: self.preferences) let expectation = self.expectation(description: "should fail")