OLD | NEW |
1 # Copyright 2012 The Chromium Authors. All rights reserved. | 1 # Copyright 2012 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import os | 5 import os |
6 import tempfile | 6 import tempfile |
7 import unittest | 7 import unittest |
8 | 8 |
9 from telemetry.page import page | 9 from telemetry.page import page |
10 from telemetry.page import page_set | 10 from telemetry.page import page_set |
11 from telemetry.util import cloud_storage | 11 from telemetry.util import cloud_storage |
12 | 12 |
13 | 13 |
14 class TestPageSet(unittest.TestCase): | 14 class TestPageSet(unittest.TestCase): |
15 | 15 |
16 def testServingDirs(self): | 16 def testServingDirs(self): |
17 directory_path = tempfile.mkdtemp() | 17 directory_path = tempfile.mkdtemp() |
18 try: | 18 try: |
19 ps = page_set.PageSet(serving_dirs=['a/b'], file_path=directory_path) | 19 ps = page_set.PageSet(serving_dirs=['a/b'], file_path=directory_path) |
20 ps.AddUserStory(page.Page('file://c/test.html', ps, ps.base_dir)) | 20 ps.AddUserStory(page.Page('file://c/test.html', ps, ps.base_dir)) |
21 ps.AddUserStory(page.Page('file://c/test.js', ps, ps.base_dir)) | 21 ps.AddUserStory(page.Page('file://c/test.js', ps, ps.base_dir)) |
22 ps.AddUserStory(page.Page('file://d/e/../test.html', ps, ps.base_dir)) | 22 ps.AddUserStory(page.Page('file://d/e/../test.html', ps, ps.base_dir)) |
23 finally: | 23 finally: |
24 os.rmdir(directory_path) | 24 os.rmdir(directory_path) |
25 | 25 |
26 real_directory_path = os.path.realpath(directory_path) | 26 real_directory_path = os.path.realpath(directory_path) |
27 expected_serving_dirs = set([os.path.join(real_directory_path, 'a', 'b'), | 27 expected_serving_dirs = set([os.path.join(real_directory_path, 'a', 'b')]) |
28 os.path.join(real_directory_path, 'c'), | |
29 os.path.join(real_directory_path, 'd')]) | |
30 self.assertEquals(ps.serving_dirs, expected_serving_dirs) | 28 self.assertEquals(ps.serving_dirs, expected_serving_dirs) |
31 self.assertEquals(ps[0].serving_dir, os.path.join(real_directory_path, 'c')) | 29 self.assertEquals(ps[0].serving_dir, os.path.join(real_directory_path, 'c')) |
32 self.assertEquals(ps[2].serving_dir, os.path.join(real_directory_path, 'd')) | 30 self.assertEquals(ps[2].serving_dir, os.path.join(real_directory_path, 'd')) |
33 | 31 |
34 def testAbsoluteServingDir(self): | 32 def testAbsoluteServingDir(self): |
35 directory_path = tempfile.mkdtemp() | 33 directory_path = tempfile.mkdtemp() |
36 try: | 34 try: |
37 absolute_dir = os.path.join(directory_path, 'a', 'b') | 35 absolute_dir = os.path.join(directory_path, 'a', 'b') |
38 ps = page_set.PageSet(file_path=directory_path, | 36 ps = page_set.PageSet(file_path=directory_path, |
39 serving_dirs=['', directory_path, absolute_dir]) | 37 serving_dirs=['', directory_path, absolute_dir]) |
(...skipping 17 matching lines...) Expand all Loading... |
57 page_set_b = page_set.PageSet() | 55 page_set_b = page_set.PageSet() |
58 for p in pages: | 56 for p in pages: |
59 p.TransferToPageSet(page_set_b) | 57 p.TransferToPageSet(page_set_b) |
60 page_set_b.AddUserStory(page.Page('http://baz.com', page_set_b)) | 58 page_set_b.AddUserStory(page.Page('http://baz.com', page_set_b)) |
61 self.assertEqual(0, len(page_set_a.pages)) | 59 self.assertEqual(0, len(page_set_a.pages)) |
62 self.assertEqual( | 60 self.assertEqual( |
63 set(['http://foo.com', 'http://bar.com', 'http://baz.com']), | 61 set(['http://foo.com', 'http://bar.com', 'http://baz.com']), |
64 set(p.url for p in page_set_b.pages)) | 62 set(p.url for p in page_set_b.pages)) |
65 for p in page_set_b.pages: | 63 for p in page_set_b.pages: |
66 self.assertIs(page_set_b, p.page_set) | 64 self.assertIs(page_set_b, p.page_set) |
OLD | NEW |