| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/protocol/webrtc_transport.h" | 5 #include "remoting/protocol/webrtc_transport.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| 11 #include "base/run_loop.h" | 11 #include "base/run_loop.h" |
| 12 #include "base/strings/string_util.h" |
| 12 #include "jingle/glue/thread_wrapper.h" | 13 #include "jingle/glue/thread_wrapper.h" |
| 13 #include "net/base/io_buffer.h" | 14 #include "net/base/io_buffer.h" |
| 14 #include "net/url_request/url_request_context_getter.h" | 15 #include "net/url_request/url_request_context_getter.h" |
| 15 #include "remoting/protocol/connection_tester.h" | 16 #include "remoting/protocol/connection_tester.h" |
| 16 #include "remoting/protocol/fake_authenticator.h" | 17 #include "remoting/protocol/fake_authenticator.h" |
| 17 #include "remoting/protocol/message_channel_factory.h" | 18 #include "remoting/protocol/message_channel_factory.h" |
| 18 #include "remoting/protocol/message_pipe.h" | 19 #include "remoting/protocol/message_pipe.h" |
| 19 #include "remoting/protocol/network_settings.h" | 20 #include "remoting/protocol/network_settings.h" |
| 20 #include "remoting/protocol/transport_context.h" | 21 #include "remoting/protocol/transport_context.h" |
| 21 #include "remoting/signaling/fake_signal_strategy.h" | 22 #include "remoting/signaling/fake_signal_strategy.h" |
| (...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 87 void TearDown() override { | 88 void TearDown() override { |
| 88 run_loop_.reset(); | 89 run_loop_.reset(); |
| 89 client_message_pipe_.reset(); | 90 client_message_pipe_.reset(); |
| 90 client_transport_.reset(); | 91 client_transport_.reset(); |
| 91 host_message_pipe_.reset(); | 92 host_message_pipe_.reset(); |
| 92 host_transport_.reset(); | 93 host_transport_.reset(); |
| 93 base::RunLoop().RunUntilIdle(); | 94 base::RunLoop().RunUntilIdle(); |
| 94 } | 95 } |
| 95 | 96 |
| 96 void ProcessTransportInfo(scoped_ptr<WebrtcTransport>* target_transport, | 97 void ProcessTransportInfo(scoped_ptr<WebrtcTransport>* target_transport, |
| 98 bool normalize_line_endings, |
| 97 scoped_ptr<buzz::XmlElement> transport_info) { | 99 scoped_ptr<buzz::XmlElement> transport_info) { |
| 98 ASSERT_TRUE(target_transport); | 100 ASSERT_TRUE(target_transport); |
| 101 |
| 102 // Reformat the message to normalize line endings by removing CR symbol. |
| 103 if (normalize_line_endings) { |
| 104 std::string xml = transport_info->Str(); |
| 105 base::ReplaceChars(xml, "\r", std::string(), &xml); |
| 106 transport_info.reset(buzz::XmlElement::ForStr(xml)); |
| 107 } |
| 108 |
| 99 EXPECT_TRUE( | 109 EXPECT_TRUE( |
| 100 (*target_transport)->ProcessTransportInfo(transport_info.get())); | 110 (*target_transport)->ProcessTransportInfo(transport_info.get())); |
| 101 } | 111 } |
| 102 | 112 |
| 103 void InitializeConnection() { | 113 void InitializeConnection() { |
| 104 host_transport_.reset( | 114 host_transport_.reset( |
| 105 new WebrtcTransport(jingle_glue::JingleThreadWrapper::current(), | 115 new WebrtcTransport(jingle_glue::JingleThreadWrapper::current(), |
| 106 TransportContext::ForTests(TransportRole::SERVER), | 116 TransportContext::ForTests(TransportRole::SERVER), |
| 107 &host_event_handler_)); | 117 &host_event_handler_)); |
| 108 host_authenticator_.reset(new FakeAuthenticator( | 118 host_authenticator_.reset(new FakeAuthenticator( |
| (...skipping 17 matching lines...) Expand all Loading... |
| 126 base::Bind(&WebrtcTransportTest::OnSessionError, base::Unretained(this), | 136 base::Bind(&WebrtcTransportTest::OnSessionError, base::Unretained(this), |
| 127 TransportRole::SERVER)); | 137 TransportRole::SERVER)); |
| 128 client_event_handler_.set_error_callback( | 138 client_event_handler_.set_error_callback( |
| 129 base::Bind(&WebrtcTransportTest::OnSessionError, base::Unretained(this), | 139 base::Bind(&WebrtcTransportTest::OnSessionError, base::Unretained(this), |
| 130 TransportRole::CLIENT)); | 140 TransportRole::CLIENT)); |
| 131 | 141 |
| 132 // Start both transports. | 142 // Start both transports. |
| 133 host_transport_->Start( | 143 host_transport_->Start( |
| 134 host_authenticator_.get(), | 144 host_authenticator_.get(), |
| 135 base::Bind(&WebrtcTransportTest::ProcessTransportInfo, | 145 base::Bind(&WebrtcTransportTest::ProcessTransportInfo, |
| 136 base::Unretained(this), &client_transport_)); | 146 base::Unretained(this), &client_transport_, true)); |
| 137 client_transport_->Start( | 147 client_transport_->Start( |
| 138 client_authenticator_.get(), | 148 client_authenticator_.get(), |
| 139 base::Bind(&WebrtcTransportTest::ProcessTransportInfo, | 149 base::Bind(&WebrtcTransportTest::ProcessTransportInfo, |
| 140 base::Unretained(this), &host_transport_)); | 150 base::Unretained(this), &host_transport_, false)); |
| 141 } | 151 } |
| 142 | 152 |
| 143 void WaitUntilConnected() { | 153 void WaitUntilConnected() { |
| 144 int counter = 2; | 154 int counter = 2; |
| 145 host_event_handler_.set_connected_callback( | 155 host_event_handler_.set_connected_callback( |
| 146 base::Bind(&WebrtcTransportTest::QuitRunLoopOnCounter, | 156 base::Bind(&WebrtcTransportTest::QuitRunLoopOnCounter, |
| 147 base::Unretained(this), &counter)); | 157 base::Unretained(this), &counter)); |
| 148 client_event_handler_.set_connected_callback( | 158 client_event_handler_.set_connected_callback( |
| 149 base::Bind(&WebrtcTransportTest::QuitRunLoopOnCounter, | 159 base::Bind(&WebrtcTransportTest::QuitRunLoopOnCounter, |
| 150 base::Unretained(this), &counter)); | 160 base::Unretained(this), &counter)); |
| (...skipping 158 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 309 run_loop_.reset(new base::RunLoop()); | 319 run_loop_.reset(new base::RunLoop()); |
| 310 run_loop_->Run(); | 320 run_loop_->Run(); |
| 311 | 321 |
| 312 // Check that OnSessionError() has been called. | 322 // Check that OnSessionError() has been called. |
| 313 EXPECT_EQ(CHANNEL_CONNECTION_ERROR, host_error_); | 323 EXPECT_EQ(CHANNEL_CONNECTION_ERROR, host_error_); |
| 314 EXPECT_FALSE(host_transport_); | 324 EXPECT_FALSE(host_transport_); |
| 315 } | 325 } |
| 316 | 326 |
| 317 } // namespace protocol | 327 } // namespace protocol |
| 318 } // namespace remoting | 328 } // namespace remoting |
| OLD | NEW |