1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "ui/base/x/selection_requestor.h"
6
7 #include "base/memory/ref_counted_memory.h"
8 #include "base/memory/scoped_ptr.h"
9 #include "base/message_loop/message_loop.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11 #include "ui/base/x/selection_utils.h"
12 #include "ui/base/x/x11_util.h"
13 #include "ui/events/platform/platform_event_source.h"
14 #include "ui/gfx/x/x11_atom_cache.h"
15 #include "ui/gfx/x/x11_types.h"
16
17 #include <X11/Xlib.h>
18
19 namespace ui {
20
21 namespace {
22
23 const char* kAtomsToCache[] = {
24 "STRING",
25 NULL
26 };
27
28 } // namespace
29
30 class SelectionRequestorTest : public testing::Test {
31 public:
SelectionRequestorTest()32 SelectionRequestorTest()
33 : x_display_(gfx::GetXDisplay()),
34 x_window_(None),
35 atom_cache_(gfx::GetXDisplay(), kAtomsToCache) {
36 atom_cache_.allow_uncached_atoms();
37 }
38
~SelectionRequestorTest()39 virtual ~SelectionRequestorTest() {
40 }
41
42 // Responds to the SelectionRequestor's XConvertSelection() request by
43 // - Setting the property passed into the XConvertSelection() request to
44 // |value|.
45 // - Sending a SelectionNotify event.
SendSelectionNotify(XAtom selection,XAtom target,const std::string & value)46 void SendSelectionNotify(XAtom selection,
47 XAtom target,
48 const std::string& value) {
49 ui::SetStringProperty(x_window_,
50 requestor_->x_property_,
51 atom_cache_.GetAtom("STRING"),
52 value);
53
54 XEvent xev;
55 xev.type = SelectionNotify;
56 xev.xselection.serial = 0u;
57 xev.xselection.display = x_display_;
58 xev.xselection.requestor = x_window_;
59 xev.xselection.selection = selection;
60 xev.xselection.target = target;
61 xev.xselection.property = requestor_->x_property_;
62 xev.xselection.time = CurrentTime;
63 xev.xselection.type = SelectionNotify;
64 requestor_->OnSelectionNotify(xev);
65 }
66
67 protected:
SetUp()68 virtual void SetUp() OVERRIDE {
69 // Make X11 synchronous for our display connection.
70 XSynchronize(x_display_, True);
71
72 // Create a window for the selection requestor to use.
73 x_window_ = XCreateWindow(x_display_,
74 DefaultRootWindow(x_display_),
75 0, 0, 10, 10, // x, y, width, height
76 0, // border width
77 CopyFromParent, // depth
78 InputOnly,
79 CopyFromParent, // visual
80 0,
81 NULL);
82
83 event_source_ = ui::PlatformEventSource::CreateDefault();
84 CHECK(ui::PlatformEventSource::GetInstance());
85 requestor_.reset(new SelectionRequestor(x_display_, x_window_, NULL));
86 }
87
TearDown()88 virtual void TearDown() OVERRIDE {
89 requestor_.reset();
90 event_source_.reset();
91 XDestroyWindow(x_display_, x_window_);
92 XSynchronize(x_display_, False);
93 }
94
95 Display* x_display_;
96
97 // |requestor_|'s window.
98 XID x_window_;
99
100 scoped_ptr<ui::PlatformEventSource> event_source_;
101 scoped_ptr<SelectionRequestor> requestor_;
102
103 base::MessageLoopForUI message_loop_;
104 X11AtomCache atom_cache_;
105
106 private:
107 DISALLOW_COPY_AND_ASSIGN(SelectionRequestorTest);
108 };
109
110 namespace {
111
112 // Converts |selection| to |target| and checks the returned values.
PerformBlockingConvertSelection(SelectionRequestor * requestor,X11AtomCache * atom_cache,XAtom selection,XAtom target,const std::string & expected_data)113 void PerformBlockingConvertSelection(SelectionRequestor* requestor,
114 X11AtomCache* atom_cache,
115 XAtom selection,
116 XAtom target,
117 const std::string& expected_data) {
118 scoped_refptr<base::RefCountedMemory> out_data;
119 size_t out_data_items = 0u;
120 XAtom out_type = None;
121 EXPECT_TRUE(requestor->PerformBlockingConvertSelection(
122 selection, target, &out_data, &out_data_items, &out_type));
123 EXPECT_EQ(expected_data, ui::RefCountedMemoryToString(out_data));
124 EXPECT_EQ(expected_data.size(), out_data_items);
125 EXPECT_EQ(atom_cache->GetAtom("STRING"), out_type);
126 }
127
128 } // namespace
129
130 // Test that SelectionRequestor correctly handles receiving a request while it
131 // is processing another request.
TEST_F(SelectionRequestorTest,NestedRequests)132 TEST_F(SelectionRequestorTest, NestedRequests) {
133 // Assume that |selection| will have no owner. If there is an owner, the owner
134 // will set the property passed into the XConvertSelection() request which is
135 // undesirable.
136 XAtom selection = atom_cache_.GetAtom("FAKE_SELECTION");
137
138 XAtom target1 = atom_cache_.GetAtom("TARGET1");
139 XAtom target2 = atom_cache_.GetAtom("TARGET2");
140
141 base::MessageLoopForUI* loop = base::MessageLoopForUI::current();
142 loop->PostTask(FROM_HERE,
143 base::Bind(&PerformBlockingConvertSelection,
144 base::Unretained(requestor_.get()),
145 base::Unretained(&atom_cache_),
146 selection,
147 target2,
148 "Data2"));
149 loop->PostTask(FROM_HERE,
150 base::Bind(&SelectionRequestorTest::SendSelectionNotify,
151 base::Unretained(this),
152 selection,
153 target1,
154 "Data1"));
155 loop->PostTask(FROM_HERE,
156 base::Bind(&SelectionRequestorTest::SendSelectionNotify,
157 base::Unretained(this),
158 selection,
159 target2,
160 "Data2"));
161 PerformBlockingConvertSelection(
162 requestor_.get(), &atom_cache_, selection, target1, "Data1");
163 }
164
165 } // namespace ui
166