profiler: flush after writing the profiler output
Otherwise, the profiler output might only partially appears until the next flush
of the buffer. Since profiling often happens for long operation, the next flush
can be a long time away.
import os
from mercurial.hgweb import hgwebdir_mod
hgwebdir = hgwebdir_mod.hgwebdir
os.mkdir(b'webdir')
os.chdir(b'webdir')
webdir = os.path.realpath(b'.')
def trivial_response(req, res):
return []
def make_hgwebdir(gc_rate=None):
config = os.path.join(webdir, b'hgwebdir.conf')
with open(config, 'wb') as configfile:
configfile.write(b'[experimental]\n')
if gc_rate is not None:
configfile.write(b'web.full-garbage-collection-rate=%d\n' % gc_rate)
hg_wd = hgwebdir(config)
hg_wd._runwsgi = trivial_response
return hg_wd
def process_requests(webdir_instance, number):
# we don't care for now about passing realistic arguments
for _ in range(number):
for chunk in webdir_instance.run_wsgi(None, None):
pass
without_gc = make_hgwebdir(gc_rate=0)
process_requests(without_gc, 5)
assert without_gc.requests_count == 5
assert without_gc.gc_full_collections_done == 0
with_gc = make_hgwebdir(gc_rate=2)
process_requests(with_gc, 5)
assert with_gc.requests_count == 5
assert with_gc.gc_full_collections_done == 2
with_systematic_gc = make_hgwebdir() # default value of the setting
process_requests(with_systematic_gc, 3)
assert with_systematic_gc.requests_count == 3
assert with_systematic_gc.gc_full_collections_done == 3