Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(227)

Side by Side Diff: device/bluetooth/bluetooth_advertisement_chromeos_unittest.cc

Issue 1415573014: Reland "Add Linux support for the Bluetooth API" (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: build fix. Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2015 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 <vector>
6
7 #include "base/bind.h"
8 #include "base/logging.h"
9 #include "base/memory/ref_counted.h"
10 #include "base/memory/scoped_ptr.h"
11 #include "base/message_loop/message_loop.h"
12 #include "device/bluetooth/bluetooth_adapter.h"
13 #include "device/bluetooth/bluetooth_adapter_factory.h"
14 #include "device/bluetooth/bluetooth_advertisement.h"
15 #include "device/bluetooth/bluetooth_advertisement_chromeos.h"
16 #include "device/bluetooth/dbus/bluez_dbus_manager.h"
17 #include "device/bluetooth/dbus/fake_bluetooth_le_advertisement_service_provider .h"
18 #include "testing/gtest/include/gtest/gtest.h"
19
20 using device::BluetoothAdapter;
21 using device::BluetoothAdapterFactory;
22 using device::BluetoothAdvertisement;
23
24 namespace chromeos {
25
26 class TestAdvertisementObserver : public BluetoothAdvertisement::Observer {
27 public:
28 explicit TestAdvertisementObserver(
29 scoped_refptr<BluetoothAdvertisement> advertisement)
30 : released_(false), advertisement_(advertisement) {
31 advertisement_->AddObserver(this);
32 }
33
34 ~TestAdvertisementObserver() override {
35 advertisement_->RemoveObserver(this);
36 }
37
38 // BluetoothAdvertisement::Observer overrides:
39 void AdvertisementReleased(BluetoothAdvertisement* advertisement) override {
40 released_ = true;
41 }
42
43 bool released() { return released_; }
44
45 private:
46 bool released_;
47 scoped_refptr<BluetoothAdvertisement> advertisement_;
48
49 DISALLOW_COPY_AND_ASSIGN(TestAdvertisementObserver);
50 };
51
52 class BluetoothAdvertisementChromeOSTest : public testing::Test {
53 public:
54 void SetUp() override {
55 bluez::BluezDBusManager::Initialize(NULL, true);
56
57 callback_count_ = 0;
58 error_callback_count_ = 0;
59
60 last_callback_count_ = 0;
61 last_error_callback_count_ = 0;
62
63 last_error_code_ = BluetoothAdvertisement::INVALID_ADVERTISEMENT_ERROR_CODE;
64
65 GetAdapter();
66 }
67
68 void TearDown() override {
69 observer_.reset();
70 // The adapter should outlive the advertisement.
71 advertisement_ = nullptr;
72 adapter_ = nullptr;
73 bluez::BluezDBusManager::Shutdown();
74 }
75
76 // Gets the existing Bluetooth adapter.
77 void GetAdapter() {
78 BluetoothAdapterFactory::GetAdapter(
79 base::Bind(&BluetoothAdvertisementChromeOSTest::GetAdapterCallback,
80 base::Unretained(this)));
81 }
82
83 // Called whenever BluetoothAdapter is retrieved successfully.
84 void GetAdapterCallback(scoped_refptr<BluetoothAdapter> adapter) {
85 adapter_ = adapter;
86 ASSERT_NE(adapter_.get(), nullptr);
87 ASSERT_TRUE(adapter_->IsInitialized());
88 }
89
90 scoped_ptr<BluetoothAdvertisement::Data> CreateAdvertisementData() {
91 scoped_ptr<BluetoothAdvertisement::Data> data =
92 make_scoped_ptr(new BluetoothAdvertisement::Data(
93 BluetoothAdvertisement::ADVERTISEMENT_TYPE_BROADCAST));
94 data->set_service_uuids(
95 make_scoped_ptr(new BluetoothAdvertisement::UUIDList()).Pass());
96 data->set_manufacturer_data(
97 make_scoped_ptr(new BluetoothAdvertisement::ManufacturerData()).Pass());
98 data->set_solicit_uuids(
99 make_scoped_ptr(new BluetoothAdvertisement::UUIDList()).Pass());
100 data->set_service_data(
101 make_scoped_ptr(new BluetoothAdvertisement::ServiceData()).Pass());
102 return data.Pass();
103 }
104
105 // Creates and registers an advertisement with the adapter.
106 scoped_refptr<BluetoothAdvertisement> CreateAdvertisement() {
107 // Clear the last advertisement we created.
108 advertisement_ = nullptr;
109
110 adapter_->RegisterAdvertisement(
111 CreateAdvertisementData().Pass(),
112 base::Bind(&BluetoothAdvertisementChromeOSTest::RegisterCallback,
113 base::Unretained(this)),
114 base::Bind(
115 &BluetoothAdvertisementChromeOSTest::AdvertisementErrorCallback,
116 base::Unretained(this)));
117
118 message_loop_.RunUntilIdle();
119 return advertisement_;
120 }
121
122 void UnregisterAdvertisement(
123 scoped_refptr<BluetoothAdvertisement> advertisement) {
124 advertisement->Unregister(
125 base::Bind(&BluetoothAdvertisementChromeOSTest::Callback,
126 base::Unretained(this)),
127 base::Bind(
128 &BluetoothAdvertisementChromeOSTest::AdvertisementErrorCallback,
129 base::Unretained(this)));
130
131 message_loop_.RunUntilIdle();
132 }
133
134 void TriggerReleased(scoped_refptr<BluetoothAdvertisement> advertisement) {
135 BluetoothAdvertisementChromeOS* adv =
136 static_cast<BluetoothAdvertisementChromeOS*>(advertisement.get());
137 bluez::FakeBluetoothLEAdvertisementServiceProvider* provider =
138 static_cast<bluez::FakeBluetoothLEAdvertisementServiceProvider*>(
139 adv->provider());
140 provider->Release();
141 }
142
143 // Called whenever RegisterAdvertisement is completed successfully.
144 void RegisterCallback(scoped_refptr<BluetoothAdvertisement> advertisement) {
145 ++callback_count_;
146 advertisement_ = advertisement;
147
148 ASSERT_NE(advertisement_.get(), nullptr);
149 }
150
151 void AdvertisementErrorCallback(
152 BluetoothAdvertisement::ErrorCode error_code) {
153 ++error_callback_count_;
154 last_error_code_ = error_code;
155 }
156
157 // Generic callbacks.
158 void Callback() { ++callback_count_; }
159
160 void ErrorCallback() { ++error_callback_count_; }
161
162 void ExpectSuccess() {
163 EXPECT_EQ(last_error_callback_count_, error_callback_count_);
164 EXPECT_EQ(last_callback_count_ + 1, callback_count_);
165 last_callback_count_ = callback_count_;
166 last_error_callback_count_ = error_callback_count_;
167 }
168
169 void ExpectError(BluetoothAdvertisement::ErrorCode error_code) {
170 EXPECT_EQ(last_callback_count_, callback_count_);
171 EXPECT_EQ(last_error_callback_count_ + 1, error_callback_count_);
172 last_callback_count_ = callback_count_;
173 last_error_callback_count_ = error_callback_count_;
174 EXPECT_EQ(error_code, last_error_code_);
175 }
176
177 protected:
178 int callback_count_;
179 int error_callback_count_;
180
181 int last_callback_count_;
182 int last_error_callback_count_;
183
184 BluetoothAdvertisement::ErrorCode last_error_code_;
185
186 base::MessageLoopForIO message_loop_;
187
188 scoped_ptr<TestAdvertisementObserver> observer_;
189 scoped_refptr<BluetoothAdapter> adapter_;
190 scoped_refptr<BluetoothAdvertisement> advertisement_;
191 };
192
193 TEST_F(BluetoothAdvertisementChromeOSTest, RegisterSucceeded) {
194 scoped_refptr<BluetoothAdvertisement> advertisement = CreateAdvertisement();
195 ExpectSuccess();
196 EXPECT_NE(nullptr, advertisement);
197
198 UnregisterAdvertisement(advertisement);
199 ExpectSuccess();
200 }
201
202 TEST_F(BluetoothAdvertisementChromeOSTest, DoubleRegisterFailed) {
203 scoped_refptr<BluetoothAdvertisement> advertisement = CreateAdvertisement();
204 ExpectSuccess();
205 EXPECT_NE(nullptr, advertisement);
206
207 // Creating a second advertisement should give us an error.
208 scoped_refptr<BluetoothAdvertisement> advertisement2 = CreateAdvertisement();
209 ExpectError(BluetoothAdvertisement::ERROR_ADVERTISEMENT_ALREADY_EXISTS);
210 EXPECT_EQ(nullptr, advertisement2);
211 }
212
213 TEST_F(BluetoothAdvertisementChromeOSTest, DoubleUnregisterFailed) {
214 scoped_refptr<BluetoothAdvertisement> advertisement = CreateAdvertisement();
215 ExpectSuccess();
216 EXPECT_NE(nullptr, advertisement);
217
218 UnregisterAdvertisement(advertisement);
219 ExpectSuccess();
220
221 // Unregistering an already unregistered advertisement should give us an
222 // error.
223 UnregisterAdvertisement(advertisement);
224 ExpectError(BluetoothAdvertisement::ERROR_ADVERTISEMENT_DOES_NOT_EXIST);
225 }
226
227 TEST_F(BluetoothAdvertisementChromeOSTest, UnregisterAfterReleasedFailed) {
228 scoped_refptr<BluetoothAdvertisement> advertisement = CreateAdvertisement();
229 ExpectSuccess();
230 EXPECT_NE(nullptr, advertisement);
231
232 observer_.reset(new TestAdvertisementObserver(advertisement));
233 TriggerReleased(advertisement);
234 EXPECT_TRUE(observer_->released());
235
236 // Unregistering an advertisement that has been released should give us an
237 // error.
238 UnregisterAdvertisement(advertisement);
239 ExpectError(BluetoothAdvertisement::ERROR_ADVERTISEMENT_DOES_NOT_EXIST);
240 }
241
242 } // namespace chromeos
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698