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) { |