| 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 |