OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/safe_browsing/local_safebrowsing_test_server.h" |
| 6 |
| 7 #include "base/command_line.h" |
| 8 #include "base/path_service.h" |
| 9 #include "base/string_number_conversions.h" |
| 10 #include "base/values.h" |
| 11 #include "net/test/python_utils.h" |
| 12 #include "net/test/test_server.h" |
| 13 |
| 14 LocalSafeBrowsingTestServer::LocalSafeBrowsingTestServer( |
| 15 const FilePath& data_file) |
| 16 : net::LocalTestServer(net::TestServer::TYPE_HTTP, |
| 17 net::TestServer::kLocalhost, |
| 18 FilePath()), |
| 19 data_file_(data_file) { |
| 20 } |
| 21 |
| 22 LocalSafeBrowsingTestServer::~LocalSafeBrowsingTestServer() {} |
| 23 |
| 24 bool LocalSafeBrowsingTestServer::GetTestServerPath( |
| 25 FilePath* testserver_path) const { |
| 26 FilePath testserver_dir; |
| 27 if (!PathService::Get(base::DIR_SOURCE_ROOT, &testserver_dir)) { |
| 28 LOG(ERROR) << "Failed to get DIR_SOURCE_ROOT"; |
| 29 return false; |
| 30 } |
| 31 |
| 32 testserver_dir = testserver_dir |
| 33 .Append(FILE_PATH_LITERAL("chrome")) |
| 34 .Append(FILE_PATH_LITERAL("browser")) |
| 35 .Append(FILE_PATH_LITERAL("safe_browsing")); |
| 36 |
| 37 *testserver_path = testserver_dir.Append(FILE_PATH_LITERAL( |
| 38 "safe_browsing_testserver.py")); |
| 39 return true; |
| 40 } |
| 41 |
| 42 bool LocalSafeBrowsingTestServer::SetPythonPath() const { |
| 43 if (!net::LocalTestServer::SetPythonPath()) |
| 44 return false; |
| 45 |
| 46 // Locate the Python code generated by the protocol buffers compiler. |
| 47 FilePath pyproto_dir; |
| 48 if (!GetPyProtoPath(&pyproto_dir)) { |
| 49 LOG(ERROR) << "Cannot find pyproto dir for generated code."; |
| 50 return false; |
| 51 } |
| 52 |
| 53 AppendToPythonPath(pyproto_dir.AppendASCII("google")); |
| 54 return true; |
| 55 } |
| 56 |
| 57 bool LocalSafeBrowsingTestServer::GenerateAdditionalArguments( |
| 58 base::DictionaryValue* arguments) const { |
| 59 arguments->SetString("data-file", data_file_.value()); |
| 60 return true; |
| 61 } |
OLD | NEW |