def test_resolve_no_scheme_ssl(self):
test_url = 'https://localhost'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'https://localhost:443')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, 'localhost')
def test_ssl_resolve(self):
test_url = 'https://localhost:9999'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'https://localhost:9999')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, 'localhost')
def test_resolve_no_scheme(self):
test_url = 'http://localhost'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'http://127.0.0.1:80')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, '127.0.0.1')
def test_resolve_no_scheme(self):
test_url = 'http://localhost'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'http://127.0.0.1:80')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, '127.0.0.1')
def test_resolve_no_scheme_ssl(self):
test_url = 'https://localhost'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'https://localhost:443')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, 'localhost')
def test_ssl_resolve(self):
test_url = 'https://localhost:9999'
url, original, resolved = resolve(test_url)
self.assertEqual(url, 'https://localhost:9999')
self.assertEqual(original, 'localhost')
self.assertEqual(resolved, 'localhost')
def test_resolve_no_scheme_ssl(self):
test_url = "https://localhost"
url, original, resolved = resolve(test_url)
self.assertEqual(url, "https://localhost:443")
self.assertEqual(original, "localhost")
self.assertEqual(resolved, "localhost")
def test_resolve_no_scheme(self):
test_url = "http://localhost"
url, original, resolved = resolve(test_url)
self.assertEqual(url, "http://127.0.0.1:80")
self.assertEqual(original, "localhost")
self.assertEqual(resolved, "127.0.0.1")
def test_ssl_resolve(self):
test_url = "https://localhost:9999"
url, original, resolved = resolve(test_url)
self.assertEqual(url, "https://localhost:9999")
self.assertEqual(original, "localhost")
self.assertEqual(resolved, "localhost")
def main():
parser = argparse.ArgumentParser(
description='Simple HTTP Load runner.')
parser.add_argument(
'--version', action='store_true', default=False,
help='Displays version and exits.')
parser.add_argument('-m', '--method', help='HTTP Method',
type=str, default='GET', choices=_VERBS)
parser.add_argument('--content-type', help='Content-Type',
type=str, default='text/plain')
parser.add_argument('-D', '--data',
help=('Data. Prefixed by "py:" to point '
'a python callable.'),
type=str)
parser.add_argument('-c', '--concurrency', help='Concurrency',
type=int, default=1)
parser.add_argument('-a', '--auth',
help='Basic authentication user:password', type=str)
parser.add_argument('--header', help='Custom header. name:value',
type=str, action='append')
parser.add_argument('--pre-hook',
help=("Python module path (eg: mymodule.pre_hook) "
"to a callable which will be executed before "
"doing a request for example: "
"pre_hook(method, url, options). "
"It must return a tupple of parameters given in "
"function definition"),
type=str)
parser.add_argument('--post-hook',
help=("Python module path (eg: mymodule.post_hook) "
"to a callable which will be executed after "
"a request is done for example: "
"eg. post_hook(response). "
"It must return a given response parameter or "
"raise an `boom._boom.RequestException` for "
"failed request."),
type=str)
parser.add_argument('--json-output',
help='Prints the results in JSON instead of the '
'default format',
action='store_true')
parser.add_argument('-q', '--quiet', help="Don't display progress bar",
action='store_true')
group = parser.add_mutually_exclusive_group()
group.add_argument('-n', '--requests', help='Number of requests',
type=int)
group.add_argument('-d', '--duration', help='Duration in seconds',
type=int)
parser.add_argument('url', help='URL to hit', nargs='?')
args = parser.parse_args()
if args.version:
print(__version__)
sys.exit(0)
if args.url is None:
print('You need to provide an URL.')
parser.print_usage()
sys.exit(0)
if args.data is not None and not args.method in _DATA_VERBS:
print("You can't provide data with %r" % args.method)
parser.print_usage()
sys.exit(0)
if args.requests is None and args.duration is None:
args.requests = 1
try:
url, original, resolved = resolve(args.url)
except gaierror as e:
print_errors(("DNS resolution failed for %s (%s)" %
(args.url, str(e)),))
sys.exit(1)
def _split(header):
header = header.split(':')
if len(header) != 2:
print("A header must be of the form name:value")
parser.print_usage()
sys.exit(0)
return header
if args.header is None:
headers = {}
else:
headers = dict([_split(header) for header in args.header])
if original != resolved and 'Host' not in headers:
headers['Host'] = original
try:
res = load(
url, args.requests, args.concurrency, args.duration,
args.method, args.data, args.content_type, args.auth,
headers=headers, pre_hook=args.pre_hook,
post_hook=args.post_hook, quiet=(args.json_output or args.quiet))
except RequestException as e:
print_errors((e, ))
sys.exit(1)
if not args.json_output:
print_errors(res.errors)
print_stats(res)
else:
print_json(res)
logger.info('Bye!')
