OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 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 tempfile | 5 import tempfile |
6 import unittest | 6 import unittest |
7 | 7 |
8 from telemetry.core import wpr_modes | 8 from telemetry.core import wpr_modes |
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 |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
69 self.assertEqual(test.options, self.options) | 69 self.assertEqual(test.options, self.options) |
70 | 70 |
71 def assertUserAgentSetCorrectly( | 71 def assertUserAgentSetCorrectly( |
72 self, shared_page_state_class, expected_user_agent): | 72 self, shared_page_state_class, expected_user_agent): |
73 us = page.Page( | 73 us = page.Page( |
74 'http://www.google.com', | 74 'http://www.google.com', |
75 shared_page_state_class=shared_page_state_class) | 75 shared_page_state_class=shared_page_state_class) |
76 test = DummyTest() | 76 test = DummyTest() |
77 uss = user_story_set.UserStorySet() | 77 uss = user_story_set.UserStorySet() |
78 uss.AddUserStory(us) | 78 uss.AddUserStory(us) |
79 us.shared_user_story_state_class(test, self.options, uss) | 79 us.shared_state_class(test, self.options, uss) |
80 browser_options = self.options.browser_options | 80 browser_options = self.options.browser_options |
81 actual_user_agent = browser_options.browser_user_agent_type | 81 actual_user_agent = browser_options.browser_user_agent_type |
82 self.assertEqual(expected_user_agent, actual_user_agent) | 82 self.assertEqual(expected_user_agent, actual_user_agent) |
83 | 83 |
84 def testPageStatesUserAgentType(self): | 84 def testPageStatesUserAgentType(self): |
85 self.assertUserAgentSetCorrectly( | 85 self.assertUserAgentSetCorrectly( |
86 shared_page_state.SharedMobilePageState, 'mobile') | 86 shared_page_state.SharedMobilePageState, 'mobile') |
87 self.assertUserAgentSetCorrectly( | 87 self.assertUserAgentSetCorrectly( |
88 shared_page_state.SharedDesktopPageState, 'desktop') | 88 shared_page_state.SharedDesktopPageState, 'desktop') |
89 self.assertUserAgentSetCorrectly( | 89 self.assertUserAgentSetCorrectly( |
90 shared_page_state.SharedTabletPageState, 'tablet') | 90 shared_page_state.SharedTabletPageState, 'tablet') |
91 self.assertUserAgentSetCorrectly( | 91 self.assertUserAgentSetCorrectly( |
92 shared_page_state.Shared10InchTabletPageState, 'tablet_10_inch') | 92 shared_page_state.Shared10InchTabletPageState, 'tablet_10_inch') |
OLD | NEW |