Home | History | Annotate | Download | only in performance 
      1 // Copyright (c) 2011 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 "base/strings/stringprintf.h"
      6 #include "chrome/browser/sync/profile_sync_service_harness.h"
      7 #include "chrome/browser/sync/test/integration/performance/sync_timing_helper.h"
      8 #include "chrome/browser/sync/test/integration/sessions_helper.h"
      9 #include "chrome/browser/sync/test/integration/sync_test.h"
     10 #include "chrome/browser/ui/browser.h"
     11 #include "chrome/browser/ui/browser_commands.h"
     12 #include "chrome/browser/ui/tabs/tab_strip_model.h"
     13 
     14 using content::OpenURLParams;
     15 using sessions_helper::GetLocalSession;
     16 using sessions_helper::GetSessionData;
     17 using sessions_helper::OpenMultipleTabs;
     18 using sessions_helper::SyncedSessionVector;
     19 using sessions_helper::SessionWindowMap;
     20 using sessions_helper::WaitForTabsToLoad;
     21 
     22 static const int kNumTabs = 150;
     23 
     24 class SessionsSyncPerfTest: public SyncTest {
     25  public:
     26   SessionsSyncPerfTest() : SyncTest(TWO_CLIENT), url_number_(0) {}
     27 
     28   // Opens |num_tabs| new tabs on |profile|.
     29   void AddTabs(int profile, int num_tabs);
     30 
     31   // Update all tabs in |profile| by visiting a new URL.
     32   void UpdateTabs(int profile);
     33 
     34   // Close all tabs in |profile|.
     35   void RemoveTabs(int profile);
     36 
     37   // Returns the number of open tabs in all sessions (local + foreign) for
     38   // |profile|.  Returns -1 on failure.
     39   int GetTabCount(int profile);
     40 
     41  private:
     42   // Returns a new unique URL.
     43   GURL NextURL();
     44 
     45   // Returns a unique URL according to the integer |n|.
     46   GURL IntToURL(int n);
     47 
     48   int url_number_;
     49   DISALLOW_COPY_AND_ASSIGN(SessionsSyncPerfTest);
     50 };
     51 
     52 void SessionsSyncPerfTest::AddTabs(int profile, int num_tabs) {
     53   std::vector<GURL> urls;
     54   for (int i = 0; i < num_tabs; ++i) {
     55     urls.push_back(NextURL());
     56   }
     57   OpenMultipleTabs(profile, urls);
     58 }
     59 
     60 void SessionsSyncPerfTest::UpdateTabs(int profile) {
     61   Browser* browser = GetBrowser(profile);
     62   GURL url;
     63   std::vector<GURL> urls;
     64   for (int i = 0; i < browser->tab_strip_model()->count(); ++i) {
     65     chrome::SelectNumberedTab(browser, i);
     66     url = NextURL();
     67     browser->OpenURL(
     68         OpenURLParams(url,
     69         content::Referrer(GURL("http://localhost"),
     70                           WebKit::WebReferrerPolicyDefault),
     71         CURRENT_TAB,
     72         content::PageTransitionFromInt(0), false));
     73     urls.push_back(url);
     74   }
     75   WaitForTabsToLoad(profile, urls);
     76 }
     77 
     78 void SessionsSyncPerfTest::RemoveTabs(int profile) {
     79   GetBrowser(profile)->tab_strip_model()->CloseAllTabs();
     80 }
     81 
     82 int SessionsSyncPerfTest::GetTabCount(int profile) {
     83   int tab_count = 0;
     84   const browser_sync::SyncedSession* local_session;
     85   SyncedSessionVector sessions;
     86 
     87   if (!GetLocalSession(profile, &local_session)) {
     88     DVLOG(1) << "GetLocalSession returned false";
     89     return -1;
     90   }
     91 
     92   if (!GetSessionData(profile, &sessions)) {
     93     // Foreign session data may be empty.  In this case we only count tabs in
     94     // the local session.
     95     DVLOG(1) << "GetSessionData returned false";
     96   }
     97 
     98   sessions.push_back(local_session);
     99   for (SyncedSessionVector::const_iterator it = sessions.begin();
    100        it != sessions.end(); ++it) {
    101     for (SessionWindowMap::const_iterator win_it = (*it)->windows.begin();
    102          win_it != (*it)->windows.end();
    103          ++win_it) {
    104       tab_count += win_it->second->tabs.size();
    105     }
    106   }
    107   return tab_count;
    108 }
    109 
    110 GURL SessionsSyncPerfTest::NextURL() {
    111   return IntToURL(url_number_++);
    112 }
    113 
    114 GURL SessionsSyncPerfTest::IntToURL(int n) {
    115   return GURL(base::StringPrintf("http://localhost/%d", n));
    116 }
    117 
    118 // TODO(lipalani): Re-enable after crbug.com/96921 is fixed.
    119 IN_PROC_BROWSER_TEST_F(SessionsSyncPerfTest, DISABLED_P0) {
    120   ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
    121 
    122   AddTabs(0, kNumTabs);
    123   base::TimeDelta dt =
    124       SyncTimingHelper::TimeMutualSyncCycle(GetClient(0), GetClient(1));
    125   ASSERT_EQ(kNumTabs, GetTabCount(0));
    126   ASSERT_EQ(kNumTabs, GetTabCount(1));
    127   SyncTimingHelper::PrintResult("tabs", "add_tabs", dt);
    128 
    129   UpdateTabs(0);
    130   dt = SyncTimingHelper::TimeMutualSyncCycle(GetClient(0), GetClient(1));
    131   ASSERT_EQ(kNumTabs, GetTabCount(0));
    132   ASSERT_EQ(kNumTabs, GetTabCount(1));
    133   SyncTimingHelper::PrintResult("tabs", "update_tabs", dt);
    134 
    135   RemoveTabs(0);
    136   dt = SyncTimingHelper::TimeMutualSyncCycle(GetClient(0), GetClient(1));
    137   // New tab page remains open on profile 0 after closing all tabs.
    138   ASSERT_EQ(1, GetTabCount(0));
    139   ASSERT_EQ(0, GetTabCount(1));
    140   SyncTimingHelper::PrintResult("tabs", "delete_tabs", dt);
    141 }
    142