commit 3d870658ad97ac1ac97216063171cf1a2d5a8d66
author vasilvv <vasilvv@google.com> Mon Jun 05 09:29:16 2023 -0700
committer Copybara-Service <copybara-worker@google.com> Mon Jun 05 09:30:44 2023 -0700
tree 1bf6de89aba1289d49e8d645362d72ce754ec563
parent 4ca6b8429c5c2c244516821f8175ee86b004a8bc

Switch uses of std::function to QUICHE callback aliases as appropriate.

PiperOrigin-RevId: 537898189

quiche/quic/qbone/qbone_session_test.cc

diff --git a/quiche/quic/qbone/qbone_session_test.cc b/quiche/quic/qbone/qbone_session_test.cc
index ab92c1c..46a41d9 100644
--- a/quiche/quic/qbone/qbone_session_test.cc
+++ b/quiche/quic/qbone/qbone_session_test.cc
@@ -26,6 +26,7 @@
 #include "quiche/quic/test_tools/quic_connection_peer.h"
 #include "quiche/quic/test_tools/quic_session_peer.h"
 #include "quiche/quic/test_tools/quic_test_utils.h"
+#include "quiche/common/quiche_callbacks.h"
 
 namespace quic {
 namespace test {
@@ -238,7 +239,7 @@
 
     void Run() {
       if (!cancelled_) {
-        task_();
+        std::move(task_)();
       }
     }
 
@@ -246,7 +247,7 @@
 
    private:
     bool cancelled_ = false;
-    std::function<void()> task_;
+    quiche::SingleUseCallback<void()> task_;
     QuicTime time_;
   };
 
@@ -260,8 +261,9 @@
 
  private:
   using TaskType = std::shared_ptr<InnerTask>;
-  std::priority_queue<TaskType, std::vector<TaskType>,
-                      std::function<bool(const TaskType&, const TaskType&)>>
+  std::priority_queue<
+      TaskType, std::vector<TaskType>,
+      quiche::UnretainedCallback<bool(const TaskType&, const TaskType&)>>
       tasks_;
   MockQuicConnectionHelper* helper_;
 };