Do not lose secondary gpus and make sure to have an active gpu on multiple gpu configurations

content/test/gpu/page_sets/gpu_process_tests.py

 # Copyright 2014 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 sys
 from telemetry.story import story_set as story_set_module
 from telemetry.page import page_test
 
 from gpu_tests import gpu_test_base
 
 class GpuProcessSharedPageState(gpu_test_base.GpuSharedPageState):
 
   gpu_switches = ['--gpu-no-complete-info-collection',
                   '--gpu-testing-os-version',
                   '--gpu-testing-vendor-id',
                   '--gpu-testing-device-id',
+                  '--gpu-testing-secondary-vendor-ids',
+                  '--gpu-testing-secondary-device-ids',
                   '--gpu-testing-gl-vendor',
                   '--gpu-testing-gl-renderer',
                   '--gpu-testing-gl-version']
 
   def __init__(self, test, finder_options, story_set):
     super(GpuProcessSharedPageState, self).__init__(
       test, finder_options, story_set)
     options = finder_options.browser_options.extra_browser_args
 
     # Clear all existing gpu testing switches.
     old_gpu_switches = []
     for opt in options:
       for gpu_switch in self.gpu_switches:
         if opt.startswith(gpu_switch):
           old_gpu_switches.append(opt)
     options.difference_update(old_gpu_switches)
 
+
+class IdentifyActiveGpuPageBase(gpu_test_base.PageBase):
+
+  def __init__(self, name=None, page_set=None, shared_page_state_class=None,
+               expectations=None, active_gpu=None, inactive_gpus=None):
+    super(IdentifyActiveGpuPageBase, self).__init__(
+      url='chrome:gpu',
+      name=name,
+      page_set=page_set,
+      shared_page_state_class=shared_page_state_class,
+      expectations=expectations)
+    self.active_gpu = active_gpu
+    self.inactive_gpus = inactive_gpus
+
+  def Validate(self, tab, results):
+    basic_infos = tab.EvaluateJavaScript('browserBridge.gpuInfo.basic_info')
+    active_gpu = []
+    inactive_gpus = []
+    index = 0
+    for info in basic_infos:
+      description = info['description']
+      value = info['value']
+      if description.startswith('GPU%d' % index) and value.startswith('VENDOR'):
+        if value.endswith('*ACTIVE*'):
+          active_gpu.append(value)
+        else:
+          inactive_gpus.append(value)
+        index += 1
+
+    if active_gpu != self.active_gpu:
+      raise page_test.Failure('Active GPU field is wrong %s' % active_gpu)
+
+    if inactive_gpus != self.inactive_gpus:
+      raise page_test.Failure('Inactive GPU field is wrong %s' % inactive_gpus)
+
+
 class GpuProcessTestsPage(gpu_test_base.PageBase):
   def __init__(self, url, name, story_set, expectations):
     super(GpuProcessTestsPage, self).__init__(url=url,
         shared_page_state_class=gpu_test_base.GpuSharedPageState,
         page_set=story_set,
         name=name,
         expectations=expectations)
 
 
 class FunctionalVideoPage(GpuProcessTestsPage):
@@ skipping 155 matching lines @@
         % (workaround, browser_list))
 
     gpu_list = tab.EvaluateJavaScript( \
       'chrome.gpuBenchmarking.getGpuDriverBugWorkarounds()')
     if not workaround in gpu_list:
       print 'Test failed. Printing page contents:'
       print tab.EvaluateJavaScript('document.body.innerHTML')
       raise page_test.Failure('%s missing in GPU process workarounds: %s' \
         % (workaround, gpu_list))
 
+
+class IdentifyActiveGpuSharedPageState1(GpuProcessSharedPageState):
+  def __init__(self, test, finder_options, story_set):
+    super(IdentifyActiveGpuSharedPageState1, self).__init__(
+      test, finder_options, story_set)
+    opts = finder_options.browser_options
+
+    if sys.platform in ('cygwin', 'win32') or sys.platform.startswith('linux'):
+      opts.AppendExtraBrowserArgs('--gpu-testing-vendor-id=0x8086')
+      opts.AppendExtraBrowserArgs('--gpu-testing-device-id=0x040a')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-vendor-ids=0x10de')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-device-ids=0x0de1')
+      opts.AppendExtraBrowserArgs('--gpu-testing-gl-vendor=nouveau')
+
+
+class IdentifyActiveGpuPage1(IdentifyActiveGpuPageBase):
+  def __init__(self, story_set, expectations):
+    active_gpu = ['VENDOR = 0x10de, DEVICE= 0x0de1 *ACTIVE*']
+    inactive_gpus = ['VENDOR = 0x8086, DEVICE= 0x040a']
+
+    super(IdentifyActiveGpuPage1, self).__init__(
+      name='GpuProcess.identify_active_gpu1',
+      page_set=story_set,
+      shared_page_state_class=IdentifyActiveGpuSharedPageState1,
+      expectations=expectations,
+      active_gpu=active_gpu,
+      inactive_gpus=inactive_gpus)
+
+  def Validate(self, tab, results):
+    super(IdentifyActiveGpuPage1, self).Validate(tab, results)
+
+
+class IdentifyActiveGpuSharedPageState2(GpuProcessSharedPageState):
+  def __init__(self, test, finder_options, story_set):
+    super(IdentifyActiveGpuSharedPageState2, self).__init__(
+      test, finder_options, story_set)
+    opts = finder_options.browser_options
+
+    if sys.platform in ('cygwin', 'win32') or sys.platform.startswith('linux'):
+      opts.AppendExtraBrowserArgs('--gpu-testing-vendor-id=0x8086')
+      opts.AppendExtraBrowserArgs('--gpu-testing-device-id=0x040a')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-vendor-ids=0x10de')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-device-ids=0x0de1')
+      opts.AppendExtraBrowserArgs('--gpu-testing-gl-vendor=Intel')
+
+
+class IdentifyActiveGpuPage2(IdentifyActiveGpuPageBase):
+  def __init__(self, story_set, expectations):
+    active_gpu = ['VENDOR = 0x8086, DEVICE= 0x040a *ACTIVE*']
+    inactive_gpus = ['VENDOR = 0x10de, DEVICE= 0x0de1']
+
+    super(IdentifyActiveGpuPage2, self).__init__(
+      name='GpuProcess.identify_active_gpu2',
+      page_set=story_set,
+      shared_page_state_class=IdentifyActiveGpuSharedPageState2,
+      expectations=expectations,
+      active_gpu=active_gpu,
+      inactive_gpus=inactive_gpus)
+
+  def Validate(self, tab, results):
+    super(IdentifyActiveGpuPage2, self).Validate(tab, results)
+
+
+class IdentifyActiveGpuSharedPageState3(GpuProcessSharedPageState):
+  def __init__(self, test, finder_options, story_set):
+    super(IdentifyActiveGpuSharedPageState3, self).__init__(
+      test, finder_options, story_set)
+    opts = finder_options.browser_options
+
+    if sys.platform in ('cygwin', 'win32') or sys.platform.startswith('linux'):
+      opts.AppendExtraBrowserArgs('--gpu-testing-vendor-id=0x8086')
+      opts.AppendExtraBrowserArgs('--gpu-testing-device-id=0x040a')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-vendor-ids= \
+                                  0x10de;0x1002')
+      opts.AppendExtraBrowserArgs('--gpu-testing-secondary-device-ids= \
+                                  0x0de1;0x6779')
+      opts.AppendExtraBrowserArgs('--gpu-testing-gl-vendor=radeon')
+
+
+class IdentifyActiveGpuPage3(IdentifyActiveGpuPageBase):
+  def __init__(self, story_set, expectations):
+    active_gpu = ['VENDOR = 0x1002, DEVICE= 0x6779 *ACTIVE*']
+    inactive_gpus = ['VENDOR = 0x8086, DEVICE= 0x040a', \
+                     'VENDOR = 0x10de, DEVICE= 0x0de1']
+
+    super(IdentifyActiveGpuPage3, self).__init__(
+      name='GpuProcess.identify_active_gpu3',
+      page_set=story_set,
+      shared_page_state_class=IdentifyActiveGpuSharedPageState3,
+      expectations=expectations,
+      active_gpu=active_gpu,
+      inactive_gpus=inactive_gpus)
+
+  def Validate(self, tab, results):
+    super(IdentifyActiveGpuPage3, self).Validate(tab, results)
+
 class GpuProcessTestsStorySet(story_set_module.StorySet):
 
   """ Tests that accelerated content triggers the creation of a GPU process """
 
   def __init__(self, expectations):
     super(GpuProcessTestsStorySet, self).__init__(
       serving_dirs=set(['../../../../content/test/data']))
 
     urls_and_names_list = [
       ('file://../../data/gpu/functional_canvas_demo.html',
        'GpuProcess.canvas2d'),
       ('file://../../data/gpu/functional_3d_css.html',
        'GpuProcess.css3d'),
       ('file://../../data/gpu/functional_webgl.html',
        'GpuProcess.webgl')
     ]
 
     for url, name in urls_and_names_list:
       self.AddStory(GpuProcessTestsPage(url, name, self, expectations))
 
     self.AddStory(FunctionalVideoPage(self, expectations))
     self.AddStory(GpuInfoCompletePage(self, expectations))
     self.AddStory(NoGpuProcessPage(self, expectations))
     self.AddStory(SoftwareGpuProcessPage(self, expectations))
     self.AddStory(SkipGpuProcessPage(self, expectations))
     self.AddStory(DriverBugWorkaroundsInGpuProcessPage(self, expectations))
+    self.AddStory(IdentifyActiveGpuPage1(self, expectations))
+    self.AddStory(IdentifyActiveGpuPage2(self, expectations))
+    self.AddStory(IdentifyActiveGpuPage3(self, expectations))
 
   @property
   def allow_mixed_story_states(self):
     # Return True here in order to be able to run pages with different browser
     # command line arguments.
     return True