Fixup SSK tests

Signal/test/util/SearcherTest.swift

@@ -112,13 +112,14 @@ class ConversationSearcherTest: XCTestCase {
     override func setUp() {
         super.setUp()
 
-        FullTextSearchFinder.syncRegisterDatabaseExtension(storage: OWSPrimaryStorage.shared())
+        FullTextSearchFinder.ensureDatabaseExtensionRegistered(storage: OWSPrimaryStorage.shared())
 
         TSContactThread.removeAllObjectsInCollection()
         TSGroupThread.removeAllObjectsInCollection()
         TSMessage.removeAllObjectsInCollection()
 
         originalEnvironment = TextSecureKitEnv.shared()
+        assert(originalEnvironment != nil)
 
         let testEnvironment: StubbableEnvironment = StubbableEnvironment(proxy: originalEnvironment!)
         testEnvironment.stubbedContactsManager = FakeContactsManager()

SignalServiceKit/src/Storage/FullTextSearchFinder.swift

@@ -264,7 +264,11 @@ public class FullTextSearchFinder: NSObject {
     }
 
     // Only for testing.
-    public class func syncRegisterDatabaseExtension(storage: OWSStorage) {
+    public class func ensureDatabaseExtensionRegistered(storage: OWSStorage) {
+        guard storage.registeredExtension(dbExtensionName) == nil else {
+            return
+        }
+
         storage.register(dbExtensionConfig, withName: dbExtensionName)
     }
 

SignalServiceKit/src/TextSecureKitEnv.m

@@ -1,5 +1,5 @@
 //
-//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 //
 
 #import "TextSecureKitEnv.h"
@@ -59,7 +59,7 @@ static TextSecureKitEnv *sharedTextSecureKitEnv;
 + (void)setSharedEnv:(TextSecureKitEnv *)env
 {
     OWSAssert(env);
-    OWSAssert(!sharedTextSecureKitEnv || !CurrentAppContext().isMainApp);
+    OWSAssert(!sharedTextSecureKitEnv || CurrentAppContext().isRunningTests);
 
     sharedTextSecureKitEnv = env;
 }