| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/proximity_auth/cryptauth/cryptauth_gcm_manager_impl.h" | |
| 6 | |
| 7 #include "base/macros.h" | |
| 8 #include "components/gcm_driver/fake_gcm_driver.h" | |
| 9 #include "components/gcm_driver/gcm_client.h" | |
| 10 #include "components/prefs/testing_pref_service.h" | |
| 11 #include "components/proximity_auth/cryptauth/pref_names.h" | |
| 12 #include "testing/gmock/include/gmock/gmock.h" | |
| 13 #include "testing/gtest/include/gtest/gtest.h" | |
| 14 | |
| 15 using ::testing::_; | |
| 16 using ::testing::SaveArg; | |
| 17 | |
| 18 namespace proximity_auth { | |
| 19 | |
| 20 namespace { | |
| 21 | |
| 22 const char kCryptAuthGCMAppId[] = "com.google.chrome.cryptauth"; | |
| 23 const char kCryptAuthGCMSenderId[] = "381449029288"; | |
| 24 const char kExistingGCMRegistrationId[] = "cirrus"; | |
| 25 const char kNewGCMRegistrationId[] = "stratus"; | |
| 26 const char kCryptAuthMessageCollapseKey[] = | |
| 27 "collapse_cryptauth_sync_DEVICES_SYNC"; | |
| 28 | |
| 29 // Mock GCMDriver implementation for testing. | |
| 30 class MockGCMDriver : public gcm::FakeGCMDriver { | |
| 31 public: | |
| 32 MockGCMDriver() {} | |
| 33 ~MockGCMDriver() override {} | |
| 34 | |
| 35 MOCK_METHOD2(AddAppHandler, | |
| 36 void(const std::string& app_id, gcm::GCMAppHandler* handler)); | |
| 37 | |
| 38 MOCK_METHOD2(RegisterImpl, | |
| 39 void(const std::string& app_id, | |
| 40 const std::vector<std::string>& sender_ids)); | |
| 41 | |
| 42 using gcm::GCMDriver::RegisterFinished; | |
| 43 | |
| 44 private: | |
| 45 DISALLOW_COPY_AND_ASSIGN(MockGCMDriver); | |
| 46 }; | |
| 47 | |
| 48 } // namespace | |
| 49 | |
| 50 class ProximityAuthCryptAuthGCMManagerImplTest | |
| 51 : public testing::Test, | |
| 52 public CryptAuthGCMManager::Observer { | |
| 53 protected: | |
| 54 ProximityAuthCryptAuthGCMManagerImplTest() | |
| 55 : gcm_manager_(&gcm_driver_, &pref_service_) {} | |
| 56 | |
| 57 // testing::Test: | |
| 58 void SetUp() override { | |
| 59 CryptAuthGCMManager::RegisterPrefs(pref_service_.registry()); | |
| 60 gcm_manager_.AddObserver(this); | |
| 61 EXPECT_CALL(gcm_driver_, AddAppHandler(kCryptAuthGCMAppId, &gcm_manager_)); | |
| 62 gcm_manager_.StartListening(); | |
| 63 } | |
| 64 | |
| 65 void TearDown() override { gcm_manager_.RemoveObserver(this); } | |
| 66 | |
| 67 void RegisterWithGCM(gcm::GCMClient::Result registration_result) { | |
| 68 std::vector<std::string> sender_ids; | |
| 69 EXPECT_CALL(gcm_driver_, RegisterImpl(kCryptAuthGCMAppId, _)) | |
| 70 .WillOnce(SaveArg<1>(&sender_ids)); | |
| 71 gcm_manager_.RegisterWithGCM(); | |
| 72 | |
| 73 ASSERT_EQ(1u, sender_ids.size()); | |
| 74 EXPECT_EQ(kCryptAuthGCMSenderId, sender_ids[0]); | |
| 75 | |
| 76 bool success = (registration_result == gcm::GCMClient::SUCCESS); | |
| 77 EXPECT_CALL(*this, OnGCMRegistrationResultProxy(success)); | |
| 78 gcm_driver_.RegisterFinished(kCryptAuthGCMAppId, kNewGCMRegistrationId, | |
| 79 registration_result); | |
| 80 } | |
| 81 | |
| 82 // CryptAuthGCMManager::Observer: | |
| 83 void OnGCMRegistrationResult(bool success) override { | |
| 84 OnGCMRegistrationResultProxy(success); | |
| 85 } | |
| 86 | |
| 87 void OnReenrollMessage() override { OnReenrollMessageProxy(); } | |
| 88 | |
| 89 void OnResyncMessage() override { OnResyncMessageProxy(); } | |
| 90 | |
| 91 MOCK_METHOD1(OnGCMRegistrationResultProxy, void(bool)); | |
| 92 MOCK_METHOD0(OnReenrollMessageProxy, void()); | |
| 93 MOCK_METHOD0(OnResyncMessageProxy, void()); | |
| 94 | |
| 95 testing::StrictMock<MockGCMDriver> gcm_driver_; | |
| 96 | |
| 97 TestingPrefServiceSimple pref_service_; | |
| 98 | |
| 99 CryptAuthGCMManagerImpl gcm_manager_; | |
| 100 | |
| 101 DISALLOW_COPY_AND_ASSIGN(ProximityAuthCryptAuthGCMManagerImplTest); | |
| 102 }; | |
| 103 | |
| 104 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, RegisterPrefs) { | |
| 105 TestingPrefServiceSimple pref_service; | |
| 106 CryptAuthGCMManager::RegisterPrefs(pref_service.registry()); | |
| 107 EXPECT_TRUE(pref_service.FindPreference(prefs::kCryptAuthGCMRegistrationId)); | |
| 108 } | |
| 109 | |
| 110 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, RegistrationSucceeds) { | |
| 111 EXPECT_EQ(std::string(), gcm_manager_.GetRegistrationId()); | |
| 112 RegisterWithGCM(gcm::GCMClient::SUCCESS); | |
| 113 EXPECT_EQ(kNewGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 114 } | |
| 115 | |
| 116 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, | |
| 117 RegistrationSucceedsWithExistingRegistration) { | |
| 118 pref_service_.SetString(prefs::kCryptAuthGCMRegistrationId, | |
| 119 kExistingGCMRegistrationId); | |
| 120 EXPECT_EQ(kExistingGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 121 RegisterWithGCM(gcm::GCMClient::SUCCESS); | |
| 122 EXPECT_EQ(kNewGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 123 EXPECT_EQ(kNewGCMRegistrationId, | |
| 124 pref_service_.GetString(prefs::kCryptAuthGCMRegistrationId)); | |
| 125 } | |
| 126 | |
| 127 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, RegisterWithGCMFails) { | |
| 128 EXPECT_EQ(std::string(), gcm_manager_.GetRegistrationId()); | |
| 129 RegisterWithGCM(gcm::GCMClient::SERVER_ERROR); | |
| 130 EXPECT_EQ(std::string(), gcm_manager_.GetRegistrationId()); | |
| 131 EXPECT_EQ(std::string(), | |
| 132 pref_service_.GetString(prefs::kCryptAuthGCMRegistrationId)); | |
| 133 } | |
| 134 | |
| 135 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, | |
| 136 RegisterWithGCMFailsWithExistingRegistration) { | |
| 137 pref_service_.SetString(prefs::kCryptAuthGCMRegistrationId, | |
| 138 kExistingGCMRegistrationId); | |
| 139 EXPECT_EQ(kExistingGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 140 RegisterWithGCM(gcm::GCMClient::SERVER_ERROR); | |
| 141 EXPECT_EQ(kExistingGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 142 EXPECT_EQ(kExistingGCMRegistrationId, | |
| 143 pref_service_.GetString(prefs::kCryptAuthGCMRegistrationId)); | |
| 144 } | |
| 145 | |
| 146 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, | |
| 147 RegistrationFailsThenSucceeds) { | |
| 148 EXPECT_EQ(std::string(), gcm_manager_.GetRegistrationId()); | |
| 149 RegisterWithGCM(gcm::GCMClient::NETWORK_ERROR); | |
| 150 EXPECT_EQ(std::string(), gcm_manager_.GetRegistrationId()); | |
| 151 RegisterWithGCM(gcm::GCMClient::SUCCESS); | |
| 152 EXPECT_EQ(kNewGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 153 } | |
| 154 | |
| 155 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, ConcurrentRegistrations) { | |
| 156 // If multiple RegisterWithGCM() calls are made concurrently, only one | |
| 157 // registration attempt should actually be made. | |
| 158 EXPECT_CALL(gcm_driver_, RegisterImpl(kCryptAuthGCMAppId, _)); | |
| 159 gcm_manager_.RegisterWithGCM(); | |
| 160 gcm_manager_.RegisterWithGCM(); | |
| 161 gcm_manager_.RegisterWithGCM(); | |
| 162 | |
| 163 EXPECT_CALL(*this, OnGCMRegistrationResultProxy(true)); | |
| 164 gcm_driver_.RegisterFinished(kCryptAuthGCMAppId, kNewGCMRegistrationId, | |
| 165 gcm::GCMClient::SUCCESS); | |
| 166 EXPECT_EQ(kNewGCMRegistrationId, gcm_manager_.GetRegistrationId()); | |
| 167 } | |
| 168 | |
| 169 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, ReenrollmentMessagesReceived) { | |
| 170 EXPECT_CALL(*this, OnReenrollMessageProxy()).Times(2); | |
| 171 | |
| 172 gcm::IncomingMessage message; | |
| 173 message.data["registrationTickleType"] = "1"; // FORCE_ENROLLMENT | |
| 174 message.collapse_key = kCryptAuthMessageCollapseKey; | |
| 175 message.sender_id = kCryptAuthGCMSenderId; | |
| 176 | |
| 177 gcm::GCMAppHandler* gcm_app_handler = | |
| 178 static_cast<gcm::GCMAppHandler*>(&gcm_manager_); | |
| 179 gcm_app_handler->OnMessage(kCryptAuthGCMAppId, message); | |
| 180 message.data["registrationTickleType"] = "2"; // UPDATE_ENROLLMENT | |
| 181 gcm_app_handler->OnMessage(kCryptAuthGCMAppId, message); | |
| 182 } | |
| 183 | |
| 184 TEST_F(ProximityAuthCryptAuthGCMManagerImplTest, ResyncMessagesReceived) { | |
| 185 EXPECT_CALL(*this, OnResyncMessageProxy()).Times(2); | |
| 186 | |
| 187 gcm::IncomingMessage message; | |
| 188 message.data["registrationTickleType"] = "3"; // DEVICES_SYNC | |
| 189 message.collapse_key = kCryptAuthMessageCollapseKey; | |
| 190 message.sender_id = kCryptAuthGCMSenderId; | |
| 191 | |
| 192 gcm::GCMAppHandler* gcm_app_handler = | |
| 193 static_cast<gcm::GCMAppHandler*>(&gcm_manager_); | |
| 194 gcm_app_handler->OnMessage(kCryptAuthGCMAppId, message); | |
| 195 gcm_app_handler->OnMessage(kCryptAuthGCMAppId, message); | |
| 196 } | |
| 197 | |
| 198 } // namespace proximity_auth | |
| OLD | NEW |