f5f03e823c
When the fake clock's time is advanced, it now ensures all pending queued messages have been dispatched. This allows us to write a "SIMULATED_WAIT" macro that ticks the simulated clock by milliseconds up until the target time. Useful in this case, where we know the STUN timeout should take a total of 9500ms, but it would be overly complex to write test code that waits for each individual timeout, ensures a STUN packet has been retransmited, etc. (The test described above *should* be written, but it belongs in p2ptransportchannel_unittest.cc, not webrtcsession_unittest.cc). Review-Url: https://codereview.webrtc.org/2024813004 Cr-Commit-Position: refs/heads/master@{#13052}
57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
/*
|
|
* Copyright 2016 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 WEBRTC_BASE_FAKECLOCK_H_
|
|
#define WEBRTC_BASE_FAKECLOCK_H_
|
|
|
|
#include "webrtc/base/criticalsection.h"
|
|
#include "webrtc/base/timedelta.h"
|
|
#include "webrtc/base/timeutils.h"
|
|
|
|
namespace rtc {
|
|
|
|
// Fake clock for use with unit tests, which does not tick on its own.
|
|
// Starts at time 0.
|
|
//
|
|
// TODO(deadbeef): Unify with webrtc::SimulatedClock.
|
|
class FakeClock : public ClockInterface {
|
|
public:
|
|
~FakeClock() override {}
|
|
|
|
// ClockInterface implementation.
|
|
uint64_t TimeNanos() const override;
|
|
|
|
// Methods that can be used by the test to control the time.
|
|
|
|
// Should only be used to set a time in the future.
|
|
void SetTimeNanos(uint64_t nanos);
|
|
|
|
void AdvanceTime(TimeDelta delta);
|
|
|
|
private:
|
|
CriticalSection lock_;
|
|
uint64_t time_ GUARDED_BY(lock_) = 0u;
|
|
};
|
|
|
|
// Helper class that sets itself as the global clock in its constructor and
|
|
// unsets it in its destructor.
|
|
class ScopedFakeClock : public FakeClock {
|
|
public:
|
|
ScopedFakeClock();
|
|
~ScopedFakeClock() override;
|
|
|
|
private:
|
|
ClockInterface* prev_clock_;
|
|
};
|
|
|
|
} // namespace rtc
|
|
|
|
#endif // WEBRTC_BASE_FAKECLOCK_H_
|