| 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 |