mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
fc8279d66c
This is a reland of b70c5c5ce9
Original change's description:
> Using simulated rtc::Thread for peer connection scenario tests.
>
> Bug: webrtc:11255
> Change-Id: I5d29e997a7209ffc64595082358cca9b2115d07a
> Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/165689
> Commit-Queue: Sebastian Jansson <srte@webrtc.org>
> Reviewed-by: Steve Anton <steveanton@webrtc.org>
> Cr-Commit-Position: refs/heads/master@{#30258}
Bug: webrtc:11255
Change-Id: If65cd56b59158cebec5609407a721fbdb47cfd1b
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/166046
Reviewed-by: Steve Anton <steveanton@webrtc.org>
Commit-Queue: Sebastian Jansson <srte@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#30294}
91 lines
3 KiB
C++
91 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2020 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
#ifndef TEST_TIME_CONTROLLER_SIMULATED_THREAD_H_
|
|
#define TEST_TIME_CONTROLLER_SIMULATED_THREAD_H_
|
|
|
|
#include <memory>
|
|
|
|
#include "test/time_controller/simulated_time_controller.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class SimulatedThread : public rtc::Thread,
|
|
public sim_time_impl::SimulatedSequenceRunner {
|
|
public:
|
|
class CurrentThreadSetter : CurrentTaskQueueSetter {
|
|
public:
|
|
explicit CurrentThreadSetter(Thread* thread)
|
|
: CurrentTaskQueueSetter(thread),
|
|
manager_(rtc::ThreadManager::Instance()),
|
|
previous_(manager_->CurrentThread()) {
|
|
manager_->ChangeCurrentThreadForTest(thread);
|
|
}
|
|
~CurrentThreadSetter() { manager_->ChangeCurrentThreadForTest(previous_); }
|
|
|
|
private:
|
|
rtc::ThreadManager* const manager_;
|
|
rtc::Thread* const previous_;
|
|
};
|
|
SimulatedThread(sim_time_impl::SimulatedTimeControllerImpl* handler,
|
|
absl::string_view name,
|
|
std::unique_ptr<rtc::SocketServer> socket_server);
|
|
~SimulatedThread() override;
|
|
|
|
void RunReady(Timestamp at_time) override;
|
|
|
|
Timestamp GetNextRunTime() const override {
|
|
rtc::CritScope lock(&lock_);
|
|
return next_run_time_;
|
|
}
|
|
|
|
TaskQueueBase* GetAsTaskQueue() override { return this; }
|
|
|
|
// Thread interface
|
|
void Send(const rtc::Location& posted_from,
|
|
rtc::MessageHandler* phandler,
|
|
uint32_t id,
|
|
rtc::MessageData* pdata) override;
|
|
void Post(const rtc::Location& posted_from,
|
|
rtc::MessageHandler* phandler,
|
|
uint32_t id,
|
|
rtc::MessageData* pdata,
|
|
bool time_sensitive) override;
|
|
void PostDelayed(const rtc::Location& posted_from,
|
|
int delay_ms,
|
|
rtc::MessageHandler* phandler,
|
|
uint32_t id,
|
|
rtc::MessageData* pdata) override;
|
|
void PostAt(const rtc::Location& posted_from,
|
|
int64_t target_time_ms,
|
|
rtc::MessageHandler* phandler,
|
|
uint32_t id,
|
|
rtc::MessageData* pdata) override;
|
|
|
|
void Stop() override;
|
|
|
|
private:
|
|
sim_time_impl::SimulatedTimeControllerImpl* const handler_;
|
|
// Using char* to be debugger friendly.
|
|
char* name_;
|
|
rtc::CriticalSection lock_;
|
|
Timestamp next_run_time_ RTC_GUARDED_BY(lock_) = Timestamp::PlusInfinity();
|
|
};
|
|
|
|
class SimulatedMainThread : public SimulatedThread {
|
|
public:
|
|
explicit SimulatedMainThread(
|
|
sim_time_impl::SimulatedTimeControllerImpl* handler);
|
|
~SimulatedMainThread();
|
|
|
|
private:
|
|
CurrentThreadSetter current_setter_;
|
|
};
|
|
} // namespace webrtc
|
|
#endif // TEST_TIME_CONTROLLER_SIMULATED_THREAD_H_
|