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

#include "chrome/browser/ash/wallpaper_handlers/mock_sea_pen_fetcher.h"

#include <optional>
#include <string>
#include <vector>

#include "ash/public/cpp/wallpaper/sea_pen_image.h"
#include "ash/webui/common/mojom/sea_pen.mojom.h"
#include "base/containers/span.h"
#include "base/functional/bind.h"
#include "base/strings/string_view_util.h"
#include "base/strings/stringprintf.h"
#include "base/task/sequenced_task_runner.h"
#include "components/manta/manta_status.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/codec/jpeg_codec.h"
#include "ui/gfx/image/image_unittest_util.h"

namespace wallpaper_handlers {

namespace {

SkBitmap CreateBitmap() {
  return gfx::test::CreateBitmap(1, SkColorSetARGB(255, 31, 63, 127));
}

// Used in `FetchWallpaper` to create a fake JPEG image.
std::string CreateJpgBytes() {
  std::optional<std::vector<uint8_t>> data =
      gfx::JPEGCodec::Encode(CreateBitmap(), /*quality=*/100);
  return std::string(base::as_string_view(data.value()));
}

std::vector<ash::SeaPenImage> MakeFakeImageResults() {
  std::vector<ash::SeaPenImage> image_results;
  for (uint32_t i = 1; i < 5; i++) {
    image_results.emplace_back(base::StringPrintf("fake_sea_pen_image_%d", i),
                               i);
  }
  return image_results;
}

}  // namespace

MockSeaPenFetcher::MockSeaPenFetcher() {
  ON_CALL(*this, FetchThumbnails)
      .WillByDefault(
          [](manta::proto::FeatureName feature_name,
             const ash::personalization_app::mojom::SeaPenQueryPtr& query,
             OnFetchThumbnailsComplete callback) {
            base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
                FROM_HERE,
                base::BindOnce(std::move(callback), MakeFakeImageResults(),
                               manta::MantaStatusCode::kOk));
          });

  ON_CALL(*this, FetchWallpaper)
      .WillByDefault(
          [](manta::proto::FeatureName feature_name,
             const ash::SeaPenImage& image,
             const ash::personalization_app::mojom::SeaPenQueryPtr& query,
             OnFetchWallpaperComplete callback) {
            std::move(callback).Run(
                ash::SeaPenImage(CreateJpgBytes(), image.id));
          });
}

MockSeaPenFetcher::~MockSeaPenFetcher() = default;

}  // namespace wallpaper_handlers