firebase-ios-sdk/Example/Messaging/Tests/FIRMessagingAnalyticsTest.m

/*
 * Copyright 2018 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import <OCMock/OCMock.h>
#import <XCTest/XCTest.h>

#import "FIRMessagingAnalytics.h"
#import <FirebaseAnalyticsInterop/FIRInteropEventNames.h>
#import <FirebaseAnalyticsInterop/FIRInteropParameterNames.h>

static NSString *const kFIRParameterLabel = @"label";
static NSString *const kReengagementSource = @"Firebase";
static NSString *const kReengagementMedium = @"notification";
static NSString *const kFIREventOriginFCM = @"fcm";
static const NSTimeInterval kAsyncTestTimout = 0.5;

typedef void (^FakeAnalyticsLogEventHandler)(NSString *origin,
                                             NSString *name,
                                             NSDictionary *parameters);
typedef void (^FakeAnalyticsUserPropertyHandler)(NSString *origin,
                                                 NSString *name,
                                                 id value);

@interface FakeAnalytics : NSObject <FIRAnalyticsInterop>

- (instancetype)initWithEventHandler:(FakeAnalyticsLogEventHandler)eventHandler;
- (instancetype)initWithUserPropertyHandler:(FakeAnalyticsUserPropertyHandler)userPropertyHandler;

@end

@implementation FakeAnalytics

static FakeAnalyticsLogEventHandler _eventHandler;
static FakeAnalyticsLogEventHandler _userPropertyHandler;

- (instancetype)initWithEventHandler:(FakeAnalyticsLogEventHandler)eventHandler {
  self = [super init];
  if (self) {
    _eventHandler = eventHandler;
  }
  return self;
}

- (instancetype)initWithUserPropertyHandler:(FakeAnalyticsUserPropertyHandler)userPropertyHandler {
  self = [super init];
  if (self) {
    _userPropertyHandler = userPropertyHandler;
  }
  return self;
}

- (void)logEventWithOrigin:(nonnull NSString *)origin
                      name:(nonnull NSString *)name
                parameters:(nullable NSDictionary<NSString *, id> *)parameters {
  if (_eventHandler) {
    _eventHandler(origin, name, parameters);
  }
}

- (void)setUserPropertyWithOrigin:(nonnull NSString *)origin
                             name:(nonnull NSString *)name
                            value:(nonnull id)value {
  if (_userPropertyHandler) {
    _userPropertyHandler(origin, name, value);
  }
}

// Stubs
- (void)clearConditionalUserProperty:(nonnull NSString *)userPropertyName
                      clearEventName:(nonnull NSString *)clearEventName
                clearEventParameters:(nonnull NSDictionary *)clearEventParameters {
}

- (nonnull NSArray<FIRAConditionalUserProperty *> *)
conditionalUserProperties:(nonnull NSString *)origin
propertyNamePrefix:(nonnull NSString *)propertyNamePrefix {
  return @[];
}

- (NSInteger)maxUserProperties:(nonnull NSString *)origin {
  return -1;
}

- (void)setConditionalUserProperty:(nonnull FIRAConditionalUserProperty *)conditionalUserProperty {
}

- (void)checkLastNotificationForOrigin:(nonnull NSString *)origin
                                 queue:(nonnull dispatch_queue_t)queue
                              callback:(nonnull void (^)(NSString *_Nullable))
                                           currentLastNotificationProperty {
}

- (void)registerAnalyticsListener:(nonnull id<FIRAnalyticsInteropListener>)listener
                       withOrigin:(nonnull NSString *)origin {
}

- (void)unregisterAnalyticsListenerWithOrigin:(nonnull NSString *)origin {
}


@end

@interface FIRMessagingAnalytics (ExposedForTest)
+ (BOOL)canLogNotification:(NSDictionary *)notification;
+ (NSMutableDictionary *)paramsForEvent:(NSString *)event
                       withNotification:(NSDictionary *)notification;
+ (void)logAnalyticsEventWithOrigin:(NSString *)origin
                               name:(NSString *)name
                         parameters:(NSDictionary *)params
                        toAnalytics:(id<FIRAnalyticsInterop> _Nullable)analytics;
+ (void)logUserPropertyForConversionTracking:(NSDictionary *)notification
                                 toAnalytics:(id<FIRAnalyticsInterop> _Nullable)analytics;
+ (void)logOpenNotification:(NSDictionary *)notification
                toAnalytics:(id<FIRAnalyticsInterop> _Nullable)analytics;
+ (void)logForegroundNotification:(NSDictionary *)notification
                      toAnalytics:(id<FIRAnalyticsInterop> _Nullable)analytics;
+ (void)logEvent:(NSString *)event
withNotification:(NSDictionary *)notification
     toAnalytics:(id<FIRAnalyticsInterop> _Nullable)analytics ;;

@end

@interface FIRMessagingAnalyticsTest : XCTestCase

@property(nonatomic, readwrite, strong) id logClassMock;

@end

@implementation FIRMessagingAnalyticsTest

- (void)setUp {
  [super setUp];
  self.logClassMock = OCMClassMock([FIRMessagingAnalytics class]);
}

- (void)tearDown {
  _eventHandler = nil;
  _userPropertyHandler = nil;
  [self.logClassMock stopMocking];
  [super tearDown];
}

- (void)testCanLogNotification {
  NSDictionary *notification = @{};
  XCTAssertFalse([FIRMessagingAnalytics canLogNotification:notification]);

  notification = @{@"google.c.a.e" : @1};
  XCTAssertFalse([FIRMessagingAnalytics canLogNotification:notification]);

  notification = @{@"aps" : @{@"alert" : @"to check the reporting format"}};
  XCTAssertFalse([FIRMessagingAnalytics canLogNotification:notification]);

  notification = @{@"aps" : @{@"alert" : @"to check the reporting format"}, @"google.c.a.e" : @"0"};
  XCTAssertFalse([FIRMessagingAnalytics canLogNotification:notification]);

  notification = @{
                   @"aps" : @{@"alert" : @"to check the reporting format"},
                   @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                   @"gcm.n.e" : @"1",
                   @"google.c.a.c_id" : @"575315420755741863",
                   @"google.c.a.e" : @"1",
                   @"google.c.a.ts" : @"1522880044",
                   @"google.c.a.udt" : @"0"
                   };
  XCTAssertTrue([FIRMessagingAnalytics canLogNotification:notification]);

  notification = @{
                   @"aps" : @{@"content-available" : @"1"},
                   @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                   @"google.c.a.e" : @"1",
                   @"google.c.a.ts" : @"1522880044",
                   @"google.c.a.udt" : @"0"
                   };
  XCTAssertTrue([FIRMessagingAnalytics canLogNotification:notification]);
}

- (void)testEmptyNotification {
  NSMutableDictionary *params = [FIRMessagingAnalytics paramsForEvent:@"" withNotification:@{}];
  XCTAssertNil(params);
}

- (void)testNoParamsIfAnalyticsIsNotEnabled {
  NSDictionary *notification = @{
                                 @"aps" : @{@"alert" : @"to check the reporting format"},
                                 @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                                 @"gcm.n.e" : @"1",
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"google.c.a.udt" : @"0"
                                 };

  NSMutableDictionary *params = [FIRMessagingAnalytics paramsForEvent:@"" withNotification:notification];
  XCTAssertNil(params);
}

- (void)testNoParamsIfEmpty {
  NSDictionary *notification = @{
                                 @"google.c.a.e" : @"1",
                                 };
  NSMutableDictionary *params = [FIRMessagingAnalytics paramsForEvent:@""
                                                     withNotification:notification];
  XCTAssertNotNil(params);

  XCTestExpectation *expectation = [self expectationWithDescription:@"completion"];
  FakeAnalytics *analytics = [[FakeAnalytics alloc]
      initWithEventHandler:^(NSString *origin, NSString *name, NSDictionary *parameters) {
        XCTAssertEqualObjects(origin, kFIREventOriginFCM);
        XCTAssertEqualObjects(name, @"_cmp");
        XCTAssertEqual([parameters count], 0);
        [expectation fulfill];
      }];
  [FIRMessagingAnalytics logEvent:kFIRIEventFirebaseCampaign
                 withNotification:notification
                      toAnalytics:analytics];
  [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
}
- (void)testParamForEventAndNotification {
  NSDictionary *notification = @{
                                 @"aps" : @{@"alert" : @"to check the reporting format"},
                                 @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                                 @"gcm.n.e" : @"1",
                                 @"google.c.a.c_l" : @"Hello World",
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.e" : @"1",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"google.c.a.udt" : @"0",
                                 @"google.c.a.m_l" : @"developer's customized label",
                                 @"from" : @"/topics/news",
                                 };

  NSMutableDictionary *params =
  [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertNotNil(params);
  XCTAssertEqualObjects(params[kFIRIParameterMessageIdentifier], @"575315420755741863");
  XCTAssertEqualObjects(params[kFIRIParameterMessageName], @"Hello World");
  XCTAssertEqualObjects(params[kFIRParameterLabel], @"developer's customized label");
  XCTAssertEqualObjects(params[kFIRIParameterTopic], @"/topics/news");
  XCTAssertEqualObjects([params[kFIRIParameterMessageTime] stringValue], @"1522880044");
  XCTAssertEqualObjects(params[kFIRIParameterMessageDeviceTime], @"0");
}

- (void)testInvalidDataInParamsForLogging {
  NSString *composerIdentifier = @"Hellow World";
  NSDictionary *notification = @{
                                 @"google.c.a.e" : @(YES),
                                 @"google.c.a.c_l" : composerIdentifier,
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.m_l" : @"developer's customized label",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"from" : @"/topics/news",
                                 @"google.c.a.udt" : @"0",
                                 };
  NSMutableDictionary *params =
  [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertNil(params);

  notification = @{
                   @"google.c.a.e" : @"1",
                   @"google.c.a.c_l" : [composerIdentifier dataUsingEncoding:NSUTF8StringEncoding],
                   @"google.c.a.c_id" : @"575315420755741863",
                   @"google.c.a.m_l" : @"developer's customized label",
                   @"google.c.a.ts" : @"1522880044",
                   @"from" : @"/topics/news",
                   @"google.c.a.udt" : @"0",
                   };
  params = [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertNil(params[kFIRIParameterMessageName]);
  XCTAssertEqualObjects(params[kFIRIParameterMessageIdentifier], @"575315420755741863");
  XCTAssertEqualObjects(params[kFIRIParameterTopic], @"/topics/news");

  notification = @{
                   @"google.c.a.e" : @"1",
                   @"google.c.a.c_l" : composerIdentifier,
                   @"google.c.a.c_id" : @(575315420755741863),
                   @"google.c.a.m_l" : @"developer's customized label",
                   @"google.c.a.ts" : @"1522880044",
                   @"from" : @"/topics/news",
                   @"google.c.a.udt" : @"0",
                   };
  params = [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertEqualObjects(params[kFIRIParameterMessageName], composerIdentifier);
  XCTAssertNil(params[kFIRIParameterMessageIdentifier]);
  XCTAssertEqualObjects(params[kFIRIParameterTopic], @"/topics/news");

  notification = @{
                   @"google.c.a.e" : @"1",
                   @"google.c.a.c_l" : composerIdentifier,
                   @"google.c.a.c_id" : @"575315420755741863",
                   @"google.c.a.m_l" : @"developer's customized label",
                   @"google.c.a.ts" : @"0",
                   @"from" : @"/topics/news",
                   @"google.c.a.udt" : @"12345678",
                   };
  params = [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertEqualObjects(params[kFIRIParameterMessageName], composerIdentifier);
  XCTAssertEqualObjects(params[kFIRIParameterMessageIdentifier], @"575315420755741863");
  XCTAssertEqualObjects(params[kFIRParameterLabel], @"developer's customized label");
  XCTAssertEqualObjects(params[kFIRIParameterTopic], @"/topics/news");
  XCTAssertNil(params[kFIRIParameterMessageTime]);
  XCTAssertEqualObjects(params[kFIRIParameterMessageDeviceTime], @"12345678");

  notification = @{
                   @"google.c.a.e" : @"1",
                   @"google.c.a.c_l" : composerIdentifier,
                   @"google.c.a.c_id" : @"575315420755741863",
                   @"google.c.a.m_l" : @"developer's customized label",
                   @"google.c.a.ts" : @(0),
                   @"from" : @"/topics/news",
                   @"google.c.a.udt" : @"12345678",
                   };
  params = [FIRMessagingAnalytics paramsForEvent:kFIRIEventNotificationOpen withNotification:notification];
  XCTAssertEqualObjects(params[kFIRIParameterMessageName], composerIdentifier);
  XCTAssertNil(params[kFIRIParameterMessageTime]);
  XCTAssertEqualObjects(params[kFIRIParameterMessageDeviceTime], @"12345678");
}

- (void)testConversionTracking {
  // Notification contains "google.c.a.tc" key.
  NSDictionary *notification = @{
                                 @"aps" : @{@"alert" : @"to check the reporting format"},
                                 @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                                 @"gcm.n.e" : @"1",
                                 @"google.c.a.c_l" : @"Hello World",
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.e" : @"1",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"google.c.a.udt" : @"0",
                                 @"google.c.a.m_l" : @"developer's customized label",
                                 @"google.c.a.tc" : @"1",
                                 @"from" : @"/topics/news",
                                 };
  NSDictionary *params = @{
                           kFIRIParameterSource : kReengagementSource,
                           kFIRIParameterMedium : kReengagementMedium,
                           kFIRIParameterCampaign : @"575315420755741863"
                           };
  __block XCTestExpectation *expectation = [self expectationWithDescription:@"completion"];
  FakeAnalytics *analytics = [[FakeAnalytics alloc]
                              initWithEventHandler:^(NSString *origin, NSString *name, NSDictionary *parameters) {
                                XCTAssertEqualObjects(origin, kFIREventOriginFCM);
                                XCTAssertEqualObjects(name, @"_cmp");
                                XCTAssertEqualObjects(parameters, params);
                                [expectation fulfill];
                                expectation = nil;
                              }];
  [FIRMessagingAnalytics logUserPropertyForConversionTracking:notification toAnalytics:analytics];
  [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
}

- (void)testConversionTrackingUserProperty {
  // Notification contains "google.c.a.tc" key.
  NSDictionary *notification = @{
                                 @"aps" : @{@"alert" : @"to check the reporting format"},
                                 @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                                 @"gcm.n.e" : @"1",
                                 @"google.c.a.c_l" : @"Hello World",
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.e" : @"1",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"google.c.a.udt" : @"0",
                                 @"google.c.a.m_l" : @"developer's customized label",
                                 @"google.c.a.tc" : @"1",
                                 @"from" : @"/topics/news",
                                 };

  XCTestExpectation *expectation = [self expectationWithDescription:@"completion"];
  FakeAnalytics *analytics = [[FakeAnalytics alloc]
                              initWithUserPropertyHandler:^(NSString *origin, NSString *name, id value) {
                                XCTAssertEqualObjects(origin, kFIREventOriginFCM);
                                XCTAssertEqualObjects(name, @"_ln");
                                XCTAssertEqualObjects(value, @"575315420755741863");
                                [expectation fulfill];
                              }];
  [FIRMessagingAnalytics logUserPropertyForConversionTracking:notification toAnalytics:analytics];
  [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
}

- (void)testNoConversionTracking {
  // Notification contains "google.c.a.tc" key.
  NSDictionary *notification = @{
                                 @"aps" : @{@"alert" : @"to check the reporting format"},
                                 @"gcm.message_id" : @"0:1522880049414338%944841cd944841cd",
                                 @"gcm.n.e" : @"1",
                                 @"google.c.a.c_l" : @"Hello World",
                                 @"google.c.a.c_id" : @"575315420755741863",
                                 @"google.c.a.e" : @"1",
                                 @"google.c.a.ts" : @"1522880044",
                                 @"google.c.a.udt" : @"0",
                                 @"google.c.a.m_l" : @"developer's customized label",
                                 @"from" : @"/topics/news",
                                 };
  FakeAnalytics *analytics = [[FakeAnalytics alloc]
                              initWithEventHandler:^(NSString *origin, NSString *name, NSDictionary *parameters) {
                                XCTAssertTrue(NO);
                              }];
  [FIRMessagingAnalytics logUserPropertyForConversionTracking:notification toAnalytics:analytics];
}

- (void)testLogMessage {
  NSDictionary *notification = @{
                                 @"google.c.a.e" : @"1",
                                 };
  [FIRMessagingAnalytics logMessage:notification toAnalytics:nil];
  OCMVerify([self.logClassMock logEvent:OCMOCK_ANY withNotification:notification toAnalytics:nil]);
}

- (void)testLogOpenNotification {
  NSDictionary *notification = @{
                                 @"google.c.a.e" : @"1",
                                 };
  [FIRMessagingAnalytics logOpenNotification:notification toAnalytics:nil];

  OCMVerify([self.logClassMock logUserPropertyForConversionTracking:notification toAnalytics:nil]);
  OCMVerify([self.logClassMock logEvent:kFIRIEventNotificationOpen
                       withNotification:notification
                            toAnalytics:nil]);
}

@end