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
76
77
78
79
80
81
82
83
84
85
86
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ui/ozone/platform/wayland/host/wayland_exchange_data_provider.h"
#include <memory>
#include <string>
#include "base/containers/span.h"
#include "base/pickle.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/clipboard/clipboard_constants.h"
#include "ui/base/clipboard/clipboard_format_type.h"
#include "ui/base/data_transfer_policy/data_transfer_endpoint.h"
#include "url/gurl.h"
namespace ui {
namespace {
template <typename StringType>
PlatformClipboard::Data ToClipboardData(const StringType& data_string) {
return base::MakeRefCounted<base::RefCountedBytes>(
base::as_byte_span(data_string));
}
} // namespace
// Regression test for https://crbug.com/1284996.
TEST(WaylandExchangeDataProviderTest, ExtractPickledData) {
WaylandExchangeDataProvider provider;
std::string extracted;
EXPECT_FALSE(provider.ExtractData(kMimeTypePlainText, &extracted));
EXPECT_FALSE(
provider.ExtractData(kMimeTypeDataTransferCustomData, &extracted));
extracted.clear();
provider.SetString(u"dnd-string");
EXPECT_TRUE(provider.ExtractData(kMimeTypePlainText, &extracted));
EXPECT_EQ("dnd-string", extracted);
extracted.clear();
base::Pickle pickle;
pickle.WriteString("pickled-str");
provider.SetPickledData(ClipboardFormatType::DataTransferCustomType(),
pickle);
EXPECT_TRUE(
provider.ExtractData(kMimeTypeDataTransferCustomData, &extracted));
// Ensure Pickle "reconstruction" works as expected.
std::string read_pickled_str;
base::Pickle read_pickle =
base::Pickle::WithData(base::as_byte_span(extracted));
base::PickleIterator iter(read_pickle);
ASSERT_TRUE(read_pickle.data());
EXPECT_FALSE(iter.ReachedEnd());
EXPECT_TRUE(iter.ReadString(&read_pickled_str));
EXPECT_EQ("pickled-str", read_pickled_str);
}
TEST(WaylandExchangeDataProviderTest, FileContents) {
constexpr std::string kName("filename");
constexpr std::string kContents("contents");
const std::string kMimeType("application/octet-stream;name=\"filename\"");
WaylandExchangeDataProvider provider;
provider.AddData(ToClipboardData(kContents), kMimeType);
std::vector<std::string> mime_types = provider.BuildMimeTypesList();
EXPECT_THAT(mime_types, ::testing::Contains(kMimeType));
std::optional<OSExchangeDataProvider::FileContentsInfo> file_contents =
provider.GetFileContents();
EXPECT_TRUE(file_contents.has_value());
EXPECT_EQ(kName, file_contents->filename.value());
EXPECT_EQ(kContents, file_contents->file_contents);
std::string extracted;
EXPECT_TRUE(provider.ExtractData(kMimeType, &extracted));
EXPECT_EQ(kContents, extracted);
}
} // namespace ui
|
