Added pepper test for SSLHandshake

chrome/test/ui/ppapi_uitest.cc

 // Copyright (c) 2012 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 "chrome/test/ui/ppapi_uitest.h"
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/stringprintf.h"
@@ skipping 141 matching lines @@
 }
 
 void PPAPITestBase::RunTestAndReload(const std::string& test_case) {
   GURL url = GetTestFileUrl(test_case);
   RunTestURL(url);
   // If that passed, we simply run the test again, which navigates again.
   RunTestURL(url);
 }
 
 void PPAPITestBase::RunTestViaHTTP(const std::string& test_case) {
-  // For HTTP tests, we use the output DIR to grab the generated files such
-  // as the NEXEs.
-  FilePath exe_dir = CommandLine::ForCurrentProcess()->GetProgram().DirName();
-  FilePath src_dir;
-  ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &src_dir));
-
-  // TestServer expects a path relative to source. So we must first
-  // generate absolute paths to SRC and EXE and from there generate
-  // a relative path.
-  if (!exe_dir.IsAbsolute()) file_util::AbsolutePath(&exe_dir);
-  if (!src_dir.IsAbsolute()) file_util::AbsolutePath(&src_dir);
-  ASSERT_TRUE(exe_dir.IsAbsolute());
-  ASSERT_TRUE(src_dir.IsAbsolute());
-
-  size_t match, exe_size, src_size;
-  std::vector<FilePath::StringType> src_parts, exe_parts;
-
-  // Determine point at which src and exe diverge, and create a relative path.
-  exe_dir.GetComponents(&exe_parts);
-  src_dir.GetComponents(&src_parts);
-  exe_size = exe_parts.size();
-  src_size = src_parts.size();
-  for (match = 0; match < exe_size && match < src_size; ++match) {
-    if (exe_parts[match] != src_parts[match])
-      break;
-  }
   FilePath web_dir;
-  for (size_t tmp_itr = match; tmp_itr < src_size; ++tmp_itr) {
-    web_dir = web_dir.Append(FILE_PATH_LITERAL(".."));
-  }
-  for (; match < exe_size; ++match) {
-    web_dir = web_dir.Append(exe_parts[match]);
-  }
-
+  ASSERT_TRUE(GetHTTPDocumentRoot(&web_dir));
   net::TestServer test_server(net::TestServer::TYPE_HTTP,
                               net::TestServer::kLocalhost,
                               web_dir);
   ASSERT_TRUE(test_server.Start());
   std::string query = BuildQuery("files/test_case.html?", test_case);
 
   GURL url = test_server.GetURL(query);
   RunTestURL(url);
 }
 
+void PPAPITestBase::RunTestWithSSLServer(const std::string& test_case) {
+  FilePath web_dir;
+  ASSERT_TRUE(GetHTTPDocumentRoot(&web_dir));
+  net::TestServer test_server(net::BaseTestServer::HTTPSOptions(), web_dir);
+  ASSERT_TRUE(test_server.Start());
+
+  uint16_t port = test_server.host_port_pair().port();
+  std::string url = StringPrintf("%s&ssl_server_port=%d",
+                                 test_case.c_str(), port);
+  RunTestViaHTTP(url);

[yzshen1, 2012/03/21 22:37:58]

nit:
It might be better to have a private method named
RunHTTPTestServer(web_dir, test_case, extra_params)
and then let RunTestViaHTTP() and RunTestWithSSLServer() call it.

So that we can save a call to GetHTTPDocumentRoot(), and it is more clear about
the extra parameter.

[raymes, 2012/03/21 23:33:07]

Hmm, I'm not sure exactly what you mean here. Can you give a more concrete
example of what you are suggesting?

[yzshen1, 2012/03/21 23:42:12]

The current code calls RunTestViaHTTP() directly, as a result,
- GetHTTPDocumentRoot is called twice (one in RunTestWithSSLServer and the other
in RunTestViaHTTP)
- the way to add ssl_server_port is not very elegant. The code actually pretends
that it is part of the test_case name.

So, maybe the following is better:
void RunTestViaHTTP(test_case) {
  ...
  GetHTTPDocumentRoot(&web_dir);
  ...
  RunHTTPTestServer(web_dir, test_case, "");
}

void RunTestWithSSLServer(test_case) {
  ...
  GetHTTPDocumentRoot(&web_dir);
  // Start the ssl server.
  ...
  RunHTTPTestServer(web_dir, test_case, "ssl_server_port=1234");
}

Does that make sense?
On 2012/03/21 23:33:07, raymes wrote:

+}
+
 void PPAPITestBase::RunTestWithWebSocketServer(const std::string& test_case) {
   FilePath websocket_root_dir;
   ASSERT_TRUE(
       PathService::Get(chrome::DIR_LAYOUT_TESTS, &websocket_root_dir));
 
   ui_test_utils::TestWebSocketServer server;
   int port = server.UseRandomPort();
   ASSERT_TRUE(server.Start(websocket_root_dir));
   std::string url = StringPrintf("%s&websocket_port=%d",
                                  test_case.c_str(), port);
@@ skipping 18 matching lines @@
   TestFinishObserver observer(
       browser()->GetSelectedWebContents()->GetRenderViewHost(), kTimeoutMs);
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
   ASSERT_TRUE(observer.WaitForFinish()) << "Test timed out.";
 
   EXPECT_STREQ("PASS", observer.result().c_str());
 }
 
+bool PPAPITestBase::GetHTTPDocumentRoot(FilePath* web_dir) {
+  // For HTTP tests, we use the output DIR to grab the generated files such
+  // as the NEXEs.
+  FilePath exe_dir = CommandLine::ForCurrentProcess()->GetProgram().DirName();
+  FilePath src_dir;
+  if (!PathService::Get(base::DIR_SOURCE_ROOT, &src_dir))
+    return false;
+
+  // TestServer expects a path relative to source. So we must first
+  // generate absolute paths to SRC and EXE and from there generate
+  // a relative path.
+  if (!exe_dir.IsAbsolute()) file_util::AbsolutePath(&exe_dir);
+  if (!src_dir.IsAbsolute()) file_util::AbsolutePath(&src_dir);
+  if (!exe_dir.IsAbsolute())
+    return false;
+  if (!src_dir.IsAbsolute())
+    return false;
+
+  size_t match, exe_size, src_size;
+  std::vector<FilePath::StringType> src_parts, exe_parts;
+
+  // Determine point at which src and exe diverge, and create a relative path.
+  exe_dir.GetComponents(&exe_parts);
+  src_dir.GetComponents(&src_parts);
+  exe_size = exe_parts.size();
+  src_size = src_parts.size();
+  for (match = 0; match < exe_size && match < src_size; ++match) {
+    if (exe_parts[match] != src_parts[match])
+      break;
+  }
+  for (size_t tmp_itr = match; tmp_itr < src_size; ++tmp_itr) {
+    *web_dir = web_dir->Append(FILE_PATH_LITERAL(".."));
+  }
+  for (; match < exe_size; ++match) {
+    *web_dir = web_dir->Append(exe_parts[match]);
+  }
+  return true;
+}
+
 PPAPITest::PPAPITest() {
 }
 
 void PPAPITest::SetUpCommandLine(CommandLine* command_line) {
   PPAPITestBase::SetUpCommandLine(command_line);
 
   // Append the switch to register the pepper plugin.
   // library name = <out dir>/<test_name>.<library_extension>
   // MIME type = application/x-ppapi-<test_name>
   FilePath plugin_dir;
@@ skipping 85 matching lines @@
 // Similar macros that test over HTTP.
 #define TEST_PPAPI_IN_PROCESS_VIA_HTTP(test_name) \
     IN_PROC_BROWSER_TEST_F(PPAPITest, test_name) { \
       RunTestViaHTTP(STRIP_PREFIXES(test_name)); \
     }
 #define TEST_PPAPI_OUT_OF_PROCESS_VIA_HTTP(test_name) \
     IN_PROC_BROWSER_TEST_F(OutOfProcessPPAPITest, test_name) { \
       RunTestViaHTTP(STRIP_PREFIXES(test_name)); \
     }
 
+// Similar macros that test with an SSL server.
+#define TEST_PPAPI_IN_PROCESS_WITH_SSL_SERVER(test_name) \
+    IN_PROC_BROWSER_TEST_F(PPAPITest, test_name) { \
+      RunTestWithSSLServer(STRIP_PREFIXES(test_name)); \
+    }
+#define TEST_PPAPI_OUT_OF_PROCESS_WITH_SSL_SERVER(test_name) \
+    IN_PROC_BROWSER_TEST_F(OutOfProcessPPAPITest, test_name) { \
+      RunTestWithSSLServer(STRIP_PREFIXES(test_name)); \
+    }
+
 // Similar macros that test with WebSocket server
 #define TEST_PPAPI_IN_PROCESS_WITH_WS(test_name) \
     IN_PROC_BROWSER_TEST_F(PPAPITest, test_name) { \
       RunTestWithWebSocketServer(STRIP_PREFIXES(test_name)); \
     }
 #define TEST_PPAPI_OUT_OF_PROCESS_WITH_WS(test_name) \
     IN_PROC_BROWSER_TEST_F(OutOfProcessPPAPITest, test_name) { \
       RunTestWithWebSocketServer(STRIP_PREFIXES(test_name)); \
     }
 
@@ skipping 437 matching lines @@
 TEST_PPAPI_NACL_VIA_HTTP(MAYBE_NetAddressPrivateUntrusted_ReplacePort)
 TEST_PPAPI_NACL_VIA_HTTP(NetAddressPrivateUntrusted_GetAnyAddress)
 TEST_PPAPI_NACL_VIA_HTTP(NetAddressPrivateUntrusted_GetFamily)
 TEST_PPAPI_NACL_VIA_HTTP(MAYBE_NetAddressPrivateUntrusted_GetPort)
 TEST_PPAPI_NACL_VIA_HTTP(NetAddressPrivateUntrusted_GetAddress)
 
 TEST_PPAPI_IN_PROCESS(NetworkMonitorPrivate_Basic)
 TEST_PPAPI_IN_PROCESS(NetworkMonitorPrivate_2Monitors)
 
 // PPB_TCPSocket_Private currently isn't supported in-process.
-IN_PROC_BROWSER_TEST_F(OutOfProcessPPAPITest, TCPSocketPrivate) {
-  RunTestViaHTTP("TCPSocketPrivate");
-}
+TEST_PPAPI_OUT_OF_PROCESS_WITH_SSL_SERVER(TCPSocketPrivate);

[yzshen1, 2012/03/21 22:37:58]

Please update this accordingly once you check in the other CL of removing
TCPSocketPrivateShared.

[raymes, 2012/03/21 23:33:07]

Done.

 
 TEST_PPAPI_IN_PROCESS(Flash_SetInstanceAlwaysOnTop)
 TEST_PPAPI_IN_PROCESS(Flash_GetProxyForURL)
 TEST_PPAPI_IN_PROCESS(Flash_MessageLoop)
 TEST_PPAPI_IN_PROCESS(Flash_GetLocalTimeZoneOffset)
 TEST_PPAPI_IN_PROCESS(Flash_GetCommandLineArgs)
 TEST_PPAPI_OUT_OF_PROCESS(Flash_SetInstanceAlwaysOnTop)
 TEST_PPAPI_OUT_OF_PROCESS(Flash_GetProxyForURL)
 TEST_PPAPI_OUT_OF_PROCESS(Flash_MessageLoop)
 TEST_PPAPI_OUT_OF_PROCESS(Flash_GetLocalTimeZoneOffset)
@@ skipping 138 matching lines @@
 TEST_PPAPI_OUT_OF_PROCESS(ResourceArray_OutOfRangeAccess)
 TEST_PPAPI_OUT_OF_PROCESS(ResourceArray_EmptyArray)
 TEST_PPAPI_OUT_OF_PROCESS(ResourceArray_InvalidElement)
 
 TEST_PPAPI_IN_PROCESS(FlashMessageLoop_Basics)
 TEST_PPAPI_IN_PROCESS(FlashMessageLoop_RunWithoutQuit)
 TEST_PPAPI_OUT_OF_PROCESS(FlashMessageLoop_Basics)
 TEST_PPAPI_OUT_OF_PROCESS(FlashMessageLoop_RunWithoutQuit)
 
 #endif // ADDRESS_SANITIZER