From df6fe05d0da556f95863d2caa381e903b2a591de Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Mon, 15 Oct 2018 13:39:45 -0400
Subject: [PATCH] Get tests running.

---
 Signal/test/Models/AccountManagerTest.swift | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/Signal/test/Models/AccountManagerTest.swift b/Signal/test/Models/AccountManagerTest.swift
index 3293f7a7b..2aedf780b 100644
--- a/Signal/test/Models/AccountManagerTest.swift
+++ b/Signal/test/Models/AccountManagerTest.swift
@@ -47,11 +47,6 @@ class VerifyingTSAccountManager: FailingTSAccountManager {
                                 success: @escaping () -> Void, failure: @escaping (Error) -> Void) {
         success()
     }
-
-    // TODO:
-//    override func updateAcountAttributes() -> Promise<Void> {
-//        return Promise(value: ())
-//    }
 }
 
 class TokenObtainingTSAccountManager: VerifyingTSAccountManager {
@@ -114,6 +109,8 @@ class AccountManagerTest: SignalBaseTest {
 
     func testSuccessfulRegistration() {
         let tsAccountManager = TokenObtainingTSAccountManager(primaryStorage: OWSPrimaryStorage.shared())
+        let sskEnvironment = SSKEnvironment.shared as! MockSSKEnvironment
+        sskEnvironment.tsAccountManager = tsAccountManager
 
         let accountManager = AccountManager()