From: Brian Warner Date: Tue, 24 Feb 2009 00:39:37 +0000 (-0700) Subject: test/no_network.py: add a basic stats provider X-Git-Url: https://git.rkrishnan.org/components/com_hotproperty/flags/module-simplejson.tests.html?a=commitdiff_plain;h=cc3c1ae8df5669c51c5324eded055dfe308c905e;p=tahoe-lafs%2Ftahoe-lafs.git test/no_network.py: add a basic stats provider --- diff --git a/src/allmydata/test/no_network.py b/src/allmydata/test/no_network.py index 731d7d99..d75cc04a 100644 --- a/src/allmydata/test/no_network.py +++ b/src/allmydata/test/no_network.py @@ -135,6 +135,24 @@ class NoNetworkClient(Client): def get_nickname_for_peerid(self, peerid): return None +class SimpleStats: + def __init__(self): + self.counters = {} + self.stats_producers = [] + + def count(self, name, delta=1): + val = self.counters.setdefault(name, 0) + self.counters[name] = val + delta + + def register_producer(self, stats_producer): + self.stats_producers.append(stats_producer) + + def get_stats(self): + stats = {} + for sp in self.stats_producers: + stats.update(sp.get_stats()) + ret = { 'counters': self.counters, 'stats': stats } + return ret class NoNetworkGrid(service.MultiService): def __init__(self, basedir, num_clients=1, num_servers=10, @@ -181,7 +199,7 @@ class NoNetworkGrid(service.MultiService): serverdir = os.path.join(self.basedir, "servers", idlib.shortnodeid_b2a(serverid)) fileutil.make_dirs(serverdir) - ss = StorageServer(serverdir, serverid) + ss = StorageServer(serverdir, serverid, stats_provider=SimpleStats()) return ss def add_server(self, i, ss):