// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "gpu/command_buffer/service/shared_context_state.h"

#include <limits>
#include <memory>
#include <string>
#include <utility>

#include "base/command_line.h"
#include "base/functional/callback_helpers.h"
#include "base/memory/ptr_util.h"
#include "gpu/command_buffer/common/gles2_cmd_utils.h"
#include "gpu/command_buffer/service/context_state_test_helpers.h"
#include "gpu/command_buffer/service/feature_info.h"
#include "gpu/command_buffer/service/test_helper.h"
#include "gpu/config/gpu_driver_bug_workarounds.h"
#include "gpu/config/gpu_feature_info.h"
#include "gpu/config/gpu_preferences.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/gl/gl_context_stub.h"
#include "ui/gl/gl_mock.h"
#include "ui/gl/gl_share_group.h"
#include "ui/gl/gl_surface_stub.h"
#include "ui/gl/init/gl_factory.h"
#include "ui/gl/test/gl_surface_test_support.h"

using ::testing::_;
using ::testing::InSequence;
using ::testing::NiceMock;
using ::testing::Return;
using ::testing::SetArgPointee;
using ::testing::StrictMock;

namespace gpu {

class SharedContextStateTest : public ::testing::Test {
 public:
  SharedContextStateTest() = default;
};

class MockGrContextOptionsProvider
    : public SharedContextState::GrContextOptionsProvider {
 public:
  MOCK_METHOD(void,
              SetCustomGrContextOptions,
              (GrContextOptions & options),
              (const, override));
};

TEST_F(SharedContextStateTest, InitFailsIfLostContext) {
  const ContextType context_type = CONTEXT_TYPE_OPENGLES2;

  // For easier substring/extension matching
  gl::SetGLGetProcAddressProc(gl::MockGLInterface::GetGLProcAddress);
  auto* display = gl::GLSurfaceTestSupport::InitializeOneOffWithMockBindings();
  ASSERT_TRUE(display);
  {
    StrictMock<gl::MockGLInterface> gl_interface;
    gl::MockGLInterface::SetGLInterface(&gl_interface);

    InSequence sequence;

    auto surface = base::MakeRefCounted<gl::GLSurfaceStub>();
    auto context = base::MakeRefCounted<gl::GLContextStub>();
    const char gl_version[] = "OpenGL ES 2.0";
    context->SetGLVersionString(gl_version);
    const char gl_extensions[] = "GL_KHR_robustness";
    context->SetExtensionsString(gl_extensions);
    // The stub ctx needs to be initialized so that the gl::GLContext can
    // store the offscreen stub |surface|.
    context->Initialize(surface.get(), {});

    context->MakeCurrent(surface.get());

    GpuFeatureInfo gpu_feature_info;
    GpuDriverBugWorkarounds workarounds;
    gles2::TestHelper::SetupFeatureInfoInitExpectationsWithGLVersion(
        &gl_interface, gl_extensions, "ANGLE", gl_version, context_type);
    auto feature_info =
        base::MakeRefCounted<gles2::FeatureInfo>(workarounds, gpu_feature_info);
    feature_info->Initialize(gpu::CONTEXT_TYPE_OPENGLES2,
                             false /* passthrough */,
                             gles2::DisallowedFeatures());

    // Setup expectations for SharedContextState::InitializeGL().
    EXPECT_CALL(gl_interface, GetIntegerv(GL_MAX_VERTEX_ATTRIBS, _))
        .WillOnce(SetArgPointee<1>(8u))
        .RetiresOnSaturation();
    EXPECT_CALL(gl_interface,
                GetIntegerv(GL_MAX_COMBINED_TEXTURE_IMAGE_UNITS, _))
        .WillOnce(SetArgPointee<1>(8u))
        .RetiresOnSaturation();
    ContextStateTestHelpers::SetupInitState(&gl_interface, feature_info.get(),
                                            gfx::Size(1, 1));

    EXPECT_CALL(gl_interface, GetGraphicsResetStatusARB())
        .WillOnce(Return(GL_GUILTY_CONTEXT_RESET));

    auto shared_context_state = base::MakeRefCounted<SharedContextState>(
        new gl::GLShareGroup(), surface, context,
        false /* use_virtualized_gl_contexts */, base::DoNothing(),
        GrContextType::kGL);

    bool result =
        shared_context_state->InitializeGLWithFeatureInfo(feature_info);
    EXPECT_FALSE(result);
  }
  gl::GLSurfaceTestSupport::ShutdownGL(display);
}

TEST_F(SharedContextStateTest, GLOptionsProviderSetsCustomOptions) {
  gl::SetGLGetProcAddressProc(gl::MockGLInterface::GetGLProcAddress);
  auto* display = gl::GLSurfaceTestSupport::InitializeOneOffWithMockBindings();
  ASSERT_TRUE(display);
  {
    NiceMock<gl::MockGLInterface> gl_interface;
    gl::MockGLInterface::SetGLInterface(&gl_interface);

    InSequence sequence;

    auto share_group = base::MakeRefCounted<gl::GLShareGroup>();
    auto surface = base::MakeRefCounted<gl::GLSurfaceStub>();
    auto context = base::MakeRefCounted<gl::GLContextStub>(share_group.get());
    const char gl_version[] = "OpenGL ES 2.0";
    context->SetGLVersionString(gl_version);
    const char gl_extensions[] = "GL_KHR_robustness";
    context->SetExtensionsString(gl_extensions);

    // The stub ctx needs to be initialized so that the gl::GLContext can
    // store the offscreen stub |surface|.
    context->Initialize(surface.get(), {});
    context->MakeCurrent(surface.get());
    GpuDriverBugWorkarounds workarounds;

    auto shared_context_state = base::MakeRefCounted<SharedContextState>(
        share_group.get(), surface, context,
        false /* use_virtualized_gl_contexts */, base::DoNothing(),
        GrContextType::kGL);
    StrictMock<MockGrContextOptionsProvider> provider;
    shared_context_state->gr_context_options_provider_ = &provider;
    EXPECT_CALL(provider, SetCustomGrContextOptions(_))
        .Times(1)
        .RetiresOnSaturation();
    shared_context_state->InitializeGanesh(
        GpuPreferences(), workarounds, /*cache=*/nullptr,
        /*use_shader_cache_shm_count=*/nullptr, /*progress_reporter=*/nullptr);
  }
  gl::GLSurfaceTestSupport::ShutdownGL(display);
}

TEST_F(SharedContextStateTest, VulkanOptionsProviderSetsCustomOptions) {
  gl::SetGLGetProcAddressProc(gl::MockGLInterface::GetGLProcAddress);
  auto* display = gl::GLSurfaceTestSupport::InitializeOneOffWithMockBindings();
  ASSERT_TRUE(display);
  {
    NiceMock<gl::MockGLInterface> gl_interface;
    gl::MockGLInterface::SetGLInterface(&gl_interface);
    // This line and passing kVulkan into InitializeGanesh should ensure we go
    // down the Vulkan code path.
    gl::SetGLImplementationParts(
        gl::GLImplementationParts(gl::ANGLEImplementation::kVulkan));

    InSequence sequence;

    auto share_group = base::MakeRefCounted<gl::GLShareGroup>();
    auto surface = base::MakeRefCounted<gl::GLSurfaceStub>();
    auto context = base::MakeRefCounted<gl::GLContextStub>(share_group.get());
    const char gl_version[] = "OpenGL ES 2.0";
    context->SetGLVersionString(gl_version);
    const char gl_extensions[] = "GL_KHR_robustness";
    context->SetExtensionsString(gl_extensions);

    // The stub ctx needs to be initialized so that the gl::GLContext can
    // store the offscreen stub |surface|.
    context->Initialize(surface.get(), {});
    context->MakeCurrent(surface.get());
    GpuDriverBugWorkarounds workarounds;

    auto shared_context_state = base::MakeRefCounted<SharedContextState>(
        share_group.get(), surface, context,
        false /* use_virtualized_gl_contexts */, base::DoNothing(),
        GrContextType::kVulkan);
    StrictMock<MockGrContextOptionsProvider> provider;
    shared_context_state->gr_context_options_provider_ = &provider;
    EXPECT_CALL(provider, SetCustomGrContextOptions(_))
        .Times(1)
        .RetiresOnSaturation();
    shared_context_state->InitializeGanesh(
        GpuPreferences(), workarounds, /*cache=*/nullptr,
        /*use_shader_cache_shm_count=*/nullptr, /*progress_reporter=*/nullptr);
  }
  gl::GLSurfaceTestSupport::ShutdownGL(display);
}

}  // namespace gpu