Commit ca432935 by Victor Shnayder

Fix cookie passing.

parent 24e4feca
...@@ -96,12 +96,18 @@ def loadtest(args): ...@@ -96,12 +96,18 @@ def loadtest(args):
server = args.server server = args.server
name = args.name name = args.name
ab_options = args.abopt if args.abopt else [] ab_options = args.abopt if args.abopt else []
ab_options += ["-Aanant:agarwal"] ab_options.append("-Aanant:agarwal")
cookies = []
if args.sessionid: if args.sessionid:
ab_options += ["-C 'sessionid={0}'".format(args.sessionid)] cookies.append("sessionid={0}".format(args.sessionid))
if args.csrftoken: if args.csrftoken:
ab_options += ["-C 'csrftoken={0}'".format(args.csrftoken)] cookies.append("csrftoken={0}".format(args.csrftoken))
if len(cookies) > 0:
cookie_str = "; ".join(cookies)
ab_options.append("-C '{0}'".format(cookie_str))
pages = args.pages if args.pages else [] pages = args.pages if args.pages else []
if args.pagelist: if args.pagelist:
...@@ -114,8 +120,6 @@ def loadtest(args): ...@@ -114,8 +120,6 @@ def loadtest(args):
# want a string # want a string
ab_options = ' '.join(str(s) for s in ab_options) if ab_options is not None else "" ab_options = ' '.join(str(s) for s in ab_options) if ab_options is not None else ""
reqs_per_thread = 2
# If there are already results for this run, just delete them. (TODO: Desired behavior?) # If there are already results for this run, just delete them. (TODO: Desired behavior?)
os.system("rm -rf {name}".format(name=name)) os.system("rm -rf {name}".format(name=name))
for page in pages: for page in pages:
...@@ -123,8 +127,8 @@ def loadtest(args): ...@@ -123,8 +127,8 @@ def loadtest(args):
outdir = "{name}/page_{page}".format(name=name, page=page.replace('/', '-')) outdir = "{name}/page_{page}".format(name=name, page=page.replace('/', '-'))
print "Testing {0}. Output in {1}".format(url, outdir) print "Testing {0}. Output in {1}".format(url, outdir)
os.makedirs(outdir) os.makedirs(outdir)
for conc in [1]: #[1, 2, 3, 10]: for conc in args.concurrency:
requests = conc * reqs_per_thread requests = conc * args.reqs_per_thread
logpath = outdir + '/{requests}-{conc}.results'.format(requests=requests, conc=conc) logpath = outdir + '/{requests}-{conc}.results'.format(requests=requests, conc=conc)
test_url(url, requests, conc, ab_options, logpath) test_url(url, requests, conc, ab_options, logpath)
...@@ -145,6 +149,9 @@ def main(): ...@@ -145,6 +149,9 @@ def main():
help='if testing non-anonymously, specify session id') help='if testing non-anonymously, specify session id')
parser.add_argument('--csrftoken', parser.add_argument('--csrftoken',
help='if posting forms, specify csrftoken') help='if posting forms, specify csrftoken')
parser.add_argument('-c', dest='concurrency', action='append', type=int, default=[1],
help="try this number of simultaneous threads. May be specified more than once.")
parser.add_argument('-r', type=int, default=2, help="Number of requests per thread", dest='reqs_per_thread')
parser.add_argument('--name', parser.add_argument('--name',
help='test name--results will be in {name}/', help='test name--results will be in {name}/',
......
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
/courses/ /courses/
/wiki/ /wiki/
/courses/BerkeleyX/CS188/fa12/wiki /courses/BerkeleyX/CS188/fa12/wiki
/courses/BerkeleyX/CS188/fa12/wiki/6.002x/
/courses/BerkeleyX/CS188/fa12/discussion/forum /courses/BerkeleyX/CS188/fa12/discussion/forum
/courses/BerkeleyX/CS188/fa12/progress /courses/BerkeleyX/CS188/fa12/progress
/courses/BerkeleyX/CS188/fa12/info /courses/BerkeleyX/CS188/fa12/info
...@@ -12,3 +13,4 @@ ...@@ -12,3 +13,4 @@
# Berkeley-specific. Perhaps generate these automatically. # Berkeley-specific. Perhaps generate these automatically.
/courses/BerkeleyX/CS188/fa12/courseware/Week_1/Project_0_Tutorial/ /courses/BerkeleyX/CS188/fa12/courseware/Week_1/Project_0_Tutorial/
/courses/BerkeleyX/CS188/fa12/wiki/6.002x/_create/?slug=DampedSecondOrderSystems
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment