// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "flutter/testing/testing.h"  // IWYU pragma: keep
#include "impeller/renderer/backend/vulkan/blit_command_vk.h"
#include "impeller/renderer/backend/vulkan/command_encoder_vk.h"
#include "impeller/renderer/backend/vulkan/test/mock_vulkan.h"

namespace impeller {
namespace testing {

TEST(BlitCommandVkTest, BlitCopyTextureToTextureCommandVK) {
  auto context = MockVulkanContextBuilder().Build();
  auto pool = context->GetCommandPoolRecycler()->Get();
  auto encoder = std::make_unique<CommandEncoderFactoryVK>(context)->Create();
  BlitCopyTextureToTextureCommandVK cmd;
  cmd.source = context->GetResourceAllocator()->CreateTexture({
      .format = PixelFormat::kR8G8B8A8UNormInt,
      .size = ISize(100, 100),
  });
  cmd.destination = context->GetResourceAllocator()->CreateTexture({
      .format = PixelFormat::kR8G8B8A8UNormInt,
      .size = ISize(100, 100),
  });
  bool result = cmd.Encode(*encoder.get());
  EXPECT_TRUE(result);
  EXPECT_TRUE(encoder->IsTracking(cmd.source));
  EXPECT_TRUE(encoder->IsTracking(cmd.destination));
}

TEST(BlitCommandVkTest, BlitCopyTextureToBufferCommandVK) {
  auto context = MockVulkanContextBuilder().Build();
  auto encoder = std::make_unique<CommandEncoderFactoryVK>(context)->Create();
  BlitCopyTextureToBufferCommandVK cmd;
  cmd.source = context->GetResourceAllocator()->CreateTexture({
      .format = PixelFormat::kR8G8B8A8UNormInt,
      .size = ISize(100, 100),
  });
  cmd.destination = context->GetResourceAllocator()->CreateBuffer({
      .size = 1,
  });
  bool result = cmd.Encode(*encoder.get());
  EXPECT_TRUE(result);
  EXPECT_TRUE(encoder->IsTracking(cmd.source));
  EXPECT_TRUE(encoder->IsTracking(cmd.destination));
}

TEST(BlitCommandVkTest, BlitCopyBufferToTextureCommandVK) {
  auto context = MockVulkanContextBuilder().Build();
  auto encoder = std::make_unique<CommandEncoderFactoryVK>(context)->Create();
  BlitCopyBufferToTextureCommandVK cmd;
  cmd.destination = context->GetResourceAllocator()->CreateTexture({
      .format = PixelFormat::kR8G8B8A8UNormInt,
      .size = ISize(100, 100),
  });
  cmd.source =
      DeviceBuffer::AsBufferView(context->GetResourceAllocator()->CreateBuffer({
          .size = 1,
      }));
  bool result = cmd.Encode(*encoder.get());
  EXPECT_TRUE(result);
  EXPECT_TRUE(encoder->IsTracking(cmd.source.buffer));
  EXPECT_TRUE(encoder->IsTracking(cmd.destination));
}

TEST(BlitCommandVkTest, BlitGenerateMipmapCommandVK) {
  auto context = MockVulkanContextBuilder().Build();
  auto encoder = std::make_unique<CommandEncoderFactoryVK>(context)->Create();
  BlitGenerateMipmapCommandVK cmd;
  cmd.texture = context->GetResourceAllocator()->CreateTexture({
      .format = PixelFormat::kR8G8B8A8UNormInt,
      .size = ISize(100, 100),
      .mip_count = 2,
  });
  bool result = cmd.Encode(*encoder.get());
  EXPECT_TRUE(result);
  EXPECT_TRUE(encoder->IsTracking(cmd.texture));
}

}  // namespace testing
}  // namespace impeller