// Copyright 2015 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "components/gcm_driver/gcm_stats_recorder_android.h" #include #include "components/gcm_driver/crypto/gcm_decryption_result.h" #include "testing/gtest/include/gtest/gtest.h" namespace gcm { namespace { const char kTestAppId[] = "test_app_id"; const char kTestSenderId[] = "test_sender_id"; class GCMStatsRecorderAndroidTest : public ::testing::Test, public GCMStatsRecorderAndroid::Delegate { public: GCMStatsRecorderAndroidTest() : activity_recorded_calls_(0) {} ~GCMStatsRecorderAndroidTest() override {} // GCMStatsRecorderAndroid::Delegate implementation. void OnActivityRecorded() override { ++activity_recorded_calls_; } size_t activity_recorded_calls() const { return activity_recorded_calls_; } private: size_t activity_recorded_calls_; }; TEST_F(GCMStatsRecorderAndroidTest, RecordsAndCallsDelegate) { GCMStatsRecorderAndroid recorder(this /* delegate */); recorder.set_is_recording(true); ASSERT_TRUE(recorder.is_recording()); EXPECT_EQ(0u, activity_recorded_calls()); recorder.RecordRegistrationSent(kTestAppId); EXPECT_EQ(1u, activity_recorded_calls()); recorder.RecordRegistrationResponse(kTestAppId, true /* success */); EXPECT_EQ(2u, activity_recorded_calls()); recorder.RecordUnregistrationSent(kTestAppId); EXPECT_EQ(3u, activity_recorded_calls()); recorder.RecordUnregistrationResponse(kTestAppId, true /* success */); EXPECT_EQ(4u, activity_recorded_calls()); recorder.RecordDataMessageReceived(kTestAppId, kTestSenderId, 42 /* message_byte_size */); EXPECT_EQ(5u, activity_recorded_calls()); recorder.RecordDecryptionFailure(kTestAppId, GCMDecryptionResult::INVALID_PAYLOAD); EXPECT_EQ(6u, activity_recorded_calls()); RecordedActivities activities; recorder.CollectActivities(&activities); EXPECT_EQ(4u, activities.registration_activities.size()); EXPECT_EQ(1u, activities.receiving_activities.size()); EXPECT_EQ(1u, activities.decryption_failure_activities.size()); recorder.CollectActivities(&activities); EXPECT_EQ(8u, activities.registration_activities.size()); EXPECT_EQ(2u, activities.receiving_activities.size()); EXPECT_EQ(2u, activities.decryption_failure_activities.size()); recorder.Clear(); RecordedActivities empty_activities; recorder.CollectActivities(&empty_activities); EXPECT_EQ(0u, empty_activities.registration_activities.size()); EXPECT_EQ(0u, empty_activities.receiving_activities.size()); EXPECT_EQ(0u, empty_activities.decryption_failure_activities.size()); } TEST_F(GCMStatsRecorderAndroidTest, NullDelegate) { GCMStatsRecorderAndroid recorder(nullptr /* delegate */); recorder.set_is_recording(true); ASSERT_TRUE(recorder.is_recording()); recorder.RecordRegistrationSent(kTestAppId); RecordedActivities activities; recorder.CollectActivities(&activities); EXPECT_EQ(1u, activities.registration_activities.size()); } TEST_F(GCMStatsRecorderAndroidTest, NotRecording) { GCMStatsRecorderAndroid recorder(this /* delegate */); ASSERT_FALSE(recorder.is_recording()); recorder.RecordRegistrationSent(kTestAppId); RecordedActivities activities; recorder.CollectActivities(&activities); EXPECT_EQ(0u, activities.registration_activities.size()); } } // namespace } // namespace gcm