| OLD | NEW |
| (Empty) |
| 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, '..', '..', '..', '..', | |
| 23 'third_party', '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()) | |
| OLD | NEW |