src/chrome/browser/safe_browsing/safe_browsing_testserver.py

   1 #!/usr/bin/env python
   2 # Copyright 2013 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Wraps the upstream safebrowsing_test_server.py to run in Chrome tests."""
   7
   8 import os
   9 import sys
  10
  11 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
  12
  13 sys.path.append(os.path.join(BASE_DIR, '..', '..', '..', 'net',
  14                              'tools', 'testserver'))
  15 import testserver_base
  16
  17
  18 class ServerRunner(testserver_base.TestServerRunner):
  19   """TestServerRunner for safebrowsing_test_server.py."""
  20
  21   def create_server(self, server_data):
  22     sys.path.append(os.path.join(BASE_DIR, '..', '..', '..', 'third_party',
  23                                  'safe_browsing', 'testing'))
  24     import safebrowsing_test_server
  25     server = safebrowsing_test_server.SetupServer(
  26         self.options.data_file, self.options.host, self.options.port,
  27         opt_enforce_caching=False, opt_validate_database=True)
  28     print 'Safebrowsing HTTP server started on port %d...' % server.server_port
  29     server_data['port'] = server.server_port
  30
  31     return server
  32
  33   def add_options(self):
  34     testserver_base.TestServerRunner.add_options(self)
  35     self.option_parser.add_option('--data-file', dest='data_file',
  36                                   help='File containing safebrowsing test '
  37                                   'data and expectations')
  38
  39
  40 if __name__ == '__main__':
  41   sys.exit(ServerRunner().main())