Chromium Code Reviews| Index: webrtc/media/base/rtpdataengine_unittest.cc |
| diff --git a/webrtc/media/base/rtpdataengine_unittest.cc b/webrtc/media/base/rtpdataengine_unittest.cc |
| index b5cfd3255189e6444788247030fe4aefae4bd280..73e68f2229151d9e5f563aff59e16e21d9e4e86d 100644 |
| --- a/webrtc/media/base/rtpdataengine_unittest.cc |
| +++ b/webrtc/media/base/rtpdataengine_unittest.cc |
| @@ -15,28 +15,11 @@ |
| #include "webrtc/base/gunit.h" |
| #include "webrtc/base/helpers.h" |
| #include "webrtc/base/ssladapter.h" |
| -#include "webrtc/base/timing.h" |
| #include "webrtc/media/base/fakenetworkinterface.h" |
| #include "webrtc/media/base/mediaconstants.h" |
| #include "webrtc/media/base/rtpdataengine.h" |
| #include "webrtc/media/base/rtputils.h" |
| -class FakeTiming : public rtc::Timing { |
| - public: |
| - FakeTiming() : now_(0.0) {} |
| - |
| - virtual double TimerNow() { |
| - return now_; |
| - } |
| - |
| - void set_now(double now) { |
| - now_ = now; |
| - } |
| - |
| - private: |
| - double now_; |
| -}; |
| - |
| class FakeDataReceiver : public sigslot::has_slots<> { |
| public: |
| FakeDataReceiver() : has_received_data_(false) {} |
| @@ -69,18 +52,16 @@ class RtpDataMediaChannelTest : public testing::Test { |
| virtual void SetUp() { |
| // Seed needed for each test to satisfy expectations. |
| iface_.reset(new cricket::FakeNetworkInterface()); |
| - timing_ = new FakeTiming(); |
| - dme_.reset(CreateEngine(timing_)); |
| + dme_.reset(CreateEngine()); |
| receiver_.reset(new FakeDataReceiver()); |
| } |
| void SetNow(double now) { |
| - timing_->set_now(now); |
| + clock_.SetTimeNanos(now * 1e9); |
|
hbos
2016/08/30 12:56:40
Use a constant from timeutils.h.
nisse-webrtc
2016/08/30 13:20:25
Done.
|
| } |
| - cricket::RtpDataEngine* CreateEngine(FakeTiming* timing) { |
| + cricket::RtpDataEngine* CreateEngine() { |
| cricket::RtpDataEngine* dme = new cricket::RtpDataEngine(); |
| - dme->SetTiming(timing); |
| return dme; |
| } |
| @@ -143,8 +124,7 @@ class RtpDataMediaChannelTest : public testing::Test { |
| private: |
| std::unique_ptr<cricket::RtpDataEngine> dme_; |
| - // Timing passed into dme_. Owned by dme_; |
| - FakeTiming* timing_; |
| + rtc::ScopedFakeClock clock_; |
| std::unique_ptr<cricket::FakeNetworkInterface> iface_; |
| std::unique_ptr<FakeDataReceiver> receiver_; |
| }; |
| @@ -291,13 +271,10 @@ TEST_F(RtpDataMediaChannelTest, SendData) { |
| } |
| TEST_F(RtpDataMediaChannelTest, SendDataMultipleClocks) { |
| - // Timings owned by RtpDataEngines. |
| - FakeTiming* timing1 = new FakeTiming(); |
| - std::unique_ptr<cricket::RtpDataEngine> dme1(CreateEngine(timing1)); |
| + std::unique_ptr<cricket::RtpDataEngine> dme1(CreateEngine()); |
| std::unique_ptr<cricket::RtpDataMediaChannel> dmc1( |
| CreateChannel(dme1.get())); |
| - FakeTiming* timing2 = new FakeTiming(); |
| - std::unique_ptr<cricket::RtpDataEngine> dme2(CreateEngine(timing2)); |
| + std::unique_ptr<cricket::RtpDataEngine> dme2(CreateEngine()); |
| std::unique_ptr<cricket::RtpDataMediaChannel> dmc2( |
| CreateChannel(dme2.get())); |
| @@ -332,11 +309,10 @@ TEST_F(RtpDataMediaChannelTest, SendDataMultipleClocks) { |
| EXPECT_TRUE(dmc2->SendData(params2, payload, &result)); |
| // Should bump timestamp by 90000 because the clock rate is 90khz. |
| - timing1->set_now(1); |
| - // Should bump timestamp by 180000 because the clock rate is 90khz. |
| - timing2->set_now(2); |
| - |
| + SetNow(1); |
| EXPECT_TRUE(dmc1->SendData(params1, payload, &result)); |
| + // Should bump timestamp by 180000 because the clock rate is 90khz. |
| + SetNow(2); |
| EXPECT_TRUE(dmc2->SendData(params2, payload, &result)); |
| ASSERT_TRUE(HasSentData(3)); |