From d6d403ce690d5acad1046429fe59b08ec5bc8c6d Mon Sep 17 00:00:00 2001
From: Michael Kirk <michael.code@endoftheworl.de>
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")