Expose GPU information to Telemetry.

tools/telemetry/telemetry/core/gpu_info_unittest.py

Index: tools/telemetry/telemetry/core/gpu_info_unittest.py
diff --git a/tools/telemetry/telemetry/core/gpu_info_unittest.py b/tools/telemetry/telemetry/core/gpu_info_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..29c69b75ad2a5dd634a07c04a8b91f792180dc65
--- /dev/null
+++ b/tools/telemetry/telemetry/core/gpu_info_unittest.py
@@ -0,0 +1,82 @@
+# Copyright (c) 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+import unittest
+
+from telemetry.core import gpu_info
+from telemetry.core import gpu_device
+
+class TestGPUInfo(unittest.TestCase):
+  def testConstruction(self):
+    data = {
+      'devices': [
+        { 'vendor_id': 1000, 'device_id': 2000,
+          'vendor_string': 'a', 'device_string': 'b' },
+        { 'vendor_id': 3000, 'device_id': 4000,
+          'vendor_string': 'k', 'device_string': 'l' }
+      ],
+      'aux_attributes': {
+        'optimus': False,
+        'amd_switchable': False,
+        'lenovo_dcute': False,
+        'driver_vendor': 'c',
+        'driver_version': 'd',
+        'driver_date': 'e',
+        'gl_version_string': 'g',
+        'gl_vendor': 'h',
+        'gl_renderer': 'i',
+        'gl_extensions': 'j',
+      }
+    }
+    info = gpu_info.GPUInfo.FromDict(data)
+    self.assertTrue(len(info.devices) == 2)
+    self.assertTrue(isinstance(info.devices[0], gpu_device.GPUDevice))
+    self.assertEquals(info.devices[0].vendor_id, 1000)
+    self.assertEquals(info.devices[0].device_id, 2000)
+    self.assertEquals(info.devices[0].vendor_string, 'a')
+    self.assertEquals(info.devices[0].device_string, 'b')
+    self.assertTrue(isinstance(info.devices[1], gpu_device.GPUDevice))
+    self.assertEquals(info.devices[1].vendor_id, 3000)
+    self.assertEquals(info.devices[1].device_id, 4000)
+    self.assertEquals(info.devices[1].vendor_string, 'k')
+    self.assertEquals(info.devices[1].device_string, 'l')
+    self.assertEquals(info.aux_attributes['optimus'], False)
+    self.assertEquals(info.aux_attributes['amd_switchable'], False)
+    self.assertEquals(info.aux_attributes['lenovo_dcute'], False)
+    self.assertEquals(info.aux_attributes['driver_vendor'], 'c')
+    self.assertEquals(info.aux_attributes['driver_version'], 'd')
+    self.assertEquals(info.aux_attributes['driver_date'], 'e')
+    self.assertEquals(info.aux_attributes['gl_version_string'], 'g')
+    self.assertEquals(info.aux_attributes['gl_vendor'], 'h')
+    self.assertEquals(info.aux_attributes['gl_renderer'], 'i')
+    self.assertEquals(info.aux_attributes['gl_extensions'], 'j')
+
+  def testMissingAttrsFromDict(self):
+    data = {
+      'devices': [{ 'vendor_id': 1000, 'device_id': 2000,
+                    'vendor_string': 'a', 'device_string': 'b' }]
+    }
+
+    for k in data:
+      data_copy = data.copy()
+      del data_copy[k]
+      try:
+        gpu_info.GPUInfo.FromDict(data_copy)
+        self.fail('Should raise exception if attribute "%s" is missing' % k)
+      except AssertionError:
+        raise
+      except:
+        pass
+
+  def testMissingDevices(self):
+    data = {
+      'devices': []
+    }
+
+    try:
+      gpu_info.GPUInfo.FromDict(data)
+      self.fail('Should raise exception if devices array is empty')
+    except AssertionError:
+      raise
+    except:
+      pass