Index: ppapi/tests/test_udp_socket_private_shared.cc |
diff --git a/ppapi/tests/test_udp_socket_private_shared.cc b/ppapi/tests/test_udp_socket_private_shared.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5707d3467f912eb0c5a53e044cdc3ae2bda7b6c4 |
--- /dev/null |
+++ b/ppapi/tests/test_udp_socket_private_shared.cc |
@@ -0,0 +1,147 @@ |
+// Copyright (c) 2011 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include <string.h> |
+ |
+#include "ppapi/cpp/module.h" |
+#include "ppapi/tests/test_udp_socket_private_shared.h" |
+#include "ppapi/tests/test_utils.h" |
+#include "ppapi/tests/testing_instance.h" |
+ |
+REGISTER_TEST_CASE(UDPSocketPrivateShared); |
+ |
+static const char* const kSuccess = ""; |
+ |
+bool TestUDPSocketPrivateShared::Init() { |
+ tcp_socket_private_interface_ = |
+ reinterpret_cast<PPB_TCPSocket_Private const*>( |
+ pp::Module::Get()->GetBrowserInterface( |
+ PPB_TCPSOCKET_PRIVATE_INTERFACE)); |
+ udp_socket_private_interface_ = |
+ reinterpret_cast<PPB_UDPSocket_Private const*>( |
+ pp::Module::Get()->GetBrowserInterface( |
+ PPB_UDPSOCKET_PRIVATE_INTERFACE)); |
+ return tcp_socket_private_interface_ && udp_socket_private_interface_ && |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ InitTestingInterface(); |
+} |
+ |
+void TestUDPSocketPrivateShared::RunTests(const std::string& filter) { |
+ RUN_TEST(Create, filter); |
+ RUN_TEST(Connect, filter); |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+} |
+ |
+void TestUDPSocketPrivateShared::QuitMessageLoop() { |
+ testing_interface_->QuitMessageLoop(instance_->pp_instance()); |
+} |
+ |
+std::string TestUDPSocketPrivateShared::GenerateNetAddress( |
+ PP_Resource* socket, PP_NetAddress_Private* address) { |
+ static const char* const kServerHost = "www.google.com"; |
polina
2011/11/29 05:20:47
would make more sense to move these to the header
ygorshenin
2011/11/29 13:25:18
Done.
|
+ static const int kServerPort = 80; |
+ |
+ *socket = tcp_socket_private_interface_->Create(instance_->pp_instance()); |
+ ASSERT_NE(0, *socket); |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ |
+ TestCompletionCallback callback(instance_->pp_instance(), true); |
+ pp::CompletionCallback cb = (pp::CompletionCallback) callback; |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ ASSERT_EQ(PP_OK_COMPLETIONPENDING, |
+ tcp_socket_private_interface_->Connect(*socket, |
+ kServerHost, kServerPort, |
+ cb.pp_completion_callback())); |
+ ASSERT_EQ(PP_OK, callback.WaitForResult()); |
+ ASSERT_EQ(PP_TRUE, |
+ tcp_socket_private_interface_->GetLocalAddress(*socket, address)); |
+ return kSuccess; |
+} |
+ |
+std::string TestUDPSocketPrivateShared::CreateAndBindUDPSocket( |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ const PP_NetAddress_Private *address, |
+ PP_Resource *socket) { |
+ *socket = udp_socket_private_interface_->Create(instance_->pp_instance()); |
+ ASSERT_NE(0, *socket); |
+ ASSERT_TRUE(udp_socket_private_interface_->IsUDPSocket(*socket)); |
+ |
+ TestCompletionCallback callback(instance_->pp_instance(), true); |
+ pp::CompletionCallback cb = (pp::CompletionCallback) callback; |
+ ASSERT_EQ(PP_OK_COMPLETIONPENDING, |
+ udp_socket_private_interface_->Bind( |
+ *socket, address, cb.pp_completion_callback()));; |
+ ASSERT_EQ(PP_OK, callback.WaitForResult()); |
+ return kSuccess; |
+} |
+ |
+std::string TestUDPSocketPrivateShared::TestCreate() { |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ PP_Resource udp_socket; |
+ std::string error_message; |
+ |
+ udp_socket = udp_socket_private_interface_->Create(0); |
+ ASSERT_EQ(0, udp_socket); |
+ udp_socket = udp_socket_private_interface_->Create(instance_->pp_instance()); |
+ ASSERT_NE(0, udp_socket); |
+ ASSERT_TRUE(udp_socket_private_interface_->IsUDPSocket(udp_socket)); |
+ |
+ PASS(); |
+} |
+ |
+std::string TestUDPSocketPrivateShared::TestConnect() { |
+ PP_NetAddress_Private server_address, client_address; |
+ PP_Resource tcp_socket_server, tcp_socket_client; |
+ std::string error_message; |
+ |
+ error_message = GenerateNetAddress(&tcp_socket_server, &server_address); |
+ if (error_message != kSuccess) |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ return error_message; |
+ error_message = GenerateNetAddress(&tcp_socket_client, &client_address); |
+ if (error_message != kSuccess) |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ return error_message; |
+ |
+ PP_Resource socket_server, socket_client; |
+ error_message = CreateAndBindUDPSocket(&server_address, &socket_server); |
+ if (error_message != kSuccess) |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ return error_message; |
+ error_message = CreateAndBindUDPSocket(&client_address, &socket_client); |
+ if (error_message != kSuccess) |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ return error_message; |
+ |
+ static const char* const kMessage = |
+ "Simple message that will be sent via UDP"; |
+ static const size_t kMessageBufferSize = 1024; |
+ char message_buffer[kMessageBufferSize]; |
+ |
+ TestCompletionCallback write_callback(instance_->pp_instance(), true); |
+ pp::CompletionCallback write_cb = (pp::CompletionCallback) write_callback; |
+ int32_t write_rv = udp_socket_private_interface_->SendTo( |
+ socket_client, |
+ kMessage, |
+ strlen(kMessage), |
+ &server_address, |
+ write_cb.pp_completion_callback()); |
+ ASSERT_EQ(PP_OK_COMPLETIONPENDING, write_rv); |
polina
2011/11/29 05:20:47
same as in other file
ygorshenin
2011/11/29 13:25:18
Done.
|
+ |
+ TestCompletionCallback read_callback(instance_->pp_instance(), true); |
+ pp::CompletionCallback read_cb = (pp::CompletionCallback) read_callback; |
+ int32_t read_rv = udp_socket_private_interface_->RecvFrom( |
+ socket_server, |
+ message_buffer, |
+ strlen(kMessage), |
+ read_cb.pp_completion_callback()); |
+ ASSERT_EQ(PP_OK_COMPLETIONPENDING, read_rv); |
+ |
+ read_rv = read_callback.WaitForResult(); |
+ ASSERT_TRUE(0 <= read_rv); |
+ ASSERT_EQ(strlen(kMessage), static_cast<size_t>(read_rv)); |
+ write_rv = write_callback.WaitForResult(); |
+ ASSERT_TRUE(0 <= write_rv); |
+ ASSERT_EQ(strlen(kMessage), static_cast<size_t>(write_rv)); |
+ |
+ ASSERT_EQ(0, strncmp(kMessage, message_buffer, strlen(kMessage))); |
+ |
+ udp_socket_private_interface_->Close(socket_server); |
+ udp_socket_private_interface_->Close(socket_client); |
+ |
+ tcp_socket_private_interface_->Disconnect(tcp_socket_server); |
+ tcp_socket_private_interface_->Disconnect(tcp_socket_client); |
+ |
+ PASS(); |
+} |