Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(641)

Side by Side Diff: net/tools/testserver/run_testserver.cc

Issue 11971025: [sync] Divorce python sync test server chromiumsync.py from testserver.py (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Address Philippe's comments. Created 7 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright 2013 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 <stdio.h> 5 #include <stdio.h>
6 6
7 #include "base/at_exit.h" 7 #include "base/at_exit.h"
8 #include "base/command_line.h" 8 #include "base/command_line.h"
9 #include "base/file_path.h" 9 #include "base/file_path.h"
10 #include "base/logging.h" 10 #include "base/logging.h"
11 #include "base/message_loop.h" 11 #include "base/message_loop.h"
12 #include "base/process_util.h" 12 #include "base/process_util.h"
13 #include "base/string_number_conversions.h" 13 #include "base/string_number_conversions.h"
14 #include "base/test/test_timeouts.h" 14 #include "base/test/test_timeouts.h"
15 #include "base/utf_string_conversions.h" 15 #include "base/utf_string_conversions.h"
16 #include "net/test/local_sync_test_server.h"
17 #include "net/test/python_utils.h" 16 #include "net/test/python_utils.h"
18 #include "net/test/test_server.h" 17 #include "net/test/test_server.h"
19 18
20 static void PrintUsage() { 19 static void PrintUsage() {
21 printf("run_testserver --doc-root=relpath\n" 20 printf("run_testserver --doc-root=relpath\n"
22 " [--http|--https|--ws|--wss|--ftp|--sync]\n" 21 " [--http|--https|--ws|--wss|--ftp]\n"
23 " [--ssl-cert=ok|mismatched-name|expired]\n" 22 " [--ssl-cert=ok|mismatched-name|expired]\n");
24 " [--port=<port>] [--xmpp-port=<xmpp_port>]\n");
25 printf("(NOTE: relpath should be relative to the 'src' directory.\n"); 23 printf("(NOTE: relpath should be relative to the 'src' directory.\n");
26 printf(" --port and --xmpp-port only work with the --sync flag.)\n");
27 }
28
29 // Launches the chromiumsync_test script, testing the --sync functionality.
30 static bool RunSyncTest() {
31 if (!net::TestServer::SetPythonPathStatic()) {
32 LOG(ERROR) << "Error trying to set python path. Exiting.";
33 return false;
34 }
35
36 FilePath sync_test_path;
37 if (!net::TestServer::GetTestServerDirectory(&sync_test_path)) {
38 LOG(ERROR) << "Error trying to get python test server path.";
39 return false;
40 }
41
42 sync_test_path =
43 sync_test_path.Append(FILE_PATH_LITERAL("chromiumsync_test.py"));
44
45 CommandLine python_command(CommandLine::NO_PROGRAM);
46 if (!GetPythonCommand(&python_command)) {
47 LOG(ERROR) << "Could not get python runtime command.";
48 return false;
49 }
50
51 python_command.AppendArgPath(sync_test_path);
52 if (!base::LaunchProcess(python_command, base::LaunchOptions(), NULL)) {
53 LOG(ERROR) << "Failed to launch test script.";
54 return false;
55 }
56 return true;
57 } 24 }
58 25
59 // Gets a port value from the switch with name |switch_name| and writes it to 26 // Gets a port value from the switch with name |switch_name| and writes it to
60 // |port|. Returns true if successful and false otherwise. 27 // |port|. Returns true if successful and false otherwise.
61 static bool GetPortFromSwitch(const std::string& switch_name, uint16* port) { 28 static bool GetPortFromSwitch(const std::string& switch_name, uint16* port) {
62 DCHECK(port != NULL) << "|port| is NULL"; 29 DCHECK(port != NULL) << "|port| is NULL";
63 CommandLine* command_line = CommandLine::ForCurrentProcess(); 30 CommandLine* command_line = CommandLine::ForCurrentProcess();
64 int port_int = 0; 31 int port_int = 0;
65 if (command_line->HasSwitch(switch_name)) { 32 if (command_line->HasSwitch(switch_name)) {
66 std::string port_str = command_line->GetSwitchValueASCII(switch_name); 33 std::string port_str = command_line->GetSwitchValueASCII(switch_name);
(...skipping 20 matching lines...) Expand all
87 logging::LOCK_LOG_FILE, 54 logging::LOCK_LOG_FILE,
88 logging::APPEND_TO_OLD_LOG_FILE, 55 logging::APPEND_TO_OLD_LOG_FILE,
89 logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS)) { 56 logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS)) {
90 printf("Error: could not initialize logging. Exiting.\n"); 57 printf("Error: could not initialize logging. Exiting.\n");
91 return -1; 58 return -1;
92 } 59 }
93 60
94 TestTimeouts::Initialize(); 61 TestTimeouts::Initialize();
95 62
96 if (command_line->GetSwitches().empty() || 63 if (command_line->GetSwitches().empty() ||
97 command_line->HasSwitch("help") || 64 command_line->HasSwitch("help")) {
98 ((command_line->HasSwitch("port") ||
99 command_line->HasSwitch("xmpp-port")) &&
100 !command_line->HasSwitch("sync"))) {
101 PrintUsage(); 65 PrintUsage();
102 return -1; 66 return -1;
103 } 67 }
104 68
105 net::TestServer::Type server_type; 69 net::TestServer::Type server_type;
106 if (command_line->HasSwitch("http")) { 70 if (command_line->HasSwitch("http")) {
107 server_type = net::TestServer::TYPE_HTTP; 71 server_type = net::TestServer::TYPE_HTTP;
108 } else if (command_line->HasSwitch("https")) { 72 } else if (command_line->HasSwitch("https")) {
109 server_type = net::TestServer::TYPE_HTTPS; 73 server_type = net::TestServer::TYPE_HTTPS;
110 } else if (command_line->HasSwitch("ws")) { 74 } else if (command_line->HasSwitch("ws")) {
111 server_type = net::TestServer::TYPE_WS; 75 server_type = net::TestServer::TYPE_WS;
112 } else if (command_line->HasSwitch("wss")) { 76 } else if (command_line->HasSwitch("wss")) {
113 server_type = net::TestServer::TYPE_WSS; 77 server_type = net::TestServer::TYPE_WSS;
114 } else if (command_line->HasSwitch("ftp")) { 78 } else if (command_line->HasSwitch("ftp")) {
115 server_type = net::TestServer::TYPE_FTP; 79 server_type = net::TestServer::TYPE_FTP;
116 } else if (command_line->HasSwitch("sync")) {
117 server_type = net::TestServer::TYPE_SYNC;
118 } else if (command_line->HasSwitch("sync-test")) {
119 return RunSyncTest() ? 0 : -1;
120 } else { 80 } else {
121 // If no scheme switch is specified, select http or https scheme. 81 // If no scheme switch is specified, select http or https scheme.
122 // TODO(toyoshim): Remove this estimation. 82 // TODO(toyoshim): Remove this estimation.
123 if (command_line->HasSwitch("ssl-cert")) 83 if (command_line->HasSwitch("ssl-cert"))
124 server_type = net::TestServer::TYPE_HTTPS; 84 server_type = net::TestServer::TYPE_HTTPS;
125 else 85 else
126 server_type = net::TestServer::TYPE_HTTP; 86 server_type = net::TestServer::TYPE_HTTP;
127 } 87 }
128 88
129 net::TestServer::SSLOptions ssl_options; 89 net::TestServer::SSLOptions ssl_options;
(...skipping 13 matching lines...) Expand all
143 ssl_options.server_certificate = 103 ssl_options.server_certificate =
144 net::TestServer::SSLOptions::CERT_EXPIRED; 104 net::TestServer::SSLOptions::CERT_EXPIRED;
145 } else { 105 } else {
146 printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str()); 106 printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str());
147 PrintUsage(); 107 PrintUsage();
148 return -1; 108 return -1;
149 } 109 }
150 } 110 }
151 111
152 FilePath doc_root = command_line->GetSwitchValuePath("doc-root"); 112 FilePath doc_root = command_line->GetSwitchValuePath("doc-root");
153 if ((server_type != net::TestServer::TYPE_SYNC) && doc_root.empty()) { 113 if (doc_root.empty()) {
154 printf("Error: --doc-root must be specified\n"); 114 printf("Error: --doc-root must be specified\n");
155 PrintUsage(); 115 PrintUsage();
156 return -1; 116 return -1;
157 } 117 }
158 118
159 scoped_ptr<net::TestServer> test_server; 119 scoped_ptr<net::TestServer> test_server;
160 if (net::TestServer::UsingSSL(server_type)) { 120 if (net::TestServer::UsingSSL(server_type)) {
161 test_server.reset(new net::TestServer(server_type, ssl_options, doc_root)); 121 test_server.reset(new net::TestServer(server_type, ssl_options, doc_root));
162 } else if (server_type == net::TestServer::TYPE_SYNC) {
163 uint16 port = 0;
164 uint16 xmpp_port = 0;
165 if (!GetPortFromSwitch("port", &port) ||
166 !GetPortFromSwitch("xmpp-port", &xmpp_port)) {
167 printf("Error: Could not extract --port and/or --xmpp-port.\n");
168 return -1;
169 }
170 test_server.reset(new net::LocalSyncTestServer(port, xmpp_port));
171 } else { 122 } else {
172 test_server.reset(new net::TestServer(server_type, 123 test_server.reset(new net::TestServer(server_type,
173 net::TestServer::kLocalhost, 124 net::TestServer::kLocalhost,
174 doc_root)); 125 doc_root));
175 } 126 }
176 127
177 if (!test_server->Start()) { 128 if (!test_server->Start()) {
178 printf("Error: failed to start test server. Exiting.\n"); 129 printf("Error: failed to start test server. Exiting.\n");
179 return -1; 130 return -1;
180 } 131 }
181 132
182 if (!file_util::DirectoryExists(test_server->document_root())) { 133 if (!file_util::DirectoryExists(test_server->document_root())) {
183 printf("Error: invalid doc root: \"%s\" does not exist!\n", 134 printf("Error: invalid doc root: \"%s\" does not exist!\n",
184 UTF16ToUTF8(test_server->document_root().LossyDisplayName()).c_str()); 135 UTF16ToUTF8(test_server->document_root().LossyDisplayName()).c_str());
185 return -1; 136 return -1;
186 } 137 }
187 138
188 printf("testserver running at %s (type ctrl+c to exit)\n", 139 printf("testserver running at %s (type ctrl+c to exit)\n",
189 test_server->host_port_pair().ToString().c_str()); 140 test_server->host_port_pair().ToString().c_str());
190 141
191 message_loop.Run(); 142 message_loop.Run();
192 return 0; 143 return 0;
193 } 144 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698