[Telemetry] Make session restore benchmark not require arguments.

tools/perf/benchmarks/session_restore.py

 # Copyright 2013 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.
 
+import os
+import tempfile
+
 from measurements import session_restore
+from profile_creators import small_profile_creator
 from telemetry import test
+from telemetry.page import profile_generator
+
+
+class _SessionRestoreTest(test.Test):
+
+  @classmethod
+  def ProcessCommandLineArgs(cls, parser, args):
+    super(_SessionRestoreTest, cls).ProcessCommandLineArgs(parser, args)
+
+    profile_type = 'small_profile'
+
+    if not args.browser_options.profile_dir:
+      profile_dir = os.path.join(tempfile.gettempdir(), profile_type)
+      args.browser_options.profile_dir = profile_dir
+
+      if not os.path.exists(profile_dir):
+        os.makedirs(profile_dir)
+        args = args.Copy()
+        args.pageset_repeat = 1
+        args.output_dir = tempfile.gettempdir()
+        profile_generator.GenerateProfiles(
+            small_profile_creator.SmallProfileCreator, profile_type, args)
 
 
 # crbug.com/325479: Disabling this test for now since it never ran before.
 @test.Disabled('android', 'linux')
-class SessionRestoreColdTypical25(test.Test):
+class SessionRestoreColdTypical25(_SessionRestoreTest):
   tag = 'cold'
   test = session_restore.SessionRestore
   page_set = 'page_sets/typical_25.py'
   options = {'cold': True,
              'pageset_repeat': 5}
 
 
-class SessionRestoreWarmTypical25(test.Test):
+class SessionRestoreWarmTypical25(_SessionRestoreTest):
   tag = 'warm'
   test = session_restore.SessionRestore
   page_set = 'page_sets/typical_25.py'
   options = {'warm': True,
              'pageset_repeat': 20}