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 |