diff options
Diffstat (limited to 'remoting/protocol/rtp_video_writer_unittest.cc')
| -rw-r--r-- | remoting/protocol/rtp_video_writer_unittest.cc | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/remoting/protocol/rtp_video_writer_unittest.cc b/remoting/protocol/rtp_video_writer_unittest.cc index 2724927..118c300 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) { |