#!/usr/bin/env python
# Copyright 2013 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.

"""Wraps the upstream safebrowsing_test_server.py to run in Chrome tests."""

import os
import sys

BASE_DIR = os.path.dirname(os.path.abspath(__file__))

sys.path.append(os.path.join(BASE_DIR, '..', '..', '..', 'net',
                             'tools', 'testserver'))
import testserver_base


class ServerRunner(testserver_base.TestServerRunner):
  """TestServerRunner for safebrowsing_test_server.py."""

  def create_server(self, server_data):
    sys.path.append(os.path.join(BASE_DIR, '..', '..', '..', 'third_party',
                                 'safe_browsing', 'testing'))
    import safebrowsing_test_server
    server = safebrowsing_test_server.SetupServer(
        self.options.data_file, self.options.host, self.options.port,
        opt_enforce_caching=False, opt_validate_database=True)
    print 'Safebrowsing HTTP server started on port %d...' % server.server_port
    server_data['port'] = server.server_port

    return server

  def add_options(self):
    testserver_base.TestServerRunner.add_options(self)
    self.option_parser.add_option('--data-file', dest='data_file',
                                  help='File containing safebrowsing test '
                                  'data and expectations')


if __name__ == '__main__':
  sys.exit(ServerRunner().main())