4 files changed, 4 insertions, 10 deletions
diff --git a/Example/Messaging/Tests/FIRMessagingDataMessageManagerTest.m b/Example/Messaging/Tests/FIRMessagingDataMessageManagerTest.m
index 2b4f407..5771337 100644
--- a/Example/Messaging/Tests/FIRMessagingDataMessageManagerTest.m
+++ b/Example/Messaging/Tests/FIRMessagingDataMessageManagerTest.m
@@ -22,7 +22,6 @@
 
 #import "FIRMessaging.h"
 #import "FIRMessagingClient.h"
-#import "FIRMessagingConfig.h"
 #import "FIRMessagingConnection.h"
 #import "FIRMessagingDataMessageManager.h"
 #import "FIRMessagingReceiver.h"
diff --git a/Example/Messaging/Tests/FIRMessagingLinkHandlingTest.m b/Example/Messaging/Tests/FIRMessagingLinkHandlingTest.m
index 94ce530..8f93240 100644
--- a/Example/Messaging/Tests/FIRMessagingLinkHandlingTest.m
+++ b/Example/Messaging/Tests/FIRMessagingLinkHandlingTest.m
@@ -19,13 +19,12 @@
 #import <OCMock/OCMock.h>
 
 #import "FIRMessaging.h"
-#import "FIRMessagingConfig.h"
 #import "FIRMessagingConstants.h"
 #import "FIRMessagingTestNotificationUtilities.h"
 
 @interface FIRMessaging ()
 
-- (instancetype)initWithConfig:(FIRMessagingConfig *)config;
+- (instancetype)initPrivately;
 - (NSURL *)linkURLFromMessage:(NSDictionary *)message;
 
 @end
@@ -41,8 +40,7 @@
 - (void)setUp {
   [super setUp];
 
-  FIRMessagingConfig *config = [FIRMessagingConfig defaultConfig];
-  _messaging = [[FIRMessaging alloc] initWithConfig:config];
+  _messaging = [[FIRMessaging alloc] initPrivately];
 }
 
 - (void)tearDown {
diff --git a/Example/Messaging/Tests/FIRMessagingServiceTest.m b/Example/Messaging/Tests/FIRMessagingServiceTest.m
index bdbc0c2..b3dce57 100644
--- a/Example/Messaging/Tests/FIRMessagingServiceTest.m
+++ b/Example/Messaging/Tests/FIRMessagingServiceTest.m
@@ -21,7 +21,6 @@
 
 #import "FIRMessaging.h"
 #import "FIRMessagingClient.h"
-#import "FIRMessagingConfig.h"
 #import "FIRMessagingPubSub.h"
 #import "FIRMessagingTopicsCommon.h"
 #import "InternalHeaders/FIRMessagingInternalUtilities.h"
diff --git a/Example/Messaging/Tests/FIRMessagingTest.m b/Example/Messaging/Tests/FIRMessagingTest.m
index 3d7c95f..09cdffc 100644
--- a/Example/Messaging/Tests/FIRMessagingTest.m
+++ b/Example/Messaging/Tests/FIRMessagingTest.m
@@ -19,7 +19,6 @@
 #import <OCMock/OCMock.h>
 
 #import "FIRMessaging.h"
-#import "FIRMessagingConfig.h"
 #import "FIRMessagingInstanceIDProxy.h"
 
 extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
@@ -30,7 +29,7 @@ extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
 @property(nonatomic, readwrite, strong) NSData *apnsTokenData;
 @property(nonatomic, readwrite, strong) FIRMessagingInstanceIDProxy *instanceIDProxy;
 
-- (instancetype)initWithConfig:(FIRMessagingConfig *)config;
+- (instancetype)initPrivately;
 // Direct Channel Methods
 - (void)updateAutomaticClientConnection;
 - (BOOL)shouldBeConnectedAutomatically;
@@ -49,8 +48,7 @@ extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
 
 - (void)setUp {
   [super setUp];
-  FIRMessagingConfig *config = [FIRMessagingConfig defaultConfig];
-  _messaging = [[FIRMessaging alloc] initWithConfig:config];
+  _messaging = [[FIRMessaging alloc] initPrivately];
   _mockMessaging = OCMPartialMock(self.messaging);
   _mockInstanceIDProxy = OCMPartialMock(self.messaging.instanceIDProxy);
   self.messaging.instanceIDProxy = _mockInstanceIDProxy;