|
|
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
|
|
|
|
