| # Copyright 2015 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. |
| |
| import argparse |
| import httplib |
| import sys |
| |
| |
| def GetArgs(): |
| """Returns the specified command line args.""" |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--server', required=True) |
| parser.add_argument('--port', required=True, type=int) |
| return parser.parse_args() |
| |
| |
| def main(): |
| """Get the webpage and assert the text == 'SUCCESS!'.""" |
| args = GetArgs() |
| conn = httplib.HTTPConnection(args.server, args.port) |
| conn.request('GET', '/') |
| response = conn.getresponse().read() |
| assert response == 'SUCCESS!', '%s != SUCCESS!' % response |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |
