OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "net/socket_stream/socket_stream.h" | 5 #include "net/socket_stream/socket_stream.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/bind_helpers.h" | 11 #include "base/bind_helpers.h" |
12 #include "base/callback.h" | 12 #include "base/callback.h" |
13 #include "base/utf_string_conversions.h" | 13 #include "base/utf_string_conversions.h" |
14 #include "net/base/auth.h" | 14 #include "net/base/auth.h" |
15 #include "net/base/mock_host_resolver.h" | 15 #include "net/base/mock_host_resolver.h" |
16 #include "net/base/net_log.h" | 16 #include "net/base/net_log.h" |
17 #include "net/base/net_log_unittest.h" | 17 #include "net/base/net_log_unittest.h" |
18 #include "net/base/test_completion_callback.h" | 18 #include "net/base/test_completion_callback.h" |
19 #include "net/http/http_network_session.h" | |
19 #include "net/proxy/proxy_service.h" | 20 #include "net/proxy/proxy_service.h" |
20 #include "net/socket/socket_test_util.h" | 21 #include "net/socket/socket_test_util.h" |
21 #include "net/url_request/url_request_test_util.h" | 22 #include "net/url_request/url_request_test_util.h" |
22 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
23 #include "testing/platform_test.h" | 24 #include "testing/platform_test.h" |
24 | 25 |
25 namespace { | 26 namespace { |
26 | 27 |
27 struct SocketStreamEvent { | 28 struct SocketStreamEvent { |
28 enum EventType { | 29 enum EventType { |
(...skipping 399 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
428 events[0].event_type); | 429 events[0].event_type); |
429 EXPECT_EQ(SocketStreamEvent::EVENT_AUTH_REQUIRED, events[1].event_type); | 430 EXPECT_EQ(SocketStreamEvent::EVENT_AUTH_REQUIRED, events[1].event_type); |
430 EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[2].event_type); | 431 EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[2].event_type); |
431 EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[3].event_type); | 432 EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[3].event_type); |
432 EXPECT_EQ(net::ERR_ABORTED, events[3].error_code); | 433 EXPECT_EQ(net::ERR_ABORTED, events[3].error_code); |
433 EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[4].event_type); | 434 EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[4].event_type); |
434 | 435 |
435 // TODO(eroman): Add back NetLogTest here... | 436 // TODO(eroman): Add back NetLogTest here... |
436 } | 437 } |
437 | 438 |
439 TEST_F(SocketStreamTest, BasicAuthProxyWithAuthCache) { | |
440 MockClientSocketFactory mock_socket_factory; | |
441 MockWrite data_writes[] = { | |
442 MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" | |
wtc
2012/08/14 18:34:37
Nit: the CONNECT method is usually used to set up
bashi
2012/08/14 21:45:45
WebSocket uses CONNECT even if it doesn't use ssl
| |
443 "Host: example.com\r\n" | |
444 "Proxy-Connection: keep-alive\r\n" | |
445 "Proxy-Authorization: Basic Zm9vOmJhcg==\r\n\r\n"), | |
446 }; | |
447 MockRead data_reads[] = { | |
448 MockRead("HTTP/1.1 200 Connection Established\r\n"), | |
449 MockRead("Proxy-agent: Apache/2.2.8\r\n"), | |
450 MockRead("\r\n"), | |
451 MockRead(ASYNC, ERR_IO_PENDING) | |
452 }; | |
453 StaticSocketDataProvider data(data_reads, arraysize(data_reads), | |
454 data_writes, arraysize(data_writes)); | |
455 mock_socket_factory.AddSocketDataProvider(&data); | |
456 | |
457 TestCompletionCallback test_callback; | |
458 scoped_ptr<SocketStreamEventRecorder> delegate( | |
459 new SocketStreamEventRecorder(test_callback.callback())); | |
460 delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, | |
461 base::Unretained(delegate.get()))); | |
462 | |
463 scoped_refptr<SocketStream> socket_stream( | |
464 new SocketStream(GURL("ws://example.com/demo"), delegate.get())); | |
465 | |
466 TestURLRequestContextWithProxy context("myproxy:70"); | |
467 HttpAuthCache* auth_cache = | |
468 context.http_transaction_factory()->GetSession()->http_auth_cache(); | |
469 auth_cache->Add(GURL("http://myproxy:70"), | |
470 "MyRealm1", | |
471 HttpAuth::AUTH_SCHEME_BASIC, | |
472 "Basic realm=MyRealm1", | |
473 AuthCredentials(ASCIIToUTF16("foo"), | |
474 ASCIIToUTF16("bar")), | |
475 "/"); | |
476 | |
477 socket_stream->set_context(&context); | |
478 socket_stream->SetClientSocketFactory(&mock_socket_factory); | |
479 | |
480 socket_stream->Connect(); | |
481 | |
482 test_callback.WaitForResult(); | |
483 | |
484 const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); | |
485 ASSERT_EQ(4U, events.size()); | |
486 EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, | |
487 events[0].event_type); | |
488 EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); | |
489 EXPECT_EQ(net::ERR_ABORTED, events[2].error_code); | |
490 EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[3].event_type); | |
491 } | |
492 | |
438 TEST_F(SocketStreamTest, IOPending) { | 493 TEST_F(SocketStreamTest, IOPending) { |
439 TestCompletionCallback test_callback; | 494 TestCompletionCallback test_callback; |
440 | 495 |
441 scoped_ptr<SocketStreamEventRecorder> delegate( | 496 scoped_ptr<SocketStreamEventRecorder> delegate( |
442 new SocketStreamEventRecorder(test_callback.callback())); | 497 new SocketStreamEventRecorder(test_callback.callback())); |
443 delegate->SetOnConnected(base::Bind( | 498 delegate->SetOnConnected(base::Bind( |
444 &SocketStreamTest::DoSendWebSocketHandshake, base::Unretained(this))); | 499 &SocketStreamTest::DoSendWebSocketHandshake, base::Unretained(this))); |
445 delegate->SetOnReceivedData(base::Bind( | 500 delegate->SetOnReceivedData(base::Bind( |
446 &SocketStreamTest::DoCloseFlushPendingWriteTest, | 501 &SocketStreamTest::DoCloseFlushPendingWriteTest, |
447 base::Unretained(this))); | 502 base::Unretained(this))); |
(...skipping 238 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
686 | 741 |
687 const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); | 742 const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); |
688 ASSERT_EQ(2U, events.size()); | 743 ASSERT_EQ(2U, events.size()); |
689 | 744 |
690 EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[0].event_type); | 745 EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[0].event_type); |
691 EXPECT_EQ(net::ERR_ACCESS_DENIED, events[0].error_code); | 746 EXPECT_EQ(net::ERR_ACCESS_DENIED, events[0].error_code); |
692 EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[1].event_type); | 747 EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[1].event_type); |
693 } | 748 } |
694 | 749 |
695 } // namespace net | 750 } // namespace net |
OLD | NEW |