Add support for python pageset

tools/telemetry/telemetry/page/page_set_unittest.py

 # Copyright (c) 2012 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
 import unittest
 
+from telemetry.core import util
 from telemetry.page import page_set
 
 
 simple_archive_info = """
 {
 "archives": {
   "data_01.wpr": ["http://www.foo.com/"],
   "data_02.wpr": ["http://www.bar.com/"]
 }
 }
 """
 
 
 simple_set = """
 {"description": "hello",
  "archive_data_file": "%s",
  "pages": [
    {"url": "http://www.foo.com/"},
    {"url": "http://www.bar.com/"}
  ]
 }
 """
 
 
 class TestPageSet(unittest.TestCase):
   def testSimpleSet(self):
     try:
-      with tempfile.NamedTemporaryFile(delete=False) as f:
+      with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as f:
         f.write(simple_archive_info)
 
-      with tempfile.NamedTemporaryFile(delete=False) as f2:
+      with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as f2:
         f2.write(simple_set % f.name.replace('\\', '\\\\'))
 
       ps = page_set.PageSet.FromFile(f2.name)
     finally:
       os.remove(f.name)
       os.remove(f2.name)
 
     self.assertEquals('hello', ps.description)
     self.assertEquals(f.name, ps.archive_data_file)
     self.assertEquals(2, len(ps.pages))
@@ skipping 29 matching lines @@
       'pages': [
         {'url': 'http://www.foo.com',
          'stress_memory': {'action': 'javasciprt'}
         },
         {'url': 'http://www.bar.com',
          'navigate_steps': {'action': 'navigate2'},
          'repaint' : {'action': 'scroll'}
         },
       ]}, 'file://foo.js')
 
-    self.assertEquals(ps.pages[0].RunNavigateSteps, {'action': 'navigate'})
+    self.assertTrue(hasattr(ps.pages[0], 'RunNavigateSteps'))
     self.assertEquals(ps.pages[0].RunSmoothness, {'action': 'scroll'})
     self.assertEquals(ps.pages[0].RunStressMemory, {'action': 'javasciprt'})
 
     self.assertEquals(ps.pages[1].RunSmoothness, {'action': 'scroll'})
     self.assertEquals(ps.pages[1].RunNavigateSteps, {'action': 'navigate2'})
     self.assertEquals(ps.pages[1].RunRepaint, {'action': 'scroll'})
+
+  def testRunNavigateStepsInheritance(self):
+    ps = page_set.PageSet.FromDict({
+      'serving_dirs': ['a/b'],
+      'navigate_steps' : { 'action' : 'navigate1' },
+      'pages': [
+        {'url': 'http://www.foo.com',
+        },
+        {'url': 'http://www.bar.com',
+         'navigate_steps': {'action': 'navigate2'},
+        },
+      ]}, 'file://foo.js')
+
+    self.assertEquals(ps.pages[0].RunNavigateSteps, {'action': 'navigate1'})
+    self.assertEquals(ps.pages[1].RunNavigateSteps, {'action': 'navigate2'})
+
+
+  def testSuccesfulPythonPageSetLoading(self):
+    test_pps_dir = os.path.join(util.GetUnittestDataDir(), 'test_page_set.py')
+    pps = page_set.PageSet.FromFile(test_pps_dir)
+    self.assertEqual('TestPageSet', pps.__class__.__name__)
+    self.assertEqual('A pageset for testing purpose', pps.description)
+    self.assertEqual('data/test.json', pps.archive_data_file)
+    self.assertEqual('data/credential', pps.credentials_path)
+    self.assertEqual('desktop', pps.user_agent_type)
+    self.assertEqual(test_pps_dir, pps.file_path)
+    self.assertEqual(1, len(pps.pages))
+    google_page = pps.pages[0]
+    self.assertEqual('https://www.google.com', google_page.url)
+    self.assertIs(pps, google_page.page_set)
+    self.assertTrue(hasattr(google_page, 'RunNavigateSteps'))
+    self.assertTrue(5, google_page.RunGetActionRunner(action_runner=5))
+
+  def testMultiplePythonPageSetsLoading(self):
+    test_pps_1_dir = os.path.join(util.GetUnittestDataDir(),
+                                'test_simple_one_page_set.py')
+    test_pps_2_dir = os.path.join(util.GetUnittestDataDir(),
+                                'test_simple_two_page_set.py')
+    pps1 = page_set.PageSet.FromFile(test_pps_1_dir)
+    pps2 = page_set.PageSet.FromFile(test_pps_2_dir)
+
+    self.assertEqual('TestSimpleOnePageSet', pps1.__class__.__name__)
+    self.assertEqual('TestSimpleTwoPageSet', pps2.__class__.__name__)