extension_gcm_app_handler_unittest.cc revision 116680a4aac90f2aa7413d9095a592090648e557
1c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch// Copyright 2014 The Chromium Authors. All rights reserved. 2c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch// found in the LICENSE file. 4c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 55c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "chrome/browser/extensions/extension_gcm_app_handler.h" 65c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 75c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include <vector> 85c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 9c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "base/bind.h" 105c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/bind_helpers.h" 11c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "base/command_line.h" 12f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/file_util.h" 135c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/files/file_path.h" 14f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/files/scoped_temp_dir.h" 155c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/location.h" 165c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/logging.h" 175c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/memory/ref_counted.h" 185c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/message_loop/message_loop.h" 19f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/path_service.h" 20cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "base/prefs/pref_service.h" 215c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/run_loop.h" 225c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/values.h" 23f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "chrome/browser/chrome_notification_types.h" 245c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "chrome/browser/extensions/extension_service.h" 25c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/extensions/test_extension_service.h" 26c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/extensions/test_extension_system.h" 275c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "chrome/browser/profiles/profile.h" 285c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "chrome/browser/services/gcm/fake_signin_manager.h" 29c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/services/gcm/gcm_profile_service.h" 30c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/services/gcm/gcm_profile_service_factory.h" 31c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/signin/signin_manager_factory.h" 32f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "chrome/common/chrome_paths.h" 33c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/common/pref_names.h" 34c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/test/base/testing_profile.h" 35cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "components/gcm_driver/fake_gcm_app_handler.h" 36cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "components/gcm_driver/fake_gcm_client.h" 37cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "components/gcm_driver/fake_gcm_client_factory.h" 38cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "components/gcm_driver/gcm_client_factory.h" 39cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "components/gcm_driver/gcm_driver.h" 405c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "components/keyed_service/core/keyed_service.h" 415c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "content/public/browser/browser_context.h" 42c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "content/public/browser/browser_thread.h" 43c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "content/public/test/test_browser_thread_bundle.h" 44f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "content/public/test/test_utils.h" 455c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "extensions/browser/extension_system.h" 46c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "extensions/common/extension.h" 475c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "extensions/common/manifest.h" 48c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "extensions/common/manifest_constants.h" 495c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "extensions/common/permissions/api_permission.h" 5046d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)#include "extensions/common/permissions/permissions_data.h" 51c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "testing/gtest/include/gtest/gtest.h" 52c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 535c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#if !defined(OS_ANDROID) 545c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "chrome/browser/extensions/api/gcm/gcm_api.h" 555c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#endif 565c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 57c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#if defined(OS_CHROMEOS) 58cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "chrome/browser/chromeos/login/users/user_manager.h" 59c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/chromeos/settings/cros_settings.h" 60c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "chrome/browser/chromeos/settings/device_settings_service.h" 61c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#endif 62c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 63c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochnamespace extensions { 64c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 65c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochnamespace { 66c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 67c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochconst char kTestExtensionName[] = "FooBar"; 68c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochconst char kTestingUsername[] = "user1@example.com"; 69c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 70c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} // namespace 71c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 725c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu// Helper class for asynchronous waiting. 735c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liuclass Waiter { 745c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu public: 755c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu Waiter() {} 765c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu ~Waiter() {} 775c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 785c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // Waits until the asynchronous operation finishes. 795c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void WaitUntilCompleted() { 805c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu run_loop_.reset(new base::RunLoop); 815c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu run_loop_->Run(); 825c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 835c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 845c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // Signals that the asynchronous operation finishes. 855c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void SignalCompleted() { 865c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu if (run_loop_ && run_loop_->running()) 875c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu run_loop_->Quit(); 885c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 895c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 905c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // Runs until UI loop becomes idle. 915c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void PumpUILoop() { 925c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu base::MessageLoop::current()->RunUntilIdle(); 935c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 945c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 955c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // Runs until IO loop becomes idle. 965c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void PumpIOLoop() { 975c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::PostTask( 985c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::IO, 995c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu FROM_HERE, 1005c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu base::Bind(&Waiter::OnIOLoopPump, base::Unretained(this))); 1015c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1025c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu WaitUntilCompleted(); 1035c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 1045c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1055c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu private: 1065c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void PumpIOLoopCompleted() { 1075c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); 1085c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1095c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu SignalCompleted(); 1105c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 1115c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1125c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void OnIOLoopPump() { 1135c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 1145c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1155c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::PostTask( 1165c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::IO, 1175c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu FROM_HERE, 1185c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu base::Bind(&Waiter::OnIOLoopPumpCompleted, base::Unretained(this))); 1195c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 1205c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1215c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu void OnIOLoopPumpCompleted() { 1225c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 1235c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1245c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::PostTask( 1255c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu content::BrowserThread::UI, 1265c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu FROM_HERE, 1275c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu base::Bind(&Waiter::PumpIOLoopCompleted, base::Unretained(this))); 1285c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 1295c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1305c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu scoped_ptr<base::RunLoop> run_loop_; 1315c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 1325c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu DISALLOW_COPY_AND_ASSIGN(Waiter); 1335c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu}; 1345c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 135c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochclass FakeExtensionGCMAppHandler : public ExtensionGCMAppHandler { 136c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch public: 137c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch FakeExtensionGCMAppHandler(Profile* profile, Waiter* waiter) 138c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch : ExtensionGCMAppHandler(profile), 139c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_(waiter), 140f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) unregistration_result_(gcm::GCMClient::UNKNOWN_ERROR), 141f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) app_handler_count_drop_to_zero_(false) { 142c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 143c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 144c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual ~FakeExtensionGCMAppHandler() { 145c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 146c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 147c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void OnMessage( 148c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch const std::string& app_id, 1495c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu const gcm::GCMClient::IncomingMessage& message) OVERRIDE { 150c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 151c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 152c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void OnMessagesDeleted(const std::string& app_id) OVERRIDE { 153c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 154c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 155c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void OnSendError( 156c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch const std::string& app_id, 1575c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu const gcm::GCMClient::SendErrorDetails& send_error_details) OVERRIDE { 158c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 159c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 160c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void OnUnregisterCompleted(const std::string& app_id, 1615c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result result) OVERRIDE { 162c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch unregistration_result_ = result; 163c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_->SignalCompleted(); 164c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 165c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 166f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) virtual void RemoveAppHandler(const std::string& app_id) OVERRIDE{ 167f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ExtensionGCMAppHandler::RemoveAppHandler(app_id); 168f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!GetGCMDriver()->app_handlers().size()) 169f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) app_handler_count_drop_to_zero_ = true; 170f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 171f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 1725c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result unregistration_result() const { 173c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return unregistration_result_; 174c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 175f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) bool app_handler_count_drop_to_zero() const { 176f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return app_handler_count_drop_to_zero_; 177f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 178c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 179c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch private: 180c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Waiter* waiter_; 1815c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result unregistration_result_; 182f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) bool app_handler_count_drop_to_zero_; 183c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 184c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch DISALLOW_COPY_AND_ASSIGN(FakeExtensionGCMAppHandler); 185c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch}; 186c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 187c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochclass ExtensionGCMAppHandlerTest : public testing::Test { 188c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch public: 189c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch static KeyedService* BuildGCMProfileService( 190c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch content::BrowserContext* context) { 191cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return new gcm::GCMProfileService( 192cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) Profile::FromBrowserContext(context), 193cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) scoped_ptr<gcm::GCMClientFactory>(new gcm::FakeGCMClientFactory( 194cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) gcm::FakeGCMClient::NO_DELAY_START, 195cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) content::BrowserThread::GetMessageLoopProxyForThread( 196cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) content::BrowserThread::UI), 197cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) content::BrowserThread::GetMessageLoopProxyForThread( 198cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) content::BrowserThread::IO)))); 199c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 200c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 201c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch ExtensionGCMAppHandlerTest() 202c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch : extension_service_(NULL), 2035c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu registration_result_(gcm::GCMClient::UNKNOWN_ERROR), 2045c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu unregistration_result_(gcm::GCMClient::UNKNOWN_ERROR) { 205c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 206c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 207c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual ~ExtensionGCMAppHandlerTest() { 208c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 209c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 210c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Overridden from test::Test: 211c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void SetUp() OVERRIDE { 212f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 213f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 214c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Make BrowserThread work in unittest. 215c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch thread_bundle_.reset(new content::TestBrowserThreadBundle( 216c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch content::TestBrowserThreadBundle::REAL_IO_THREAD)); 217c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 218f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // Allow extension update to unpack crx in process. 219f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) in_process_utility_thread_helper_.reset( 220f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) new content::InProcessUtilityThreadHelper); 221f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 222c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // This is needed to create extension service under CrOS. 223c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#if defined(OS_CHROMEOS) 224c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch test_user_manager_.reset(new chromeos::ScopedTestUserManager()); 225c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#endif 226c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 227c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Create a new profile. 228c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch TestingProfile::Builder builder; 229c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch builder.AddTestingFactory(SigninManagerFactory::GetInstance(), 230c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch gcm::FakeSigninManager::Build); 231c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch profile_ = builder.Build(); 232c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch signin_manager_ = static_cast<gcm::FakeSigninManager*>( 233c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch SigninManagerFactory::GetInstance()->GetForProfile(profile_.get())); 234c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 235c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Create extension service in order to uninstall the extension. 236c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch TestExtensionSystem* extension_system( 237c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch static_cast<TestExtensionSystem*>(ExtensionSystem::Get(profile()))); 238f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::FilePath extensions_install_dir = 239f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) temp_dir_.path().Append(FILE_PATH_LITERAL("Extensions")); 240c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension_system->CreateExtensionService( 241f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) CommandLine::ForCurrentProcess(), extensions_install_dir, false); 242c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension_service_ = extension_system->Get(profile())->extension_service(); 243f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extension_service_->set_extensions_enabled(true); 244f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extension_service_->set_show_extensions_prompts(false); 245f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extension_service_->set_install_updates_when_idle_for_test(false); 246c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 247c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Enable GCM such that tests could be run on all channels. 248c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch profile()->GetPrefs()->SetBoolean(prefs::kGCMChannelEnabled, true); 249c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 250c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Create GCMProfileService that talks with fake GCMClient. 251cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) gcm::GCMProfileServiceFactory::GetInstance()->SetTestingFactoryAndUse( 252cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) profile(), &ExtensionGCMAppHandlerTest::BuildGCMProfileService); 253c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 254c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Create a fake version of ExtensionGCMAppHandler. 255c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch gcm_app_handler_.reset(new FakeExtensionGCMAppHandler(profile(), &waiter_)); 256c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 257c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 258c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual void TearDown() OVERRIDE { 259c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#if defined(OS_CHROMEOS) 260c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch test_user_manager_.reset(); 261c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#endif 262c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 263c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_.PumpUILoop(); 264c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 265c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 266c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Returns a barebones test extension. 267c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_refptr<Extension> CreateExtension() { 268c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch base::DictionaryValue manifest; 269c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch manifest.SetString(manifest_keys::kVersion, "1.0.0.0"); 270c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch manifest.SetString(manifest_keys::kName, kTestExtensionName); 271c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch base::ListValue* permission_list = new base::ListValue; 272c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch permission_list->Append(base::Value::CreateStringValue("gcm")); 273c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch manifest.Set(manifest_keys::kPermissions, permission_list); 274c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 275c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch std::string error; 276c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_refptr<Extension> extension = Extension::Create( 277f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) temp_dir_.path(), 278f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) Manifest::UNPACKED, 279c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch manifest, 280c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Extension::NO_FLAGS, 281f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) "ldnnhddmnhbkjipkidpdiheffobcpfmf", 282c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch &error); 283c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EXPECT_TRUE(extension.get()) << error; 28446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) EXPECT_TRUE( 28546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) extension->permissions_data()->HasAPIPermission(APIPermission::kGcm)); 286c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 287c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return extension; 288c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 289c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 290c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void LoadExtension(const Extension* extension) { 291c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension_service_->AddExtension(extension); 292c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 293c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 294f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) static bool IsCrxInstallerDone(extensions::CrxInstaller** installer, 295f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) const content::NotificationSource& source, 296f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) const content::NotificationDetails& details) { 297f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return content::Source<extensions::CrxInstaller>(source).ptr() == 298f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) *installer; 299f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 300f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 301f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) void UpdateExtension(const Extension* extension, 302f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) const std::string& update_crx) { 303f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::FilePath data_dir; 304f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!PathService::Get(chrome::DIR_TEST_DATA, &data_dir)) { 305f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ADD_FAILURE(); 306f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return; 307f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 308f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) data_dir = data_dir.AppendASCII("extensions"); 309f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) data_dir = data_dir.AppendASCII(update_crx); 310f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 311f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::FilePath path = temp_dir_.path(); 312f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) path = path.Append(data_dir.BaseName()); 313f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_TRUE(base::CopyFile(data_dir, path)); 314f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 315f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extensions::CrxInstaller* installer = NULL; 316f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) content::WindowedNotificationObserver observer( 317f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) chrome::NOTIFICATION_CRX_INSTALLER_DONE, 318f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::Bind(&IsCrxInstallerDone, &installer)); 319f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extension_service_->UpdateExtension( 320f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) extension->id(), path, true, &installer); 321f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 322f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (installer) 323f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) observer.Wait(); 324f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 325f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 326c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void DisableExtension(const Extension* extension) { 327c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension_service_->DisableExtension( 328c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension->id(), Extension::DISABLE_USER_ACTION); 329c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 330c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 331c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void EnableExtension(const Extension* extension) { 332c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch extension_service_->EnableExtension(extension->id()); 333c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 334c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 335c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void UninstallExtension(const Extension* extension) { 336116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch extension_service_->UninstallExtension( 337116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch extension->id(), ExtensionService::UNINSTALL_REASON_FOR_TESTING, NULL); 338c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 339c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 340c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void SignIn(const std::string& username) { 341c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch signin_manager_->SignIn(username); 342c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_.PumpIOLoop(); 343c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 344c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 345c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void SignOut() { 346116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch signin_manager_->SignOut(signin_metrics::SIGNOUT_TEST); 347c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_.PumpIOLoop(); 348c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 349c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 350c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void Register(const std::string& app_id, 351c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch const std::vector<std::string>& sender_ids) { 352cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) GetGCMDriver()->Register( 353c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch app_id, 354c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch sender_ids, 355c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch base::Bind(&ExtensionGCMAppHandlerTest::RegisterCompleted, 356c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch base::Unretained(this))); 357c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 358c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 359c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void RegisterCompleted(const std::string& registration_id, 3605c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result result) { 361c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch registration_result_ = result; 362c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter_.SignalCompleted(); 363c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 364c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 365cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) gcm::GCMDriver* GetGCMDriver() const { 366cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return gcm::GCMProfileServiceFactory::GetForProfile(profile())->driver(); 367c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 368c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 369c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch bool HasAppHandlers(const std::string& app_id) const { 370cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) return GetGCMDriver()->app_handlers().count(app_id); 371c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 372c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 373c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Profile* profile() const { return profile_.get(); } 374c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Waiter* waiter() { return &waiter_; } 375c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch FakeExtensionGCMAppHandler* gcm_app_handler() const { 376c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return gcm_app_handler_.get(); 377c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 3785c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result registration_result() const { 3795c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu return registration_result_; 3805c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 3815c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result unregistration_result() const { 382c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return unregistration_result_; 383c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 384c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 385c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch private: 386c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_ptr<content::TestBrowserThreadBundle> thread_bundle_; 387f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_ptr<content::InProcessUtilityThreadHelper> 388f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) in_process_utility_thread_helper_; 389c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_ptr<TestingProfile> profile_; 390c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch ExtensionService* extension_service_; // Not owned. 391c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch gcm::FakeSigninManager* signin_manager_; // Not owned. 392f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::ScopedTempDir temp_dir_; 393c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 394c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // This is needed to create extension service under CrOS. 395c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#if defined(OS_CHROMEOS) 396c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch chromeos::ScopedTestDeviceSettingsService test_device_settings_service_; 397c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch chromeos::ScopedTestCrosSettings test_cros_settings_; 398c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_ptr<chromeos::ScopedTestUserManager> test_user_manager_; 399c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#endif 400c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 401c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Waiter waiter_; 402c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_ptr<FakeExtensionGCMAppHandler> gcm_app_handler_; 4035c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result registration_result_; 4045c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm::GCMClient::Result unregistration_result_; 405c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 406c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch DISALLOW_COPY_AND_ASSIGN(ExtensionGCMAppHandlerTest); 407c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch}; 408c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 409c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen MurdochTEST_F(ExtensionGCMAppHandlerTest, AddAndRemoveAppHandler) { 410c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_refptr<Extension> extension(CreateExtension()); 411c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 412c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // App handler is added when extension is loaded. 413c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch LoadExtension(extension); 414c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->PumpUILoop(); 415c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EXPECT_TRUE(HasAppHandlers(extension->id())); 416c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 417c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // App handler is removed when extension is unloaded. 418c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch DisableExtension(extension); 419c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->PumpUILoop(); 420c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EXPECT_FALSE(HasAppHandlers(extension->id())); 421c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 422c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // App handler is added when extension is reloaded. 423c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EnableExtension(extension); 424c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->PumpUILoop(); 425c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EXPECT_TRUE(HasAppHandlers(extension->id())); 426c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 427c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // App handler is removed when extension is uninstalled. 428c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch UninstallExtension(extension); 429c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->PumpUILoop(); 430c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch EXPECT_FALSE(HasAppHandlers(extension->id())); 431c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 432c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 433c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen MurdochTEST_F(ExtensionGCMAppHandlerTest, UnregisterOnExtensionUninstall) { 434c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch scoped_refptr<Extension> extension(CreateExtension()); 435c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch LoadExtension(extension); 436c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 437c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Sign-in is needed for registration. 438c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch SignIn(kTestingUsername); 439c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 440c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Kick off registration. 441c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch std::vector<std::string> sender_ids; 442c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch sender_ids.push_back("sender1"); 443c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch Register(extension->id(), sender_ids); 444c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->WaitUntilCompleted(); 4455c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(gcm::GCMClient::SUCCESS, registration_result()); 446c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 447cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Add another app handler in order to prevent the GCM service from being 448cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // stopped when the extension is uninstalled. This is needed because otherwise 449cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // we are not able to receive the unregistration result. 450cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) GetGCMDriver()->AddAppHandler("Foo", gcm_app_handler()); 451cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 452c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch // Unregistration should be triggered when the extension is uninstalled. 453c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch UninstallExtension(extension); 454c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch waiter()->WaitUntilCompleted(); 4555c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(gcm::GCMClient::SUCCESS, 4565c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu gcm_app_handler()->unregistration_result()); 457cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 458cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Clean up. 459cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) GetGCMDriver()->RemoveAppHandler("Foo"); 460c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} 461c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 462f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)TEST_F(ExtensionGCMAppHandlerTest, UpdateExtensionWithGcmPermissionKept) { 463f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_refptr<Extension> extension(CreateExtension()); 464f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 465f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // App handler is added when the extension is loaded. 466f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) LoadExtension(extension); 467f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) waiter()->PumpUILoop(); 468f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_TRUE(HasAppHandlers(extension->id())); 469f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 470f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // App handler count should not drop to zero when the extension is updated. 471f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) UpdateExtension(extension, "gcm2.crx"); 472f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) waiter()->PumpUILoop(); 473f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_FALSE(gcm_app_handler()->app_handler_count_drop_to_zero()); 474f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_TRUE(HasAppHandlers(extension->id())); 475f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 476f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 477f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)TEST_F(ExtensionGCMAppHandlerTest, UpdateExtensionWithGcmPermissionRemoved) { 478f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_refptr<Extension> extension(CreateExtension()); 479f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 480f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // App handler is added when the extension is loaded. 481f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) LoadExtension(extension); 482f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) waiter()->PumpUILoop(); 483f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_TRUE(HasAppHandlers(extension->id())); 484f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 485f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // App handler is removed when the extension is updated to the version that 486f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // has GCM permission removed. 487f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) UpdateExtension(extension, "good2.crx"); 488f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) waiter()->PumpUILoop(); 489f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_TRUE(gcm_app_handler()->app_handler_count_drop_to_zero()); 490f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_FALSE(HasAppHandlers(extension->id())); 491f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 492f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 493c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch} // namespace extensions 494