| Index: tools/telemetry/telemetry/core/user_agent_unittest.py
|
| diff --git a/tools/telemetry/telemetry/core/user_agent_unittest.py b/tools/telemetry/telemetry/core/user_agent_unittest.py
|
| index 1b50fb24a5eaa10e9e0ec460b6253080da34320f..242ef7657b5f230740a34402d42d2de94e2055a0 100644
|
| --- a/tools/telemetry/telemetry/core/user_agent_unittest.py
|
| +++ b/tools/telemetry/telemetry/core/user_agent_unittest.py
|
| @@ -6,7 +6,16 @@ from telemetry.core import user_agent
|
| from telemetry.unittest import tab_test_case
|
|
|
|
|
| -class UserAgentTest(tab_test_case.TabTestCase):
|
| +class MobileUserAgentTest(tab_test_case.TabTestCase):
|
| + @classmethod
|
| + def CustomizeBrowserOptions(cls, options):
|
| + options.browser_user_agent_type = 'mobile'
|
| +
|
| + def testUserAgent(self):
|
| + ua = self._tab.EvaluateJavaScript('window.navigator.userAgent')
|
| + self.assertEquals(ua, user_agent.UA_TYPE_MAPPING['mobile'])
|
| +
|
| +class TabletUserAgentTest(tab_test_case.TabTestCase):
|
| @classmethod
|
| def CustomizeBrowserOptions(cls, options):
|
| options.browser_user_agent_type = 'tablet'
|
| @@ -14,3 +23,12 @@ class UserAgentTest(tab_test_case.TabTestCase):
|
| def testUserAgent(self):
|
| ua = self._tab.EvaluateJavaScript('window.navigator.userAgent')
|
| self.assertEquals(ua, user_agent.UA_TYPE_MAPPING['tablet'])
|
| +
|
| +class DesktopUserAgentTest(tab_test_case.TabTestCase):
|
| + @classmethod
|
| + def CustomizeBrowserOptions(cls, options):
|
| + options.browser_user_agent_type = 'desktop'
|
| +
|
| + def testUserAgent(self):
|
| + ua = self._tab.EvaluateJavaScript('window.navigator.userAgent')
|
| + self.assertEquals(ua, user_agent.UA_TYPE_MAPPING['desktop'])
|
|
|