bluetooth_task_manager_win_unittest.cc 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. // Copyright (c) 2012 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. #include <stddef.h>
  5. #include "base/bind.h"
  6. #include "base/callback.h"
  7. #include "base/memory/ref_counted.h"
  8. #include "base/test/test_pending_task.h"
  9. #include "base/test/test_simple_task_runner.h"
  10. #include "device/bluetooth/bluetooth_classic_win.h"
  11. #include "device/bluetooth/bluetooth_init_win.h"
  12. #include "device/bluetooth/bluetooth_task_manager_win.h"
  13. #include "testing/gtest/include/gtest/gtest.h"
  14. namespace {
  15. class BluetoothTaskObserver : public device::BluetoothTaskManagerWin::Observer {
  16. public:
  17. BluetoothTaskObserver()
  18. : num_adapter_state_changed_(0),
  19. num_discovery_started_(0),
  20. num_discovery_stopped_(0) {
  21. }
  22. ~BluetoothTaskObserver() override {}
  23. void AdapterStateChanged(
  24. const device::BluetoothTaskManagerWin::AdapterState& state) override {
  25. num_adapter_state_changed_++;
  26. }
  27. void DiscoveryStarted(bool success) override { num_discovery_started_++; }
  28. void DiscoveryStopped() override { num_discovery_stopped_++; }
  29. int num_adapter_state_changed() const {
  30. return num_adapter_state_changed_;
  31. }
  32. int num_discovery_started() const {
  33. return num_discovery_started_;
  34. }
  35. int num_discovery_stopped() const {
  36. return num_discovery_stopped_;
  37. }
  38. private:
  39. int num_adapter_state_changed_;
  40. int num_discovery_started_;
  41. int num_discovery_stopped_;
  42. };
  43. } // namespace
  44. namespace device {
  45. class BluetoothTaskManagerWinTest : public testing::Test {
  46. public:
  47. BluetoothTaskManagerWinTest()
  48. : ui_task_runner_(new base::TestSimpleTaskRunner()),
  49. bluetooth_task_runner_(new base::TestSimpleTaskRunner()),
  50. task_manager_(new BluetoothTaskManagerWin(ui_task_runner_)),
  51. has_bluetooth_stack_(device::bluetooth_init_win::HasBluetoothStack()) {
  52. task_manager_->InitializeWithBluetoothTaskRunner(bluetooth_task_runner_);
  53. }
  54. void SetUp() override { task_manager_->AddObserver(&observer_); }
  55. void TearDown() override { task_manager_->RemoveObserver(&observer_); }
  56. int GetPollingIntervalMs() const {
  57. return BluetoothTaskManagerWin::kPollIntervalMs;
  58. }
  59. protected:
  60. scoped_refptr<base::TestSimpleTaskRunner> ui_task_runner_;
  61. scoped_refptr<base::TestSimpleTaskRunner> bluetooth_task_runner_;
  62. scoped_refptr<BluetoothTaskManagerWin> task_manager_;
  63. BluetoothTaskObserver observer_;
  64. const bool has_bluetooth_stack_;
  65. };
  66. TEST_F(BluetoothTaskManagerWinTest, StartPolling) {
  67. EXPECT_EQ(1u, bluetooth_task_runner_->NumPendingTasks());
  68. }
  69. TEST_F(BluetoothTaskManagerWinTest, PollAdapterIfBluetoothStackIsAvailable) {
  70. bluetooth_task_runner_->RunPendingTasks();
  71. size_t num_expected_pending_tasks = has_bluetooth_stack_ ? 1 : 0;
  72. EXPECT_EQ(num_expected_pending_tasks,
  73. bluetooth_task_runner_->NumPendingTasks());
  74. }
  75. TEST_F(BluetoothTaskManagerWinTest, Polling) {
  76. if (!has_bluetooth_stack_)
  77. return;
  78. int num_polls = 5;
  79. for (int i = 0; i < num_polls; i++) {
  80. bluetooth_task_runner_->RunPendingTasks();
  81. }
  82. ui_task_runner_->RunPendingTasks();
  83. EXPECT_EQ(num_polls, observer_.num_adapter_state_changed());
  84. }
  85. TEST_F(BluetoothTaskManagerWinTest, SetPowered) {
  86. if (!has_bluetooth_stack_)
  87. return;
  88. bluetooth_task_runner_->ClearPendingTasks();
  89. task_manager_->PostSetPoweredBluetoothTask(true, base::OnceClosure(),
  90. base::OnceClosure());
  91. EXPECT_EQ(1u, bluetooth_task_runner_->NumPendingTasks());
  92. bluetooth_task_runner_->RunPendingTasks();
  93. EXPECT_TRUE(ui_task_runner_->NumPendingTasks() >= 1);
  94. }
  95. TEST_F(BluetoothTaskManagerWinTest, Discovery) {
  96. if (!has_bluetooth_stack_)
  97. return;
  98. bluetooth_task_runner_->RunPendingTasks();
  99. bluetooth_task_runner_->ClearPendingTasks();
  100. task_manager_->PostStartDiscoveryTask();
  101. bluetooth_task_runner_->RunPendingTasks();
  102. ui_task_runner_->RunPendingTasks();
  103. EXPECT_EQ(1, observer_.num_discovery_started());
  104. task_manager_->PostStopDiscoveryTask();
  105. bluetooth_task_runner_->RunPendingTasks();
  106. ui_task_runner_->RunPendingTasks();
  107. EXPECT_EQ(1, observer_.num_discovery_stopped());
  108. }
  109. } // namespace device