adding unit test for auto init enable function (#710)

1 files changed, 14 insertions, 0 deletions

diff --git a/Example/Messaging/Tests/FIRMessagingTest.m b/Example/Messaging/Tests/FIRMessagingTest.m
index 09cdffc..86ed647 100644
--- a/Example/Messaging/Tests/FIRMessagingTest.m
+++ b/Example/Messaging/Tests/FIRMessagingTest.m
@@ -20,6 +20,7 @@
 
 #import "FIRMessaging.h"
 #import "FIRMessagingInstanceIDProxy.h"
+#import "FIRMessaging_Private.h"
 
 extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
 
@@ -28,6 +29,7 @@ extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
 @property(nonatomic, readwrite, strong) NSString *defaultFcmToken;
 @property(nonatomic, readwrite, strong) NSData *apnsTokenData;
 @property(nonatomic, readwrite, strong) FIRMessagingInstanceIDProxy *instanceIDProxy;
+@property(nonatomic, readwrite, strong) NSUserDefaults *messagingUserDefaults;
 
 - (instancetype)initPrivately;
 // Direct Channel Methods
@@ -52,6 +54,9 @@ extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
   _mockMessaging = OCMPartialMock(self.messaging);
   _mockInstanceIDProxy = OCMPartialMock(self.messaging.instanceIDProxy);
   self.messaging.instanceIDProxy = _mockInstanceIDProxy;
+  [self.messaging.messagingUserDefaults removePersistentDomainForName:kFIRMessagingSuiteName];
+  self.messaging.messagingUserDefaults =
+      [[NSUserDefaults alloc] initWithSuiteName:kFIRMessagingSuiteName];
 }
 
 - (void)tearDown {
@@ -60,6 +65,15 @@ extern NSString *const kFIRMessagingFCMTokenFetchAPNSOption;
   [super tearDown];
 }
 
+- (void)testAutoInitEnableFlag {
+  // Should read from Info.plist
+  XCTAssertFalse(_messaging.isAutoInitEnabled);
+
+  // Now set the flag should overwrite Info.plist value.
+  _messaging.autoInitEnabled = YES;
+  XCTAssertTrue(_messaging.isAutoInitEnabled);
+}
+
 #pragma mark - Direct Channel Establishment Testing
 
 // Should connect with valid token and application in foreground