test_callback_receiver_unittest.cc 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. // Copyright 2018 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 "device/fido/test_callback_receiver.h"
  5. #include <memory>
  6. #include <tuple>
  7. #include <utility>
  8. #include "base/test/bind.h"
  9. #include "base/test/task_environment.h"
  10. #include "base/threading/thread_task_runner_handle.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. namespace device {
  13. namespace test {
  14. namespace {
  15. // Simple class that resets a boolean flag when moved out from.
  16. class MoveResets {
  17. public:
  18. MoveResets() = default;
  19. MoveResets(bool value) : value_(value) {} // NOLINT(runtime/explicit)
  20. MoveResets(const MoveResets&) = default;
  21. MoveResets(MoveResets&& other) : value_(std::exchange(other.value_, false)) {}
  22. MoveResets& operator=(const MoveResets&) = default;
  23. MoveResets& operator=(MoveResets&& other) {
  24. value_ = std::exchange(other.value_, false);
  25. return *this;
  26. }
  27. ~MoveResets() = default;
  28. bool value() const { return value_; }
  29. private:
  30. bool value_ = false;
  31. };
  32. } // namespace
  33. TEST(TestCallbackReceiver, BasicClosure) {
  34. base::test::TaskEnvironment task_environment;
  35. TestCallbackReceiver<> closure_receiver;
  36. auto closure = closure_receiver.callback();
  37. EXPECT_FALSE(closure_receiver.was_called());
  38. EXPECT_FALSE(closure_receiver.result().has_value());
  39. EXPECT_TRUE(closure);
  40. std::move(closure).Run();
  41. EXPECT_TRUE(closure_receiver.was_called());
  42. EXPECT_TRUE(closure_receiver.result().has_value());
  43. EXPECT_EQ(std::tuple<>(), closure_receiver.TakeResult());
  44. EXPECT_TRUE(closure_receiver.was_called());
  45. EXPECT_FALSE(closure_receiver.result().has_value());
  46. }
  47. TEST(TestCallbackReceiver, BasicCopyableArgument) {
  48. base::test::TaskEnvironment task_environment;
  49. TestCallbackReceiver<int> callback_receiver;
  50. auto callback = callback_receiver.callback();
  51. EXPECT_FALSE(callback_receiver.was_called());
  52. EXPECT_FALSE(callback_receiver.result().has_value());
  53. EXPECT_TRUE(callback);
  54. std::move(callback).Run(42);
  55. EXPECT_TRUE(callback_receiver.was_called());
  56. EXPECT_TRUE(callback_receiver.result().has_value());
  57. EXPECT_EQ(std::make_tuple<int>(42), callback_receiver.TakeResult());
  58. EXPECT_TRUE(callback_receiver.was_called());
  59. EXPECT_FALSE(callback_receiver.result().has_value());
  60. }
  61. TEST(TestCallbackReceiver, MoveOnlyArgumentIsMoved) {
  62. base::test::TaskEnvironment task_environment;
  63. TestCallbackReceiver<std::unique_ptr<int>> callback_receiver;
  64. auto callback = callback_receiver.callback();
  65. std::move(callback).Run(std::make_unique<int>(42));
  66. auto result = callback_receiver.TakeResult();
  67. static_assert(std::tuple_size<decltype(result)>::value == 1u,
  68. "The returned tuple should be one-dimensional");
  69. ASSERT_TRUE(std::get<0>(result));
  70. EXPECT_EQ(42, *std::get<0>(result));
  71. }
  72. TEST(TestCallbackReceiver, ReferenceArgumentIsCopied) {
  73. base::test::TaskEnvironment task_environment;
  74. TestCallbackReceiver<MoveResets&> callback_receiver;
  75. MoveResets passed_in_value = true;
  76. auto callback = callback_receiver.callback();
  77. EXPECT_FALSE(callback_receiver.result().has_value());
  78. // Make sure |passed_in_value| is not moved from.
  79. std::move(callback).Run(passed_in_value);
  80. EXPECT_TRUE(passed_in_value.value());
  81. EXPECT_TRUE(callback_receiver.result().has_value());
  82. const MoveResets& received_value = std::get<0>(*callback_receiver.result());
  83. EXPECT_EQ(passed_in_value.value(), received_value.value());
  84. // Make sure |received_value| is not a reference to |passed_in_value|.
  85. passed_in_value = false;
  86. EXPECT_NE(passed_in_value.value(), received_value.value());
  87. callback_receiver.TakeResult();
  88. EXPECT_FALSE(callback_receiver.result().has_value());
  89. }
  90. TEST(TestCallbackReceiver, StatusAndValue) {
  91. enum class TestStatus { NOT_OK, OK };
  92. base::test::TaskEnvironment task_environment;
  93. StatusAndValueCallbackReceiver<TestStatus, std::unique_ptr<int>>
  94. callback_receiver;
  95. auto callback = callback_receiver.callback();
  96. EXPECT_FALSE(callback_receiver.was_called());
  97. EXPECT_TRUE(callback);
  98. std::move(callback).Run(TestStatus::OK, std::make_unique<int>(1337));
  99. EXPECT_TRUE(callback_receiver.was_called());
  100. EXPECT_EQ(TestStatus::OK, callback_receiver.status());
  101. EXPECT_EQ(1337, *callback_receiver.value());
  102. }
  103. TEST(TestCallbackReceiver, WaitForCallback) {
  104. base::test::TaskEnvironment task_environment;
  105. TestCallbackReceiver<> closure_receiver;
  106. auto closure = closure_receiver.callback();
  107. EXPECT_FALSE(closure_receiver.was_called());
  108. EXPECT_TRUE(closure);
  109. base::ThreadTaskRunnerHandle::Get()->PostTask(
  110. FROM_HERE,
  111. base::BindLambdaForTesting([&closure]() { std::move(closure).Run(); }));
  112. EXPECT_FALSE(closure_receiver.was_called());
  113. closure_receiver.WaitForCallback();
  114. EXPECT_TRUE(closure_receiver.was_called());
  115. }
  116. TEST(TestCallbackReceiver, WaitForCallbackAfterCallback) {
  117. base::test::TaskEnvironment task_environment;
  118. TestCallbackReceiver<> closure_receiver;
  119. auto closure = closure_receiver.callback();
  120. EXPECT_FALSE(closure_receiver.was_called());
  121. EXPECT_TRUE(closure);
  122. std::move(closure).Run();
  123. EXPECT_TRUE(closure_receiver.was_called());
  124. closure_receiver.WaitForCallback();
  125. EXPECT_TRUE(closure_receiver.was_called());
  126. }
  127. } // namespace test
  128. } // namespace device