[System Health] Add browsing user stories for pinterest and tumblr.

tools/perf/page_sets/system_health/browsing_stories.py

 # 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.
 
 from page_sets.system_health import platforms
 from page_sets.system_health import system_health_story
 
+from page_sets.login_helpers import pinterest_login
+
 from telemetry import decorators
 
 
 class _BrowsingStory(system_health_story.SystemHealthStory):
   """Abstract base class for browsing stories.
 
   A browsing story visits items on the main page. Subclasses provide
   CSS selector to identify the items and implement interaction using
   the helper methods of this class.
   """
@@ skipping 209 matching lines @@
   A media story imitates browsing a website with photo or video content:
   1. Load a page showing a media item
   2. Click on the next link to go to the next media item
   3. etc.
   """
 
   ABSTRACT_STORY = True
   ITEM_VIEW_TIME_IN_SECONDS = 3
   ITEMS_TO_VISIT = 15
   ITEM_SELECTOR_INDEX = 0
+  INCREMENT_INDEX_AFTER_EACH_ITEM = False
 
   def _DidLoadDocument(self, action_runner):
+    index = self.ITEM_SELECTOR_INDEX
     for _ in xrange(self.ITEMS_TO_VISIT):
-      self._NavigateToItem(action_runner, self.ITEM_SELECTOR_INDEX)
-      self._ViewMediaItem(action_runner)
+      self._NavigateToItem(action_runner, index)
+      self._ViewMediaItem(action_runner, index)
+      if self.INCREMENT_INDEX_AFTER_EACH_ITEM:
+        index += 1
 
-  def _ViewMediaItem(self, action_runner):
+
+  def _ViewMediaItem(self, action_runner, index):
+    del index  # Unused.
     action_runner.tab.WaitForDocumentReadyStateToBeComplete()
     action_runner.Wait(self.ITEM_VIEW_TIME_IN_SECONDS)
 
 
 class ImgurMobileStory(_MediaBrowsingStory):
   NAME = 'browse:media:imgur'
   URL = 'http://imgur.com/gallery/5UlBN'
   ITEM_SELECTOR = '.Navbar-customAction'
   SUPPORTED_PLATFORMS = platforms.MOBILE_ONLY
   IS_SINGLE_PAGE_APP = True
@@ skipping 40 matching lines @@
 
 class FacebookPhotosDesktopStory(_MediaBrowsingStory):
   NAME = 'browse:media:facebook_photos'
   URL = (
       'https://www.facebook.com/rihanna/photos/a.207477806675.138795.10092511675/10153911739606676/?type=3&theater')
   ITEM_SELECTOR = '.snowliftPager.next'
   # Recording currently does not work. The page gets stuck in the
   # theater viewer.
   SUPPORTED_PLATFORMS = platforms.NO_PLATFORMS
   IS_SINGLE_PAGE_APP = True
+
+
+class TumblrDesktopStory(_MediaBrowsingStory):
+  NAME = 'browse:media:tumblr'
+  URL = 'https://tumblr.com/search/gifs'
+  ITEM_SELECTOR = '.photo'
+  IS_SINGLE_PAGE_APP = True
+  ITEMS_TO_VISIT = 2  # Increase when crbug.com/651909 is implemented.
+  ITEM_VIEW_TIME_IN_SECONDS = 5
+  INCREMENT_INDEX_AFTER_EACH_ITEM = True
+  SUPPORTED_PLATFORMS = platforms.NO_PLATFORMS  # crbug.com/651909.
+
+  def _ViewMediaItem(self, action_runner, index):
+    super(TumblrDesktopStory, self)._ViewMediaItem(action_runner, index)
+    action_runner.MouseClick(selector='#tumblr_lightbox_center_image')
+
+
+class PinterestDesktopStory(_MediaBrowsingStory):
+  NAME = 'browse:media:pinterest'
+  URL = 'https://pinterest.com'
+  ITEM_SELECTOR = '.pinImageDim'
+  IS_SINGLE_PAGE_APP = True
+  ITEMS_TO_VISIT = 5  # Increase when crbug.com/651909 is implemented.
+  ITEM_VIEW_TIME_IN_SECONDS = 5
+  INCREMENT_INDEX_AFTER_EACH_ITEM = True
+  SUPPORTED_PLATFORMS = platforms.DESKTOP_ONLY
+
+  def _Login(self, action_runner):
+    pinterest_login.LoginDesktopAccount(action_runner, 'googletest',
+                                        self.credentials_path)
+
+  def _ViewMediaItem(self, action_runner, index):
+    super(PinterestDesktopStory, self)._ViewMediaItem(action_runner, index)
+    # To imitate real user interaction, we do not want to pin every post.
+    # We will only pin every other post.
+    if index % 2 == 0:
+      # Pin the selection.
+      save_function = ('document.querySelector('
+                       '".Button.Module.ShowModalButton.btn.hasIcon.hasText.'
+                       'isBrioFlat.medium.primary.primaryOnHover.repin.'
+                       'pinActionBarButton.isBrioFlat.rounded")')
+      action_runner.ClickElement(element_function=save_function)
+      action_runner.Wait(1)  # Wait to make navigation realistic.
+      # Select which board to pin to.
+      inner_save_function = 'document.querySelector(".nameAndIcons")'
+      action_runner.WaitForElement(element_function=inner_save_function)
+      action_runner.ClickElement(element_function=inner_save_function)
+      action_runner.Wait(1)  # Wait to make navigation realistic.
+
+    # Close selection.
+    x_element_function = ('document.querySelector('
+                          '".Button.borderless.close.visible")')
+    action_runner.ClickElement(element_function=x_element_function)
+    action_runner.Wait(1)  # Wait to make navigation realistic.