[telemetry] Prototype for specifying page sets using Python imports and Telemetry discovery.

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
 
+from unittest_data import test_page_set
+
 
 class TestPageSet(unittest.TestCase):
 
   def testServingDirs(self):
     directory_path = tempfile.mkdtemp()
     try:
       ps = page_set.PageSet(serving_dirs=['a/b'], file_path=directory_path)
       ps.AddPageWithDefaultRunNavigate('file://c/test.html')
       ps.AddPageWithDefaultRunNavigate('file://c/test.js')
       ps.AddPageWithDefaultRunNavigate('file://d/e/../test.html')
@@ skipping 13 matching lines @@
       ps = page_set.PageSet(file_path=directory_path,
                             serving_dirs=['', directory_path, absolute_dir])
       real_directory_path = os.path.realpath(directory_path)
       real_absolute_dir = os.path.realpath(absolute_dir)
       self.assertEquals(ps.serving_dirs, set([real_directory_path,
                                               real_absolute_dir]))
     finally:
       os.rmdir(directory_path)
 
   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__)
+    pps = test_page_set.TestPageSet()
     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(test_page_set.__file__, pps.file_path)
     self.assertEqual(3, 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(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__)
-
   def testPageFilePath(self):
-    test_pps_dir = os.path.join(util.GetUnittestDataDir(), 'test_page_set.py')
-    pps = page_set.PageSet.FromFile(test_pps_dir)
+    pps = test_page_set.TestPageSet()
     internal_page = pps.pages[1]
     external_page = pps.pages[2]
     self.assertEqual(
       os.path.normpath(os.path.join(
         util.GetUnittestDataDir(), 'bar.html')), internal_page.file_path)
     self.assertEqual(
       os.path.normpath(os.path.join(
         util.GetUnittestDataDir(), 'pages/foo.html')), external_page.file_path)