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/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "base/single_thread_task_runner.h" | 10 #include "base/single_thread_task_runner.h" |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
69 SyncWebSocket::kOk, | 69 SyncWebSocket::kOk, |
70 sock.ReceiveNextMessage(&message, long_timeout())); | 70 sock.ReceiveNextMessage(&message, long_timeout())); |
71 ASSERT_STREQ("hi", message.c_str()); | 71 ASSERT_STREQ("hi", message.c_str()); |
72 } | 72 } |
73 | 73 |
74 TEST_F(SyncWebSocketImplTest, SendReceiveTimeout) { | 74 TEST_F(SyncWebSocketImplTest, SendReceiveTimeout) { |
75 SyncWebSocketImpl sock(context_getter_.get()); | 75 SyncWebSocketImpl sock(context_getter_.get()); |
76 | 76 |
77 // The server might reply too quickly so that the response will be received | 77 // The server might reply too quickly so that the response will be received |
78 // before we call ReceiveNextMessage; we must prevent it. | 78 // before we call ReceiveNextMessage; we must prevent it. |
79 base::WaitableEvent server_reply_allowed(false, false); | 79 base::WaitableEvent server_reply_allowed( |
| 80 base::WaitableEvent::ResetPolicy::AUTOMATIC, |
| 81 base::WaitableEvent::InitialState::NOT_SIGNALED); |
80 server_.SetMessageCallback(base::Bind( | 82 server_.SetMessageCallback(base::Bind( |
81 &base::WaitableEvent::Wait, base::Unretained(&server_reply_allowed))); | 83 &base::WaitableEvent::Wait, base::Unretained(&server_reply_allowed))); |
82 | 84 |
83 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 85 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
84 ASSERT_TRUE(sock.Send("hi")); | 86 ASSERT_TRUE(sock.Send("hi")); |
85 std::string message; | 87 std::string message; |
86 ASSERT_EQ( | 88 ASSERT_EQ( |
87 SyncWebSocket::kTimeout, | 89 SyncWebSocket::kTimeout, |
88 sock.ReceiveNextMessage( | 90 sock.ReceiveNextMessage( |
89 &message, Timeout(base::TimeDelta()))); | 91 &message, Timeout(base::TimeDelta()))); |
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
168 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); | 170 ASSERT_TRUE(sock.Connect(server_.web_socket_url())); |
169 ASSERT_FALSE(sock.HasNextMessage()); | 171 ASSERT_FALSE(sock.HasNextMessage()); |
170 ASSERT_TRUE(sock.Send("3")); | 172 ASSERT_TRUE(sock.Send("3")); |
171 std::string message; | 173 std::string message; |
172 ASSERT_EQ( | 174 ASSERT_EQ( |
173 SyncWebSocket::kOk, | 175 SyncWebSocket::kOk, |
174 sock.ReceiveNextMessage(&message, long_timeout())); | 176 sock.ReceiveNextMessage(&message, long_timeout())); |
175 ASSERT_STREQ("3", message.c_str()); | 177 ASSERT_STREQ("3", message.c_str()); |
176 ASSERT_FALSE(sock.HasNextMessage()); | 178 ASSERT_FALSE(sock.HasNextMessage()); |
177 } | 179 } |
OLD | NEW |