From: Brian Warner <warner@lothar.com>
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/about.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):