1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
// 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
|
