Add test case for unloading Easy Unlock app when system suspends.

chrome/browser/signin/easy_unlock_service_browsertest_chromeos.cc

 // Copyright 2014 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 "base/command_line.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/values.h"
 #include "chrome/browser/chromeos/login/login_manager_test.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/ui/user_adding_screen.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/easy_unlock_service.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/fake_power_manager_client.h"
 #include "components/policy/core/browser/browser_policy_connector.h"
 #include "components/policy/core/common/mock_configuration_policy_provider.h"
 #include "components/policy/core/common/policy_map.h"
 #include "components/policy/core/common/policy_types.h"
 #include "components/user_manager/user_manager.h"
 #include "content/public/common/content_switches.h"
 #include "device/bluetooth/bluetooth_adapter_factory.h"
 #include "device/bluetooth/test/mock_bluetooth_adapter.h"
 #include "extensions/browser/extension_system.h"
 #include "policy/policy_constants.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
+using chromeos::DBusThreadManagerSetter;
+using chromeos::FakePowerManagerClient;
+using chromeos::PowerManagerClient;
 using chromeos::ProfileHelper;
 using chromeos::LoginManagerTest;
 using chromeos::StartupUtils;
 using chromeos::UserAddingScreen;
 using user_manager::UserManager;
 using device::MockBluetoothAdapter;
 using testing::_;
 using testing::Return;
 
 namespace {
@@ skipping 52 matching lines @@
 #endif
 
   // InProcessBrowserTest:
   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
     EXPECT_CALL(provider_, IsInitializationComplete(_))
         .WillRepeatedly(Return(true));
     policy::BrowserPolicyConnector::SetPolicyProviderForTesting(&provider_);
 
     mock_adapter_ = new testing::NiceMock<MockBluetoothAdapter>();
     SetUpBluetoothMock(mock_adapter_, is_bluetooth_adapter_present_);
+
+    scoped_ptr<DBusThreadManagerSetter> dbus_setter =
+        chromeos::DBusThreadManager::GetSetterForTesting();
+    power_manager_client_ = new FakePowerManagerClient;
+    dbus_setter->SetPowerManagerClient(
+        scoped_ptr<PowerManagerClient>(power_manager_client_));
   }
 
   Profile* profile() const { return browser()->profile(); }
 
   EasyUnlockService* service() const {
     return EasyUnlockService::Get(profile());
   }
 
   void set_is_bluetooth_adapter_present(bool is_present) {
     is_bluetooth_adapter_present_ = is_present;
   }
 
+  FakePowerManagerClient* power_manager_client() {
+    return power_manager_client_;
+  }
+
  private:
   policy::MockConfigurationPolicyProvider provider_;
   scoped_refptr<testing::NiceMock<MockBluetoothAdapter> > mock_adapter_;
   bool is_bluetooth_adapter_present_;
+  FakePowerManagerClient* power_manager_client_;
 
   DISALLOW_COPY_AND_ASSIGN(EasyUnlockServiceTest);
 };
 
 // Tests that EasyUnlock is on by default.
 IN_PROC_BROWSER_TEST_F(EasyUnlockServiceTest, DefaultOn) {
   EXPECT_TRUE(service()->IsAllowed());
 #if defined(GOOGLE_CHROME_BUILD)
   EXPECT_TRUE(HasEasyUnlockApp());
 #endif
 }
 
+#if defined(GOOGLE_CHROME_BUILD)
+IN_PROC_BROWSER_TEST_F(EasyUnlockServiceTest, UnloadsOnSuspend) {
+  EXPECT_TRUE(HasEasyUnlockApp());
+  power_manager_client()->SendSuspendImminent();
+  EXPECT_FALSE(HasEasyUnlockApp());
+  power_manager_client()->SendSuspendDone();
+  EXPECT_TRUE(HasEasyUnlockApp());
+}
+#endif
+
 class EasyUnlockServiceNoBluetoothTest : public EasyUnlockServiceTest {
  public:
   EasyUnlockServiceNoBluetoothTest() {}
   virtual ~EasyUnlockServiceNoBluetoothTest() {}
 
   // InProcessBrowserTest:
   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
     set_is_bluetooth_adapter_present(false);
     EasyUnlockServiceTest::SetUpInProcessBrowserTestFixture();
   }
@@ skipping 117 matching lines @@
   Profile* secondary_profile = ProfileHelper::Get()->GetProfileByUserIdHash(
       secondary_user->username_hash());
 
   EXPECT_TRUE(EasyUnlockService::Get(primary_profile)->IsAllowed());
   EXPECT_FALSE(EasyUnlockService::Get(secondary_profile)->IsAllowed());
 #if defined(GOOGLE_CHROME_BUILD)
   EXPECT_TRUE(HasEasyUnlockAppForProfile(primary_profile));
   EXPECT_FALSE(HasEasyUnlockAppForProfile(secondary_profile));
 #endif
 }