GetDocumentURL is added to PPB_Testing_Dev.

ppapi/tests/test_tcp_socket_private_shared.cc

 // 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 "ppapi/tests/test_tcp_socket_private_shared.h"
 
-#include <string.h>
-#include <new>
-#include <string>
 #include <vector>
 
 #include "ppapi/cpp/module.h"
 #include "ppapi/tests/test_utils.h"
 #include "ppapi/tests/testing_instance.h"
 
 REGISTER_TEST_CASE(TCPSocketPrivateShared);
 
-// TODO(ygorshenin): get rid of using external server in tests,
-// http://crbug.com/105863
-const char* const TestTCPSocketPrivateShared::kHost = "www.google.com";
-
 TestTCPSocketPrivateShared::TestTCPSocketPrivateShared(
     TestingInstance* instance)
     : TestCase(instance), tcp_socket_private_interface_(NULL) {
 }
 
 bool TestTCPSocketPrivateShared::Init() {
   tcp_socket_private_interface_ = static_cast<const PPB_TCPSocket_Private*>(
       pp::Module::Get()->GetBrowserInterface(PPB_TCPSOCKET_PRIVATE_INTERFACE));
   if (!tcp_socket_private_interface_)
     instance_->AppendError("TCPSocketPrivate interface not available");
-  return tcp_socket_private_interface_ && InitTestingInterface();
+
+  bool init_host_port = false;
+  if (!GetLocalHostPort(instance_->pp_instance(), &host_, &port_))
+    instance_->AppendError("Can't init host and port");
+  else
+    init_host_port = true;
+
+  return tcp_socket_private_interface_ &&
+      init_host_port &&
+      InitTestingInterface();
 }
 
 void TestTCPSocketPrivateShared::RunTests(const std::string& filter) {
   RUN_TEST(Create, filter);
   RUN_TEST_FORCEASYNC_AND_NOT(GetAddress, filter);
   RUN_TEST_FORCEASYNC_AND_NOT(Connect, filter);
   RUN_TEST_FORCEASYNC_AND_NOT(Reconnect, filter);
 }
 
 void TestTCPSocketPrivateShared::QuitMessageLoop() {
   testing_interface_->QuitMessageLoop(instance_->pp_instance());
 }
 
 std::string TestTCPSocketPrivateShared::CreateSocket(PP_Resource* socket) {
   *socket = tcp_socket_private_interface_->Create(instance_->pp_instance());
   if (0 == *socket)
     return "PPB_TCPSocket_Private::Create failed";
   if (!tcp_socket_private_interface_->IsTCPSocket(*socket))
     return "PPB_TCPSocket_Private::IsTCPSocket failed";
   PASS();
 }
 
 std::string TestTCPSocketPrivateShared::SyncConnect(PP_Resource socket,
-                                                    const char* host,
-                                                    int port) {
+                                                    const std::string& host,
+                                                    uint16_t port) {
   TestCompletionCallback callback(instance_->pp_instance(), force_async_);
 
   int32_t rv = tcp_socket_private_interface_->Connect(
-      socket, host, port,
+      socket, host.c_str(), port,
       static_cast<pp::CompletionCallback>(callback).pp_completion_callback());
 
   if (force_async_ && rv != PP_OK_COMPLETIONPENDING)
     return ReportError("PPB_TCPSocket_Private::Connect force_async", rv);
   if (rv == PP_OK_COMPLETIONPENDING)
     rv = callback.WaitForResult();
   if (rv != PP_OK)
     return ReportError("PPB_TCPSocket_Private::Connect", rv);
   PASS();
 }
 
 std::string TestTCPSocketPrivateShared::SyncConnectWithNetAddress(
     PP_Resource socket, const PP_NetAddress_Private& addr) {
   TestCompletionCallback callback(instance_->pp_instance(), force_async_);
 
   int32_t rv = tcp_socket_private_interface_->ConnectWithNetAddress(
       socket, &addr,
       static_cast<pp::CompletionCallback>(callback).pp_completion_callback());
 
   if (force_async_ && rv != PP_OK_COMPLETIONPENDING)
     return ReportError(
         "PPB_TCPSocket_Private::ConnectWithNetAddress force_async", rv);
   if (rv == PP_OK_COMPLETIONPENDING)
     rv = callback.WaitForResult();
   if (rv != PP_OK)
     return ReportError("PPB_TCPSocket_Private::ConnectWithNetAddress", rv);
   PASS();
 }
 
-std::string TestTCPSocketPrivateShared::SyncSSLHandshake(PP_Resource socket,
-                                                         const char* host,
-                                                         int port) {
-  TestCompletionCallback callback(instance_->pp_instance(), force_async_);
-
-  int32_t rv = tcp_socket_private_interface_->SSLHandshake(
-      socket, host, port,
-      static_cast<pp::CompletionCallback>(callback).pp_completion_callback());
-
-  if (force_async_ && rv != PP_OK_COMPLETIONPENDING)
-    return ReportError("PPB_TCPSocket_Private::SSLHandshake force_async", rv);
-  if (rv == PP_OK_COMPLETIONPENDING)
-    rv = callback.WaitForResult();
-  if (rv != PP_OK)
-    return ReportError("PPB_TCPSocket_Private::SSLHandshake", rv);
-  PASS();
-}
-
 std::string TestTCPSocketPrivateShared::SyncRead(PP_Resource socket,
                                                  char* buffer,
                                                  int32_t num_bytes,
                                                  int32_t* bytes_read) {
   TestCompletionCallback callback(instance_->pp_instance(), force_async_);
 
   int32_t rv = tcp_socket_private_interface_->Read(
       socket, buffer, num_bytes,
       static_cast<pp::CompletionCallback>(callback).pp_completion_callback());
 
@@ skipping 63 matching lines @@
 }
 
 std::string TestTCPSocketPrivateShared::TestGetAddress() {
   PP_Resource socket;
   std::string error_message;
 
   error_message = CreateSocket(&socket);
   if (!error_message.empty())
     return error_message;
 
-  error_message = SyncConnect(socket, kHost, kPort);
+  error_message = SyncConnect(socket, host_, port_);
   if (!error_message.empty())
     return error_message;
 
   PP_NetAddress_Private local_address, remote_address;
 
   if (PP_TRUE != tcp_socket_private_interface_->GetLocalAddress(
           socket, &local_address))
     return "PPB_TCPSocketPrivate::GetLocalAddress failed";
   if (PP_TRUE != tcp_socket_private_interface_->GetRemoteAddress(
           socket, &remote_address))
     return "PPB_TCPSocketPrivate::GetRemoteAddress failed";
 
   tcp_socket_private_interface_->Disconnect(socket);
 
   PASS();
 }
 
 std::string TestTCPSocketPrivateShared::TestConnect() {
   PP_Resource socket;
   std::string error_message;
 
   error_message = CreateSocket(&socket);
   if (!error_message.empty())
     return error_message;
-  error_message = SyncConnect(socket, kHost, kPort);
-  if (!error_message.empty())
-    return error_message;
-  error_message = SyncSSLHandshake(socket, kHost, kPort);
+  error_message = SyncConnect(socket, host_, port_);
   if (!error_message.empty())
     return error_message;
   error_message =
-      CheckHTTPResponse(socket, "GET /robots.txt\r\n", "HTTP/1.0 200 OK");
+      CheckHTTPResponse(socket, "GET / HTTP/1.0\r\n\r\n", "HTTP/1.0 200 OK");
   if (!error_message.empty())
     return error_message;
   tcp_socket_private_interface_->Disconnect(socket);
 
   PASS();
 }
 
 std::string TestTCPSocketPrivateShared::TestReconnect() {
   PP_Resource socket;
   std::string error_message;
 
   error_message = CreateSocket(&socket);
   if (!error_message.empty())
     return error_message;
-  error_message = SyncConnect(socket, kHost, kPort);
-  if (!error_message.empty())
-    return error_message;
-  error_message = SyncSSLHandshake(socket, kHost, kPort);
+  error_message = SyncConnect(socket, host_, port_);
   if (!error_message.empty())
     return error_message;
 
   PP_NetAddress_Private remote_address;
   if (PP_TRUE !=
       tcp_socket_private_interface_->GetRemoteAddress(socket,
                                                       &remote_address))
     return "PPB_TCPSocketPrivate::GetRemoteAddress failed";
   tcp_socket_private_interface_->Disconnect(socket);
 
   error_message = CreateSocket(&socket);
   if (!error_message.empty())
     return error_message;
   error_message = SyncConnectWithNetAddress(socket, remote_address);
   if (!error_message.empty())
     return error_message;
-  error_message = SyncSSLHandshake(socket, kHost, kPort);
-  if (!error_message.empty())
-    return error_message;
-  error_message =
-      CheckHTTPResponse(socket, "GET /robots.txt\r\n", "HTTP/1.0 200 OK");
+  error_message = CheckHTTPResponse(socket,
+                                    "GET / HTTP/1.0\r\n\r\n",
+                                    "HTTP/1.0 200 OK");
   if (!error_message.empty())
     return error_message;
   tcp_socket_private_interface_->Disconnect(socket);
 
   PASS();
 }
+
+// TODO(ygorshenin): test SSLHandshake somehow