[system-health] Add support for recording stories directly through story set

tools/perf/page_sets/system_health/system_health_story.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 telemetry.page import page
 
 
 # Extra wait time after the page has loaded required by the loading metric. We
@@ skipping 29 matching lines @@
 
   def __init__(self, story_set, take_memory_measurement):
     case, group, _ = self.NAME.split(':')
     super(SystemHealthStory, self).__init__(
         page_set=story_set, name=self.NAME, url=self.URL,
         credentials_path='../data/credentials.json',
         grouping_keys={'case': case, 'group': group})
     self._take_memory_measurement = take_memory_measurement
 
   def _Measure(self, action_runner):
-    if self._ShouldMeasureMemory(action_runner):
+    if self._take_memory_measurement:
       action_runner.MeasureMemory(deterministic_mode=True)
     else:
       action_runner.Wait(_WAIT_TIME_AFTER_LOAD)
 
-  def _ShouldMeasureMemory(self, action_runner):
-    if not self._take_memory_measurement:
-      return False
-    # The check below is also performed in action_runner.MeasureMemory().
-    # However, we need to duplicate it here so that the story would wait for
-    # |_WAIT_TIME_AFTER_LOAD| seconds after load when recorded via the
-    # system_health.memory_* benchmarks.
-    # TODO(petrcermak): Make it possible (and mandatory) to record the story
-    # directly through the story set and remove this check.
-    tracing_controller = action_runner.tab.browser.platform.tracing_controller
-    if not tracing_controller.is_tracing_running:
-      return False  # Tracing is not running, e.g. when recording a WPR archive.
-    return True
-
   def _Login(self, action_runner):
     pass
 
   def _DidLoadDocument(self, action_runner):
     pass
 
   def RunNavigateSteps(self, action_runner):
     self._Login(action_runner)
     super(SystemHealthStory, self).RunNavigateSteps(action_runner)
 
   def RunPageInteractions(self, action_runner):
     action_runner.tab.WaitForDocumentReadyStateToBeComplete()
     self._DidLoadDocument(action_runner)
     self._Measure(action_runner)