// 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 "third_party/blink/renderer/platform/graphics/test/gpu_test_utils.h"

#include "components/viz/test/test_context_provider.h"
#include "components/viz/test/test_raster_interface.h"
#include "third_party/blink/renderer/platform/graphics/gpu/shared_gpu_context.h"
#include "third_party/blink/renderer/platform/graphics/test/fake_web_graphics_context_3d_provider.h"
#include "third_party/blink/renderer/platform/wtf/functional.h"

namespace blink {

void InitializeSharedGpuContext(viz::TestContextProvider* test_context_provider,
                                cc::ImageDecodeCache* cache,
                                SetIsContextLost set_context_lost) {
  auto factory = [](viz::TestRasterInterface* raster,
                    cc::ImageDecodeCache* cache,
                    viz::TestContextProvider* raster_context_provider,
                    SetIsContextLost set_context_lost)
      -> std::unique_ptr<WebGraphicsContext3DProvider> {

    if (set_context_lost == SetIsContextLost::kSetToFalse) {
      raster->set_context_lost(false);
    } else if (set_context_lost == SetIsContextLost::kSetToTrue) {
      raster->set_context_lost(true);
    }
    // else set_context_lost will not be modified

    auto context_provider = std::make_unique<FakeWebGraphicsContext3DProvider>(
        raster, cache, raster_context_provider);
    context_provider->SetCapabilities(raster->capabilities());
    return context_provider;
  };
  test_context_provider->BindToCurrentSequence();
  viz::TestRasterInterface* raster =
      test_context_provider->GetTestRasterInterface();
  SharedGpuContext::SetContextProviderFactoryForTesting(blink::BindRepeating(
      factory, blink::Unretained(raster), blink::Unretained(cache),
      blink::Unretained(test_context_provider), set_context_lost));
}

}  // namespace blink