commit ba2ba59c4ba1009ad04e56bb3804c882bc63441b
author Danil Chapovalov <danilchap@webrtc.org> Fri Sep 27 09:10:45 2019
committer Commit Bot <commit-bot@chromium.org> Fri Sep 27 10:26:04 2019
tree 3fbcbdcfb0731a6310e19cc0f5661799e13c2d6d
parent bc8049ef0bc692c1ed3cb3b665b8fa56f51492df

Rewrite test::DirectTransport to work with any TaskQueue implementation

Bug: webrtc:10933
Change-Id: Ib207a5dac57e0200f1298097edb52689c4748d07
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/154568
Commit-Queue: Danil Chapovalov <danilchap@webrtc.org>
Reviewed-by: Sebastian Jansson <srte@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#29331}

test/direct_transport.cc

diff --git a/test/direct_transport.cc b/test/direct_transport.cc
index 13d1bd2..0d626de 100644
--- a/test/direct_transport.cc
+++ b/test/direct_transport.cc
@@ -10,8 +10,11 @@
 #include "test/direct_transport.h"
 
 #include "absl/memory/memory.h"
+#include "api/task_queue/task_queue_base.h"
+#include "api/units/time_delta.h"
 #include "call/call.h"
 #include "call/fake_network_pipe.h"
+#include "rtc_base/task_utils/repeating_task.h"
 #include "rtc_base/time_utils.h"
 #include "test/rtp_header_parser.h"
 #include "test/single_threaded_task_queue.h"
@@ -37,7 +40,7 @@
 }
 
 DirectTransport::DirectTransport(
-    DEPRECATED_SingleThreadedTaskQueueForTesting* task_queue,
+    TaskQueueBase* task_queue,
     std::unique_ptr<SimulatedPacketReceiverInterface> pipe,
     Call* send_call,
     const std::map<uint8_t, MediaType>& payload_type_map)
@@ -49,14 +52,7 @@
 }
 
 DirectTransport::~DirectTransport() {
-  if (next_process_task_)
-    task_queue_->CancelTask(*next_process_task_);
-}
-
-void DirectTransport::StopSending() {
-  rtc::CritScope cs(&process_lock_);
-  if (next_process_task_)
-    task_queue_->CancelTask(*next_process_task_);
+  next_process_task_.Stop();
 }
 
 void DirectTransport::SetReceiver(PacketReceiver* receiver) {
@@ -90,7 +86,7 @@
   fake_network_->DeliverPacket(media_type, rtc::CopyOnWriteBuffer(data, length),
                                send_time_us);
   rtc::CritScope cs(&process_lock_);
-  if (!next_process_task_)
+  if (!next_process_task_.Running())
     ProcessPackets();
 }
 
@@ -107,17 +103,22 @@
 }
 
 void DirectTransport::ProcessPackets() {
-  next_process_task_.reset();
-  auto delay_ms = fake_network_->TimeUntilNextProcess();
-  if (delay_ms) {
-    next_process_task_ = task_queue_->PostDelayedTask(
-        [this]() {
-          fake_network_->Process();
-          rtc::CritScope cs(&process_lock_);
-          ProcessPackets();
-        },
-        *delay_ms);
-  }
+  absl::optional<int64_t> initial_delay_ms =
+      fake_network_->TimeUntilNextProcess();
+  if (initial_delay_ms == absl::nullopt)
+    return;
+
+  next_process_task_ = RepeatingTaskHandle::DelayedStart(
+      task_queue_, TimeDelta::ms(*initial_delay_ms), [this] {
+        fake_network_->Process();
+        if (auto delay_ms = fake_network_->TimeUntilNextProcess())
+          return TimeDelta::ms(*delay_ms);
+        // Otherwise stop the task.
+        rtc::CritScope cs(&process_lock_);
+        next_process_task_.Stop();
+        // Since this task is stopped, return value doesn't matter.
+        return TimeDelta::Zero();
+      });
 }
 }  // namespace test
 }  // namespace webrtc