| Index: tools/telemetry/telemetry/core/browser_unittest.py
|
| diff --git a/tools/telemetry/telemetry/core/browser_unittest.py b/tools/telemetry/telemetry/core/browser_unittest.py
|
| index 81a80f71b70192df343d6aaa23797868bcd1fb44..8fdb8383daf2617582e8d76dca37470d95a5ee18 100644
|
| --- a/tools/telemetry/telemetry/core/browser_unittest.py
|
| +++ b/tools/telemetry/telemetry/core/browser_unittest.py
|
| @@ -6,6 +6,8 @@ import os
|
| import unittest
|
|
|
| from telemetry.core import browser_finder
|
| +from telemetry.core import gpu_device
|
| +from telemetry.core import gpu_info
|
| from telemetry.unittest import options_for_unittests
|
|
|
| class BrowserTest(unittest.TestCase):
|
| @@ -123,3 +125,18 @@ class BrowserTest(unittest.TestCase):
|
| return
|
|
|
| self.assertEquals(1, len(b.tabs))
|
| +
|
| + def testGetGPUInfo(self):
|
| + b = self.CreateBrowser()
|
| + if not b.supports_gpu_info:
|
| + logging.warning(
|
| + 'Browser does not support getting GPU info, skipping test.')
|
| + return
|
| +
|
| + info = b.GetGPUInfo()
|
| +
|
| + self.assertTrue(isinstance(info, gpu_info.GPUInfo))
|
| + self.assertTrue(hasattr(info, 'gpu'))
|
| + self.assertIsInstance(info.gpu, gpu_device.GPUDevice)
|
| + self.assertTrue(hasattr(info, 'machine_model'))
|
| + self.assertTrue(hasattr(info, 'secondary_gpus'))
|
|
|