[shared_page_state] Fix unit test

telemetry/telemetry/page/shared_page_state_unittest.py

Index: telemetry/telemetry/page/shared_page_state_unittest.py
diff --git a/telemetry/telemetry/page/shared_page_state_unittest.py b/telemetry/telemetry/page/shared_page_state_unittest.py
index 7909f8f0919a2f941f14c7d197539ae57e894d4e..f82de45876bfb3db984f898d985907a6dad91027 100644
--- a/telemetry/telemetry/page/shared_page_state_unittest.py
+++ b/telemetry/telemetry/page/shared_page_state_unittest.py
@@ -4,6 +4,7 @@
 
 import unittest
 
+from telemetry.core import platform as platform_module
 from telemetry.internal import story_runner
 from telemetry.page import page
 from telemetry.page import legacy_page_test
@@ -82,8 +83,12 @@ class SharedPageStateTests(unittest.TestCase):
   def testPageStatesUserAgentType(self):
     self.assertUserAgentSetCorrectly(
         shared_page_state.SharedMobilePageState, 'mobile')
+    if platform_module.GetHostPlatform().GetOSName() == 'chromeos':

[nednguyen, 2017/04/25 12:54:44]

can you just change this flow to:
if platform_module.GetHostPlatform().GetOSName() == 'chromeos':
  
 self.assertUserAgentSetCorrectly(shared_page_state.SharedDesktopPageState,
'chromeos')
else:
 self.assert...('desktop')

[cywang, 2017/04/25 13:14:18]

Done.

+      device_type = 'chromeos'
+    else:
+      device_type = 'desktop'
     self.assertUserAgentSetCorrectly(
-        shared_page_state.SharedDesktopPageState, 'desktop')
+        shared_page_state.SharedDesktopPageState, device_type)
     self.assertUserAgentSetCorrectly(
         shared_page_state.SharedTabletPageState, 'tablet')
     self.assertUserAgentSetCorrectly(