ui/gl/glx_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_glx_api_implementation.h"
   9 #include "ui/gl/gl_implementation.h"
  10
  11 namespace gfx {
  12
  13 class GLXApiTest : public testing::Test {
  14  public:
  15   void SetUp() override {
  16     fake_extension_string_ = "";
  17
  18     g_driver_glx.ClearBindings();
  19     g_driver_glx.fn.glXQueryExtensionsStringFn = &FakeQueryExtensionsString;
  20     SetGLImplementation(kGLImplementationMockGL);
  21     SetGLGetProcAddressProc(
  22         static_cast<GLGetProcAddressProc>(&FakeGLGetProcAddress));
  23   }
  24
  25   void TearDown() override {
  26     g_current_glx_context = nullptr;
  27     api_.reset(nullptr);
  28     g_driver_glx.ClearBindings();
  29
  30     fake_extension_string_ = "";
  31   }
  32
  33   void InitializeAPI(base::CommandLine* command_line) {
  34     api_.reset(new RealGLXApi());
  35     g_current_glx_context = api_.get();
  36     if (command_line)
  37       api_->InitializeWithCommandLine(&g_driver_glx, command_line);
  38     else
  39       api_->Initialize(&g_driver_glx);
  40     g_driver_glx.InitializeExtensionBindings();
  41   }
  42
  43   void SetFakeExtensionString(const char* fake_string) {
  44     fake_extension_string_ = fake_string;
  45   }
  46
  47   const char* GetExtensions() {
  48     return api_->glXQueryExtensionsStringFn(reinterpret_cast<Display*>(0x1), 0);
  49   }
  50
  51   static GLXContext FakeCreateContextAttribsARB(Display* dpy,
  52                                                 GLXFBConfig config,
  53                                                 GLXContext share_context,
  54                                                 int direct,
  55                                                 const int* attrib_list) {
  56     return static_cast<GLXContext>(nullptr);
  57   }
  58
  59   static void* GL_BINDING_CALL FakeGLGetProcAddress(const char *proc) {
  60     if (!strcmp("glXCreateContextAttribsARB", proc)) {
  61       return reinterpret_cast<void *>(&FakeCreateContextAttribsARB);
  62     }
  63     return NULL;
  64   }
  65
  66   static const char* GL_BINDING_CALL FakeQueryExtensionsString(Display* dpy,
  67                                                                int screen) {
  68     return fake_extension_string_;
  69   }
  70
  71  protected:
  72   static const char* fake_extension_string_;
  73
  74   scoped_ptr<RealGLXApi> api_;
  75 };
  76
  77 const char* GLXApiTest::fake_extension_string_ = "";
  78
  79 TEST_F(GLXApiTest, DisabledExtensionBitTest) {
  80   static const char* kFakeExtensions = "GLX_ARB_create_context";
  81   static const char* kFakeDisabledExtensions = "GLX_ARB_create_context";
  82
  83   SetFakeExtensionString(kFakeExtensions);
  84   InitializeAPI(nullptr);
  85
  86   EXPECT_TRUE(g_driver_glx.ext.b_GLX_ARB_create_context);
  87
  88   base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
  89   command_line.AppendSwitchASCII(switches::kDisableGLExtensions,
  90                                  kFakeDisabledExtensions);
  91   InitializeAPI(&command_line);
  92
  93   EXPECT_FALSE(g_driver_glx.ext.b_GLX_ARB_create_context);
  94 }
  95
  96 TEST_F(GLXApiTest, DisabledExtensionStringTest) {
  97   static const char* kFakeExtensions = "EGL_EXT_1 EGL_EXT_2"
  98                                        " EGL_EXT_3 EGL_EXT_4";
  99   static const char* kFakeDisabledExtensions =
 100       "EGL_EXT_1,EGL_EXT_2,EGL_FAKE";
 101   static const char* kFilteredExtensions = "EGL_EXT_3 EGL_EXT_4";
 102
 103   SetFakeExtensionString(kFakeExtensions);
 104   InitializeAPI(nullptr);
 105
 106   EXPECT_STREQ(kFakeExtensions, GetExtensions());
 107
 108   base::CommandLine command_line(base::CommandLine::NO_PROGRAM);
 109   command_line.AppendSwitchASCII(switches::kDisableGLExtensions,
 110                                  kFakeDisabledExtensions);
 111   InitializeAPI(&command_line);
 112
 113   EXPECT_STREQ(kFilteredExtensions, GetExtensions());
 114 }
 115
 116 }  // namespace gfx