Expose GPU information to Telemetry.

tools/telemetry/telemetry/core/gpu_device_unittest.py

Index: tools/telemetry/telemetry/core/gpu_device_unittest.py
diff --git a/tools/telemetry/telemetry/core/gpu_device_unittest.py b/tools/telemetry/telemetry/core/gpu_device_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..08b02633e3cc4e8bb7ec4c9809fe1320647de992
--- /dev/null
+++ b/tools/telemetry/telemetry/core/gpu_device_unittest.py
@@ -0,0 +1,44 @@
+# 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_device
+
+class TestGPUDevice(unittest.TestCase):
+  def testConstruction(self):
+    device = gpu_device.GPUDevice(1000, 2000, 'test_vendor', 'test_device')
+    self.assertEquals(device.vendor_id, 1000)
+    self.assertEquals(device.device_id, 2000)
+    self.assertEquals(device.vendor_string, 'test_vendor')
+    self.assertEquals(device.device_string, 'test_device')
+
+  def testFromDict(self):
+    dictionary = { 'vendor_id': 3000,
+                   'device_id': 4000,
+                   'vendor_string': 'test_vendor_2',
+                   'device_string': 'test_device_2' }
+    device = gpu_device.GPUDevice.FromDict(dictionary)
+    self.assertEquals(device.vendor_id, 3000)
+    self.assertEquals(device.device_id, 4000)
+    self.assertEquals(device.vendor_string, 'test_vendor_2')
+    self.assertEquals(device.device_string, 'test_device_2')
+
+  def testMissingAttrsFromDict(self):
+    data = {
+      'vendor_id': 1,
+      'device_id': 2,
+      'vendor_string': 'a',
+      'device_string': 'b'
+    }
+
+    for k in data:
+      data_copy = data.copy()
+      del data_copy[k]
+      try:
+        gpu_device.GPUDevice.FromDict(data_copy)
+        self.fail('Should raise exception if attribute "%s" is missing' % k)
+      except AssertionError:
+        raise
+      except:
+        pass