chrome/browser/safe_browsing/local_safebrowsing_test_server.cc

   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/strings/string_number_conversions.h"
  10 #include "base/values.h"
  11 #include "net/test/python_utils.h"
  12 #include "net/test/spawned_test_server/spawned_test_server.h"
  13
  14 LocalSafeBrowsingTestServer::LocalSafeBrowsingTestServer(
  15     const base::FilePath& data_file)
  16     : net::LocalTestServer(net::SpawnedTestServer::TYPE_HTTP,
  17                            net::SpawnedTestServer::kLocalhost,
  18                            base::FilePath()),
  19       data_file_(data_file) {
  20 }
  21
  22 LocalSafeBrowsingTestServer::~LocalSafeBrowsingTestServer() {}
  23
  24 bool LocalSafeBrowsingTestServer::GetTestServerPath(
  25     base::FilePath* testserver_path) const {
  26   base::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   base::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 }