ui/gl/wgl_api_unittest.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/command_line.h"
   6 #include "base/memory/scoped_ptr.h"
   7 #include "testing/gtest/include/gtest/gtest.h"
   8 #include "ui/gl/gl_implementation.h"
   9 #include "ui/gl/gl_surface_wgl.h"
  10 #include "ui/gl/gl_wgl_api_implementation.h"
  11
  12 typedef std::pair<std::string, std::string> stringpair;
  13
  14 namespace gfx {
  15
  16 class WGLApiTest : public testing::Test {
  17  public:
  18   void SetUp() override {
  19     GLSurfaceWGL::InitializeOneOffForTesting();
  20     fake_arb_extension_string_ = "";
  21     fake_ext_extension_string_ = "";
  22
  23     g_driver_wgl.ClearBindings();
  24     g_driver_wgl.fn.wglGetExtensionsStringARBFn = &FakeGetExtensionsStringARB;
  25     g_driver_wgl.fn.wglGetExtensionsStringEXTFn = &FakeGetExtensionsStringEXT;
  26     SetGLImplementation(kGLImplementationDesktopGL);
  27     SetGLGetProcAddressProc(
  28         static_cast<GLGetProcAddressProc>(&FakeGLGetProcAddress));
  29   }
  30
  31   void TearDown() override {
  32     g_current_wgl_context = nullptr;
  33     api_.reset(nullptr);
  34     g_driver_wgl.ClearBindings();
  35
  36     fake_ext_extension_string_ = "";
  37     fake_arb_extension_string_ = "";
  38   }
  39
  40   void InitializeAPI(base::CommandLine* command_line) {
  41     api_.reset(new RealWGLApi());
  42     g_current_wgl_context = api_.get();
  43     if (command_line)
  44       api_->InitializeWithCommandLine(&g_driver_wgl, command_line);
  45     else
  46       api_->Initialize(&g_driver_wgl);
  47     g_driver_wgl.InitializeExtensionBindings();
  48   }
  49
  50   void SetFakeEXTExtensionString(const char* fake_string) {
  51     fake_ext_extension_string_ = fake_string;
  52   }
  53
  54   void SetFakeARBExtensionString(const char* fake_string) {
  55     fake_arb_extension_string_ = fake_string;
  56   }
  57
  58   stringpair GetExtensions() {
  59     auto stringify = [](const char* str) -> std::string {
  60       return str ? str : "";
  61     };
  62     return stringpair(
  63         stringify(wglGetExtensionsStringARB(NULL)),
  64         stringify(wglGetExtensionsStringEXT()));
  65   }
  66
  67   static void* GL_BINDING_CALL FakeGLGetProcAddress(const char *proc) {
  68     return NULL;
  69   }
  70
  71   static const char* GL_BINDING_CALL FakeGetExtensionsStringARB(HDC dc) {
  72     return fake_arb_extension_string_;
  73   }
  74
  75   static const char* GL_BINDING_CALL FakeGetExtensionsStringEXT() {
  76     return fake_ext_extension_string_;
  77   }
  78
  79  protected:
  80   static const char* fake_ext_extension_string_;
  81   static const char* fake_arb_extension_string_;
  82
  83   scoped_ptr<RealWGLApi> api_;
  84 };
  85
  86 const char* WGLApiTest::fake_ext_extension_string_ = "";
  87 const char* WGLApiTest::fake_arb_extension_string_ = "";
  88
  89 TEST_F(WGLApiTest, DisabledExtensionBitTest) {
  90   static const char* kFakeExtensions = "WGL_ARB_extensions_string";
  91   static const char* kFakeDisabledExtensions = "WGL_ARB_extensions_string";
  92
  93   InitializeAPI(nullptr);
  94
  95   EXPECT_FALSE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
  96
  97   base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  98   command_line.AppendSwitchASCII(switches::kDisableGLExtensions,
  99                                  kFakeDisabledExtensions);
 100
 101   // NULL simulates not being able to resolve wglGetExtensionsStringARB
 102   SetFakeARBExtensionString(NULL);
 103   SetFakeEXTExtensionString(kFakeExtensions);
 104
 105   InitializeAPI(nullptr);
 106   EXPECT_TRUE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
 107
 108   InitializeAPI(&command_line);
 109   EXPECT_FALSE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
 110
 111   SetFakeARBExtensionString("");
 112   SetFakeEXTExtensionString(kFakeExtensions);
 113
 114   InitializeAPI(nullptr);
 115   // We expect false here, because wglGetExtensionsStringARB
 116   // always takes precedence over wglGetExtensionsStringEXT
 117   // if it is available.
 118   EXPECT_FALSE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
 119
 120   SetFakeARBExtensionString(kFakeExtensions);
 121   SetFakeEXTExtensionString("");
 122
 123   InitializeAPI(nullptr);
 124   EXPECT_TRUE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
 125
 126   InitializeAPI(&command_line);
 127   EXPECT_FALSE(g_driver_wgl.ext.b_WGL_ARB_extensions_string);
 128 }
 129
 130 TEST_F(WGLApiTest, DisabledExtensionStringTest) {
 131   static const char* kFakeExtensions = "EGL_EXT_1 EGL_EXT_2"
 132                                        " EGL_EXT_3 EGL_EXT_4";
 133   static const char* kFakeDisabledExtensions =
 134       "EGL_EXT_1,EGL_EXT_2,EGL_FAKE";
 135   static const char* kFilteredExtensions = "EGL_EXT_3 EGL_EXT_4";
 136
 137   base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
 138   command_line.AppendSwitchASCII(switches::kDisableGLExtensions,
 139                                  kFakeDisabledExtensions);
 140
 141   SetFakeARBExtensionString(kFakeExtensions);
 142   SetFakeEXTExtensionString(kFakeExtensions);
 143
 144   InitializeAPI(nullptr);
 145   EXPECT_EQ(stringpair(kFakeExtensions, kFakeExtensions),
 146             GetExtensions());
 147
 148   InitializeAPI(&command_line);
 149   EXPECT_EQ(stringpair(kFilteredExtensions, kFilteredExtensions),
 150             GetExtensions());
 151 }
 152
 153 }  // namespace gfx