| Index: remoting/protocol/rtp_video_writer_unittest.cc
|
| diff --git a/remoting/protocol/rtp_video_writer_unittest.cc b/remoting/protocol/rtp_video_writer_unittest.cc
|
| index 272492745320d558fa5277c1c52c05ce5074906c..118c3005ea6e81f08a00766c36f5083b271d82f6 100644
|
| --- a/remoting/protocol/rtp_video_writer_unittest.cc
|
| +++ b/remoting/protocol/rtp_video_writer_unittest.cc
|
| @@ -5,8 +5,10 @@
|
| #include <string>
|
| #include <vector>
|
|
|
| +#include "base/bind.h"
|
| #include "base/message_loop.h"
|
| #include "base/string_number_conversions.h"
|
| +#include "remoting/base/constants.h"
|
| #include "remoting/proto/video.pb.h"
|
| #include "remoting/protocol/fake_session.h"
|
| #include "remoting/protocol/rtp_reader.h"
|
| @@ -57,7 +59,13 @@ class RtpVideoWriterTest : public testing::Test {
|
|
|
| virtual void SetUp() {
|
| session_.reset(new FakeSession());
|
| - writer_.Init(session_.get());
|
| + writer_.Init(session_.get(),
|
| + base::Bind(&RtpVideoWriterTest::OnWriterInitialized,
|
| + base::Unretained(this)));
|
| + }
|
| +
|
| + void OnWriterInitialized(bool success) {
|
| + ASSERT_TRUE(success);
|
| }
|
|
|
| void InitData(int size) {
|
| @@ -88,7 +96,7 @@ class RtpVideoWriterTest : public testing::Test {
|
| void VerifyResult(const ExpectedPacket expected[],
|
| int count) {
|
| const vector<string>& rtp_packets =
|
| - session_->video_rtp_channel()->written_packets();
|
| + session_->GetDatagramChannel(kVideoRtpChannelName)->written_packets();
|
| ASSERT_EQ(count, static_cast<int>(rtp_packets.size()));
|
| int pos = 0;
|
| for (int i = 0; i < count; ++i) {
|
|
|