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 <string> | 5 #include <string> |
6 | 6 |
7 #include "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
10 #include "base/message_loop/message_loop_proxy.h" | 10 #include "base/message_loop/message_loop_proxy.h" |
(...skipping 29 matching lines...) Expand all Loading... |
40 | 40 |
41 protected: | 41 protected: |
42 base::Thread client_thread_; | 42 base::Thread client_thread_; |
43 TestHttpServer server_; | 43 TestHttpServer server_; |
44 scoped_refptr<URLRequestContextGetter> context_getter_; | 44 scoped_refptr<URLRequestContextGetter> context_getter_; |
45 }; | 45 }; |
46 | 46 |
47 } // namespace | 47 } // namespace |
48 | 48 |
49 TEST_F(SyncWebSocketImplTest, CreateDestroy) { | 49 TEST_F(SyncWebSocketImplTest, CreateDestroy) { |
50 SyncWebSocketImpl sock(context_getter_); | 50 SyncWebSocketImpl sock(context_getter_.get()); |
51 } | 51 } |
52 | 52 |
53 TEST_F(SyncWebSocketImplTest, Connect) { | 53 TEST_F(SyncWebSocketImplTest, Connect) { |
54 SyncWebSocketImpl sock(context_getter_); | 54 SyncWebSocketImpl sock(context_getter_.get()); |
55 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 55 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
56 } | 56 } |
57 | 57 |
58 TEST_F(SyncWebSocketImplTest, ConnectFail) { | 58 TEST_F(SyncWebSocketImplTest, ConnectFail) { |
59 SyncWebSocketImpl sock(context_getter_); | 59 SyncWebSocketImpl sock(context_getter_.get()); |
60 ASSERT_FALSE(sock.Connect(GURL("ws://127.0.0.1:33333"))); | 60 ASSERT_FALSE(sock.Connect(GURL("ws://127.0.0.1:33333"))); |
61 } | 61 } |
62 | 62 |
63 TEST_F(SyncWebSocketImplTest, SendReceive) { | 63 TEST_F(SyncWebSocketImplTest, SendReceive) { |
64 SyncWebSocketImpl sock(context_getter_); | 64 SyncWebSocketImpl sock(context_getter_.get()); |
65 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 65 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
66 ASSERT_TRUE(sock.Send("hi")); | 66 ASSERT_TRUE(sock.Send("hi")); |
67 std::string message; | 67 std::string message; |
68 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 68 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
69 ASSERT_STREQ("hi", message.c_str()); | 69 ASSERT_STREQ("hi", message.c_str()); |
70 } | 70 } |
71 | 71 |
72 TEST_F(SyncWebSocketImplTest, SendReceiveLarge) { | 72 TEST_F(SyncWebSocketImplTest, SendReceiveLarge) { |
73 SyncWebSocketImpl sock(context_getter_); | 73 SyncWebSocketImpl sock(context_getter_.get()); |
74 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 74 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
75 std::string wrote_message(10 << 20, 'a'); | 75 std::string wrote_message(10 << 20, 'a'); |
76 ASSERT_TRUE(sock.Send(wrote_message)); | 76 ASSERT_TRUE(sock.Send(wrote_message)); |
77 std::string message; | 77 std::string message; |
78 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 78 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
79 ASSERT_EQ(wrote_message.length(), message.length()); | 79 ASSERT_EQ(wrote_message.length(), message.length()); |
80 ASSERT_EQ(wrote_message, message); | 80 ASSERT_EQ(wrote_message, message); |
81 } | 81 } |
82 | 82 |
83 TEST_F(SyncWebSocketImplTest, SendReceiveMany) { | 83 TEST_F(SyncWebSocketImplTest, SendReceiveMany) { |
84 SyncWebSocketImpl sock(context_getter_); | 84 SyncWebSocketImpl sock(context_getter_.get()); |
85 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 85 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
86 ASSERT_TRUE(sock.Send("1")); | 86 ASSERT_TRUE(sock.Send("1")); |
87 ASSERT_TRUE(sock.Send("2")); | 87 ASSERT_TRUE(sock.Send("2")); |
88 std::string message; | 88 std::string message; |
89 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 89 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
90 ASSERT_STREQ("1", message.c_str()); | 90 ASSERT_STREQ("1", message.c_str()); |
91 ASSERT_TRUE(sock.Send("3")); | 91 ASSERT_TRUE(sock.Send("3")); |
92 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 92 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
93 ASSERT_STREQ("2", message.c_str()); | 93 ASSERT_STREQ("2", message.c_str()); |
94 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 94 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
95 ASSERT_STREQ("3", message.c_str()); | 95 ASSERT_STREQ("3", message.c_str()); |
96 } | 96 } |
97 | 97 |
98 TEST_F(SyncWebSocketImplTest, CloseOnReceive) { | 98 TEST_F(SyncWebSocketImplTest, CloseOnReceive) { |
99 server_.SetMessageAction(TestHttpServer::kCloseOnMessage); | 99 server_.SetMessageAction(TestHttpServer::kCloseOnMessage); |
100 SyncWebSocketImpl sock(context_getter_); | 100 SyncWebSocketImpl sock(context_getter_.get()); |
101 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 101 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
102 ASSERT_TRUE(sock.Send("1")); | 102 ASSERT_TRUE(sock.Send("1")); |
103 std::string message; | 103 std::string message; |
104 ASSERT_FALSE(sock.ReceiveNextMessage(&message)); | 104 ASSERT_FALSE(sock.ReceiveNextMessage(&message)); |
105 ASSERT_STREQ("", message.c_str()); | 105 ASSERT_STREQ("", message.c_str()); |
106 } | 106 } |
107 | 107 |
108 TEST_F(SyncWebSocketImplTest, CloseOnSend) { | 108 TEST_F(SyncWebSocketImplTest, CloseOnSend) { |
109 SyncWebSocketImpl sock(context_getter_); | 109 SyncWebSocketImpl sock(context_getter_.get()); |
110 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 110 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
111 server_.Stop(); | 111 server_.Stop(); |
112 ASSERT_FALSE(sock.Send("1")); | 112 ASSERT_FALSE(sock.Send("1")); |
113 } | 113 } |
114 | 114 |
115 TEST_F(SyncWebSocketImplTest, Reconnect) { | 115 TEST_F(SyncWebSocketImplTest, Reconnect) { |
116 SyncWebSocketImpl sock(context_getter_); | 116 SyncWebSocketImpl sock(context_getter_.get()); |
117 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 117 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
118 ASSERT_TRUE(sock.Send("1")); | 118 ASSERT_TRUE(sock.Send("1")); |
119 // Wait for SyncWebSocket to receive the response from the server. | 119 // Wait for SyncWebSocket to receive the response from the server. |
120 base::Time deadline = base::Time::Now() + base::TimeDelta::FromSeconds(20); | 120 base::Time deadline = base::Time::Now() + base::TimeDelta::FromSeconds(20); |
121 while (base::Time::Now() < deadline) { | 121 while (base::Time::Now() < deadline) { |
122 if (sock.IsConnected() && !sock.HasNextMessage()) | 122 if (sock.IsConnected() && !sock.HasNextMessage()) |
123 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(10)); | 123 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(10)); |
124 else | 124 else |
125 break; | 125 break; |
126 } | 126 } |
127 server_.Stop(); | 127 server_.Stop(); |
128 ASSERT_FALSE(sock.Send("2")); | 128 ASSERT_FALSE(sock.Send("2")); |
129 ASSERT_FALSE(sock.IsConnected()); | 129 ASSERT_FALSE(sock.IsConnected()); |
130 server_.Start(); | 130 server_.Start(); |
131 ASSERT_TRUE(sock.HasNextMessage()); | 131 ASSERT_TRUE(sock.HasNextMessage()); |
132 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 132 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
133 ASSERT_FALSE(sock.HasNextMessage()); | 133 ASSERT_FALSE(sock.HasNextMessage()); |
134 ASSERT_TRUE(sock.Send("3")); | 134 ASSERT_TRUE(sock.Send("3")); |
135 std::string message; | 135 std::string message; |
136 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); | 136 ASSERT_TRUE(sock.ReceiveNextMessage(&message)); |
137 ASSERT_STREQ("3", message.c_str()); | 137 ASSERT_STREQ("3", message.c_str()); |
138 ASSERT_FALSE(sock.HasNextMessage()); | 138 ASSERT_FALSE(sock.HasNextMessage()); |
139 } | 139 } |
OLD | NEW |