Add TabLoaderTest

chrome/browser/sessions/tab_loader.h

 // 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.
 
 #ifndef CHROME_BROWSER_SESSIONS_TAB_LOADER_H_
 #define CHROME_BROWSER_SESSIONS_TAB_LOADER_H_
 
 #include <stddef.h>
 
 #include <list>
 #include <memory>
 #include <set>
 
+#include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/memory/memory_coordinator_client.h"
 #include "base/memory/memory_pressure_listener.h"
 #include "base/timer/timer.h"
 #include "chrome/browser/sessions/session_restore_delegate.h"
 #include "chrome/browser/sessions/tab_loader_delegate.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 
 namespace content {
 class NavigationController;
 class RenderWidgetHost;
 }
 
 class SessionRestoreStatsCollector;
+class TabLoaderTest;
 
 // TabLoader is responsible for loading tabs after session restore has finished
 // creating all the tabs. Tabs are loaded after a previously tab finishes
 // loading or a timeout is reached. If the timeout is reached before a tab
 // finishes loading the timeout delay is doubled.
 //
 // TabLoader keeps a reference to itself when it's loading. When it has finished
 // loading, it drops the reference. If another profile is restored while the
 // TabLoader is loading, it will schedule its tabs to get loaded by the same
 // TabLoader. When doing the scheduling, it holds a reference to the TabLoader.
@@ skipping 14 matching lines @@
                const content::NotificationDetails& details) override;
 
   // TabLoaderCallback:
   void SetTabLoadingEnabled(bool enable_tab_loading) override;
 
   // Called to start restoring tabs.
   static void RestoreTabs(const std::vector<RestoredTab>& tabs_,
                           const base::TimeTicks& restore_started);
 
  private:
+  FRIEND_TEST_ALL_PREFIXES(TabLoaderTest, OnMemoryStateChange);
+
   friend class base::RefCounted<TabLoader>;
 
   using TabsLoading = std::set<content::NavigationController*>;
   using TabsToLoad = std::list<content::NavigationController*>;
 
+  static bool GetLoadingEnabledForTesting();

[sky, 2016/11/01 15:23:14]

As the test is a friend, can't it call shared_tab_loader_->loading_enabled_
directly? That way you don't need to expose awkward test only functions like
this.

[hajimehoshi, 2016/11/02 09:34:13]

Done.

+
   explicit TabLoader(base::TimeTicks restore_started);
   ~TabLoader() override;
 
   // This is invoked once by RestoreTabs to start loading.
   void StartLoading(const std::vector<RestoredTab>& tabs);
 
   // Loads the next tab. If there are no more tabs to load this deletes itself,
   // otherwise |force_load_timer_| is restarted.
   void LoadNextTab();
 
@@ skipping 75 matching lines @@
   // explicitly referenced so that it can be notified of deferred tab loads due
   // to memory pressure.
   scoped_refptr<SessionRestoreStatsCollector> stats_collector_;
 
   static TabLoader* shared_tab_loader_;
 
   DISALLOW_COPY_AND_ASSIGN(TabLoader);
 };
 
 #endif  // CHROME_BROWSER_SESSIONS_TAB_LOADER_H_