##// END OF EJS Templates
tests: use argparse in get-with-headers.py...
Gregory Szorc -
r35798:32317f8b default
parent child Browse files
Show More
@@ -5,6 +5,7 b' a subset of the headers plus the body of'
5 5
6 6 from __future__ import absolute_import, print_function
7 7
8 import argparse
8 9 import json
9 10 import os
10 11 import sys
@@ -22,25 +23,21 b' try:'
22 23 except ImportError:
23 24 pass
24 25
25 twice = False
26 if '--twice' in sys.argv:
27 sys.argv.remove('--twice')
28 twice = True
29 headeronly = False
30 if '--headeronly' in sys.argv:
31 sys.argv.remove('--headeronly')
32 headeronly = True
33 formatjson = False
34 if '--json' in sys.argv:
35 sys.argv.remove('--json')
36 formatjson = True
26 parser = argparse.ArgumentParser()
27 parser.add_argument('--twice', action='store_true')
28 parser.add_argument('--headeronly', action='store_true')
29 parser.add_argument('--json', action='store_true')
30 parser.add_argument('--hgproto')
31 parser.add_argument('host')
32 parser.add_argument('path')
33 parser.add_argument('show', nargs='*')
37 34
38 hgproto = None
39 if '--hgproto' in sys.argv:
40 idx = sys.argv.index('--hgproto')
41 hgproto = sys.argv[idx + 1]
42 sys.argv.pop(idx)
43 sys.argv.pop(idx)
35 args = parser.parse_args()
36
37 twice = args.twice
38 headeronly = args.headeronly
39 formatjson = args.json
40 hgproto = args.hgproto
44 41
45 42 tag = None
46 43 def request(host, path, show):
@@ -83,9 +80,9 b' def request(host, path, show):'
83 80
84 81 return response.status
85 82
86 status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
83 status = request(args.host, args.path, args.show)
87 84 if twice:
88 status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
85 status = request(args.host, args.path, args.show)
89 86
90 87 if 200 <= status <= 305:
91 88 sys.exit(0)
General Comments 0
You need to be logged in to leave comments. Login now