Replace BrowserProces::AddRefModule/RemoveModule by ScopedKeepAlive

chrome/browser/lifetime/keep_alive_registry_unittest.cc

 // Copyright 2016 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 "chrome/browser/lifetime/keep_alive_registry.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chrome/browser/lifetime/keep_alive_state_observer.h"
 #include "chrome/browser/lifetime/keep_alive_types.h"
@@ skipping 29 matching lines @@
  protected:
   int on_restart_allowed_call_count_;
   int on_restart_forbidden_call_count_;
   KeepAliveRegistry* registry_;
 };
 
 // Test the IsKeepingAlive state and when we interact with the browser with
 // a KeepAlive registered.
 TEST_F(KeepAliveRegistryTest, BasicKeepAliveTest) {
   TestingBrowserProcess* browser_process = TestingBrowserProcess::GetGlobal();
-  const unsigned int base_module_ref_count =
-      browser_process->module_ref_count();
+  EXPECT_FALSE(browser_process->is_pinned());
   KeepAliveRegistry* registry = KeepAliveRegistry::GetInstance();
 
   EXPECT_FALSE(registry->IsKeepingAlive());
 
   {
     // Arbitrarily chosen Origin
     ScopedKeepAlive test_keep_alive(KeepAliveOrigin::CHROME_APP_DELEGATE,
                                     KeepAliveRestartOption::DISABLED);
 
     // We should require the browser to stay alive
-    EXPECT_EQ(base_module_ref_count + 1, browser_process->module_ref_count());
+    EXPECT_TRUE(browser_process->is_pinned());
     EXPECT_TRUE(registry_->IsKeepingAlive());
   }
 
   // We should be back to normal now.
-  EXPECT_EQ(base_module_ref_count, browser_process->module_ref_count());
+  EXPECT_FALSE(browser_process->is_pinned());
   EXPECT_FALSE(registry_->IsKeepingAlive());
 }
 
 // Test the IsKeepingAlive state and when we interact with the browser with
 // more than one KeepAlive registered.
 TEST_F(KeepAliveRegistryTest, DoubleKeepAliveTest) {
   TestingBrowserProcess* browser_process = TestingBrowserProcess::GetGlobal();
-  const unsigned int base_module_ref_count =
-      browser_process->module_ref_count();
+  EXPECT_FALSE(browser_process->is_pinned());
   scoped_ptr<ScopedKeepAlive> keep_alive_1, keep_alive_2;
 
   keep_alive_1.reset(new ScopedKeepAlive(KeepAliveOrigin::CHROME_APP_DELEGATE,
                                          KeepAliveRestartOption::DISABLED));
-  EXPECT_EQ(base_module_ref_count + 1, browser_process->module_ref_count());
+  EXPECT_TRUE(browser_process->is_pinned());
   EXPECT_TRUE(registry_->IsKeepingAlive());
 
   keep_alive_2.reset(new ScopedKeepAlive(KeepAliveOrigin::CHROME_APP_DELEGATE,
                                          KeepAliveRestartOption::DISABLED));
   // We should not increment the count twice
-  EXPECT_EQ(base_module_ref_count + 1, browser_process->module_ref_count());
+  EXPECT_TRUE(browser_process->is_pinned());
   EXPECT_TRUE(registry_->IsKeepingAlive());
 
   keep_alive_1.reset();
   // We should not decrement the count before the last keep alive is released.
-  EXPECT_EQ(base_module_ref_count + 1, browser_process->module_ref_count());
+  EXPECT_TRUE(browser_process->is_pinned());
   EXPECT_TRUE(registry_->IsKeepingAlive());
 
   keep_alive_2.reset();
-  EXPECT_EQ(base_module_ref_count, browser_process->module_ref_count());
+  EXPECT_FALSE(browser_process->is_pinned());
   EXPECT_FALSE(registry_->IsKeepingAlive());
 }
 
 // Test the IsKeepingAlive state and when we interact with the browser with
 // more than one KeepAlive registered.
 TEST_F(KeepAliveRegistryTest, RestartOptionTest) {
   scoped_ptr<ScopedKeepAlive> keep_alive, keep_alive_restart;
 
   EXPECT_EQ(0, on_restart_allowed_call_count_);
   EXPECT_EQ(0, on_restart_forbidden_call_count_);
@@ skipping 13 matching lines @@
   ASSERT_EQ(1, on_restart_allowed_call_count_--);
 
   // No keep alive, we should no prevent restarts.
   keep_alive.reset();
   EXPECT_EQ(0, on_restart_forbidden_call_count_);
 
   // Make sure all calls were checked.
   EXPECT_EQ(0, on_restart_allowed_call_count_);
   EXPECT_EQ(0, on_restart_forbidden_call_count_);
 }