Pepper: Add a test for PPB_TCPSocket_Private.

ppapi/tests/test_tcp_socket_private.cc

Index: ppapi/tests/test_tcp_socket_private.cc
diff --git a/ppapi/tests/test_tcp_socket_private.cc b/ppapi/tests/test_tcp_socket_private.cc
new file mode 100644
index 0000000000000000000000000000000000000000..e5cc561554cc200aae0681f19113cf1c42af3ab3
--- /dev/null
+++ b/ppapi/tests/test_tcp_socket_private.cc
@@ -0,0 +1,218 @@
+// 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.h"
+
+#include <stdlib.h>
+
+#include <string>
+#include <vector>
+//FIXME

[yzshen1, 2011/11/15 00:47:07]

Please use "// TODO(somebody): What to do."

[viettrungluu, 2011/11/15 01:08:26]

Oops, leftover cruft.

+#include <iostream>
+
+#include "base/string_split.h"
+#include "ppapi/c/dev/ppb_url_util_dev.h"
+#include "ppapi/cpp/dev/url_util_dev.h"
+#include "ppapi/cpp/private/flash_tcp_socket.h"
+#include "ppapi/c/pp_stdint.h"

[yzshen1, 2011/11/15 00:47:07]

This is not needed since it has been included in .h file. (And also line 9.)

[viettrungluu, 2011/11/15 01:08:26]

Fixed. (But, AFAIK, the C++ headers don't necessarily define uint16_t (or
int32_t, for that matter).)

[yzshen1, 2011/11/15 01:16:56]

What do you mean by "C++ headers"?

+#include "ppapi/cpp/var.h"
+#include "ppapi/tests/testing_instance.h"
+#include "ppapi/tests/test_utils.h"
+
+namespace {
+
+// Validates the first line of an HTTP response.
+bool ValidateHttpResponse(const std::string& s) {
+  // Just check that it begins with "HTTP/" and ends with a "\r\n".
+  return s.size() >= 5 &&
+         s.substr(0, 5) == "HTTP/" &&
+         s.substr(s.size() - 2) == "\r\n";
+}
+
+}  // namespace
+
+REGISTER_TEST_CASE(TCPSocketPrivate);
+
+TestTCPSocketPrivate::TestTCPSocketPrivate(TestingInstance* instance)
+    : TestCase(instance) {
+}
+
+bool TestTCPSocketPrivate::Init() {
+  if (!TCPSocketPrivate::IsAvailable())
+    return false;
+
+  // This test currently only works out-of-process (since the API is really only
+  // implemented in that case).
+  const PPB_Testing_Dev* testing = GetTestingInterface();
+  if (!testing)
+    return false;
+  if (!testing->IsOutOfProcess())
+    return false;
+
+  // We need something to connect to, so we connect to the HTTP server whence we

[yzshen1, 2011/11/15 00:47:07]

Haha!
So what is your plan to test the UDP API? :)

[viettrungluu, 2011/11/15 01:08:26]

?!? Who knows. I don't even know how I'll properly test SSL.

+  // came. Grab the host and port.
+
+  PP_URLComponents_Dev c;

[yzshen1, 2011/11/15 00:47:07]

This name is not very descriptive.

[viettrungluu, 2011/11/15 01:08:26]

But it's wonderfully short? I changed it to "components".

+  pp::Var pp_url = pp::URLUtil_Dev::Get()->GetDocumentURL(*instance_, &c);
+  if (!pp_url.is_string())
+    return false;
+  std::string url = pp_url.AsString();
+
+  // Only support HTTP schGeme.
+  if (c.scheme.len < 0)
+    return false;
+  if (url.substr(c.scheme.begin, c.scheme.len) != "http")
+    return false;

[yzshen1, 2011/11/15 00:47:07]

You may want to use EnsureRunningOverHTTP

[viettrungluu, 2011/11/15 01:08:26]

I'll do that in addition to what I do here....

+
+  // Get host.
+  if (c.host.len < 0)
+    return false;
+  host_ = url.substr(c.host.begin, c.host.len);
+
+  // Get port (it's optional).
+  port_ = 80;  // Default value.
+  if (c.port.len > 0) {
+    int i = atoi(url.substr(c.port.begin, c.port.len).c_str());
+    if (i < 0 || i > 65535)
+      return false;
+    port_ = static_cast<uint16_t>(i);
+  }
+
+  return true;
+}
+
+void TestTCPSocketPrivate::RunTest() {
+  RUN_TEST_FORCEASYNC_AND_NOT(Basic);
+  RUN_TEST_FORCEASYNC_AND_NOT(ReadWrite);
+  RUN_TEST_FORCEASYNC_AND_NOT(ConnectAddress);
+}
+
+std::string TestTCPSocketPrivate::TestBasic() {
+  TCPSocketPrivate socket(instance_);
+  TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+
+  int32_t rv = socket.Connect(host_.c_str(), port_, cb);
+  ASSERT_TRUE(!force_async_ || rv == PP_OK_COMPLETIONPENDING);
+  if (rv == PP_OK_COMPLETIONPENDING)
+    rv = cb.WaitForResult();
+  ASSERT_EQ(PP_OK, rv);
+
+  PP_NetAddress_Private unused;
+  // TODO(viettrungluu): check the values somehow.
+  ASSERT_TRUE(socket.GetLocalAddress(&unused));
+  ASSERT_TRUE(socket.GetRemoteAddress(&unused));
+
+  socket.Disconnect();
+
+  PASS();
+}
+
+std::string TestTCPSocketPrivate::TestReadWrite() {
+  TCPSocketPrivate socket(instance_);
+  TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+
+  int32_t rv = socket.Connect(host_.c_str(), port_, cb);
+  ASSERT_TRUE(!force_async_ || rv == PP_OK_COMPLETIONPENDING);
+  if (rv == PP_OK_COMPLETIONPENDING)
+    rv = cb.WaitForResult();
+  ASSERT_EQ(PP_OK, rv);
+
+  ASSERT_EQ(PP_OK, WriteStringToSocket(&socket, "GET / HTTP/1.0\r\n\r\n"));
+
+  // Read up to the first \n and check that it looks like valid HTTP response.
+  std::string s;
+  ASSERT_EQ(PP_OK, ReadFirstLineFromSocket(&socket, &s));
+  ASSERT_TRUE(ValidateHttpResponse(s));
+
+  socket.Disconnect();
+
+  PASS();
+}
+
+std::string TestTCPSocketPrivate::TestConnectAddress() {
+  PP_NetAddress_Private address;
+
+  // First, bring up a connection and grab the address.
+  {
+    TCPSocketPrivate socket(instance_);
+    TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+    int32_t rv = socket.Connect(host_.c_str(), port_, cb);
+    ASSERT_TRUE(!force_async_ || rv == PP_OK_COMPLETIONPENDING);
+    if (rv == PP_OK_COMPLETIONPENDING)
+      rv = cb.WaitForResult();
+    ASSERT_EQ(PP_OK, rv);
+    ASSERT_TRUE(socket.GetRemoteAddress(&address));
+    // Omit the |Disconnect()| here to make sure we don't crash if we just let
+    // the resource be destroyed.
+  }
+
+  // Connect to that address.
+  TCPSocketPrivate socket(instance_);
+  TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+  int32_t rv = socket.ConnectWithNetAddress(&address, cb);
+  ASSERT_TRUE(!force_async_ || rv == PP_OK_COMPLETIONPENDING);
+  if (rv == PP_OK_COMPLETIONPENDING)
+    rv = cb.WaitForResult();
+  ASSERT_EQ(PP_OK, rv);
+
+  // Make sure we can read/write to it properly (see |TestReadWrite()|).
+  ASSERT_EQ(PP_OK, WriteStringToSocket(&socket, "GET / HTTP/1.0\r\n\r\n"));
+  std::string s;
+  ASSERT_EQ(PP_OK, ReadFirstLineFromSocket(&socket, &s));
+  ASSERT_TRUE(ValidateHttpResponse(s));
+
+  socket.Disconnect();
+
+  PASS();
+}
+
+// TODO(viettrungluu): Try testing SSL somehow.
+
+int32_t TestTCPSocketPrivate::ReadFirstLineFromSocket(TCPSocketPrivate* socket,
+                                                      std::string* s) {
+  char buffer[10000];
+
+  s->clear();
+  // Make sure we don't just hang if |Read()| spews.
+  while (s->size() < 1000000) {
+    TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+    int32_t rv = socket->Read(buffer, sizeof(buffer), cb);
+    if (force_async_ && rv != PP_OK_COMPLETIONPENDING)
+      return PP_ERROR_FAILED;
+    if (rv == PP_OK_COMPLETIONPENDING)
+      rv = cb.WaitForResult();
+    if (rv < 0)
+      return rv;
+    if (rv == 0)
+      return PP_ERROR_FAILED;  // Didn't get a \n-terminated line.
+    for (int32_t i = 0; i < rv; i++) {
+      s->push_back(buffer[i]);  // Inefficiency FTW.

[yzshen1, 2011/11/15 00:47:07]

Maybe this is better: find out whether there is '\n' and where it is, and then
do an append().

[viettrungluu, 2011/11/15 01:08:26]

How about I just add a .reserve()?

[yzshen1, 2011/11/15 01:16:56]

Okay.

+      if (buffer[i] == '\n')
+        return PP_OK;
+    }
+  }
+  return PP_ERROR_FAILED;
+}
+
+int32_t TestTCPSocketPrivate::WriteStringToSocket(TCPSocketPrivate* socket,
+                                                  const std::string& s) {
+  const char* buffer = s.data();
+  size_t written = 0;
+  while (written < s.size()) {
+    TestCompletionCallback cb(instance_->pp_instance(), force_async_);
+    int32_t rv = socket->Write(buffer + written, s.size() - written, cb);
+    if (force_async_ && rv != PP_OK_COMPLETIONPENDING)
+      return PP_ERROR_FAILED;
+    if (rv == PP_OK_COMPLETIONPENDING)
+      rv = cb.WaitForResult();
+    if (rv < 0)
+      return rv;
+    if (rv == 0)
+      return PP_ERROR_FAILED;
+    written += rv;
+  }
+  if (written != s.size())
+    return PP_ERROR_FAILED;
+  return PP_OK;
+}