3 from twisted.trial import unittest
4 from twisted.application import service
5 from twisted.python import log
8 from allmydata import client
9 from allmydata.storage_client import StorageFarmBroker
10 from allmydata.introducer.client import IntroducerClient
11 from allmydata.util import base32
12 from foolscap.api import flushEventualQueue
13 import common_util as testutil
15 class FakeIntroducerClient(IntroducerClient):
17 self._connections = set()
18 def add_peer(self, nodeid):
19 entry = (nodeid, "storage", "rref")
20 self._connections.add(entry)
21 def remove_all_peers(self):
22 self._connections.clear()
24 class Basic(unittest.TestCase):
25 def test_loadable(self):
26 basedir = "test_client.Basic.test_loadable"
28 open(os.path.join(basedir, "introducer.furl"), "w").write("")
29 open(os.path.join(basedir, "vdrive.furl"), "w").write("")
30 c = client.Client(basedir)
32 def test_loadable_without_vdrive(self):
33 basedir = "test_client.Basic.test_loadable_without_vdrive"
35 open(os.path.join(basedir, "introducer.furl"), "w").write("")
36 c = client.Client(basedir)
38 def test_loadable_old_config_bits(self):
39 basedir = "test_client.Basic.test_loadable_old_config_bits"
41 open(os.path.join(basedir, "introducer.furl"), "w").write("")
42 open(os.path.join(basedir, "vdrive.furl"), "w").write("")
43 open(os.path.join(basedir, "no_storage"), "w").write("")
44 open(os.path.join(basedir, "readonly_storage"), "w").write("")
45 open(os.path.join(basedir, "debug_discard_storage"), "w").write("")
46 c = client.Client(basedir)
48 c.getServiceNamed("storage")
49 self.fail("that was supposed to fail")
53 def test_loadable_old_storage_config_bits(self):
54 basedir = "test_client.Basic.test_loadable_old_storage_config_bits"
56 open(os.path.join(basedir, "introducer.furl"), "w").write("")
57 open(os.path.join(basedir, "vdrive.furl"), "w").write("")
58 open(os.path.join(basedir, "readonly_storage"), "w").write("")
59 open(os.path.join(basedir, "debug_discard_storage"), "w").write("")
60 c = client.Client(basedir)
61 s = c.getServiceNamed("storage")
62 self.failUnless(s.no_storage)
63 self.failUnless(s.readonly_storage)
65 def test_secrets(self):
66 basedir = "test_client.Basic.test_secrets"
68 open(os.path.join(basedir, "introducer.furl"), "w").write("")
69 open(os.path.join(basedir, "vdrive.furl"), "w").write("")
70 c = client.Client(basedir)
71 secret_fname = os.path.join(basedir, "private", "secret")
72 self.failUnless(os.path.exists(secret_fname), secret_fname)
73 renew_secret = c.get_renewal_secret()
74 self.failUnless(base32.b2a(renew_secret))
75 cancel_secret = c.get_cancel_secret()
76 self.failUnless(base32.b2a(cancel_secret))
78 BASECONFIG = ("[client]\n"
79 "introducer.furl = \n"
82 def test_reserved_1(self):
83 basedir = "client.Basic.test_reserved_1"
85 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
86 f.write(self.BASECONFIG)
87 f.write("[storage]\n")
88 f.write("enabled = true\n")
89 f.write("reserved_space = 1000\n")
91 c = client.Client(basedir)
92 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 1000)
94 def test_reserved_2(self):
95 basedir = "client.Basic.test_reserved_2"
97 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
98 f.write(self.BASECONFIG)
99 f.write("[storage]\n")
100 f.write("enabled = true\n")
101 f.write("reserved_space = 10K\n")
103 c = client.Client(basedir)
104 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 10*1000)
106 def test_reserved_3(self):
107 basedir = "client.Basic.test_reserved_3"
109 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
110 f.write(self.BASECONFIG)
111 f.write("[storage]\n")
112 f.write("enabled = true\n")
113 f.write("reserved_space = 5mB\n")
115 c = client.Client(basedir)
116 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
119 def test_reserved_4(self):
120 basedir = "client.Basic.test_reserved_4"
122 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
123 f.write(self.BASECONFIG)
124 f.write("[storage]\n")
125 f.write("enabled = true\n")
126 f.write("reserved_space = 78Gb\n")
128 c = client.Client(basedir)
129 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
132 def test_reserved_bad(self):
133 basedir = "client.Basic.test_reserved_bad"
135 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
136 f.write(self.BASECONFIG)
137 f.write("[storage]\n")
138 f.write("enabled = true\n")
139 f.write("reserved_space = bogus\n")
141 c = client.Client(basedir)
142 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 0)
144 def _permute(self, sb, key):
146 for (peerid,rref) in sb.get_servers_for_index(key) ]
148 def test_permute(self):
149 sb = StorageFarmBroker()
150 for k in ["%d" % i for i in range(5)]:
151 sb.add_server(k, None)
153 self.failUnlessEqual(self._permute(sb, "one"), ['3','1','0','4','2'])
154 self.failUnlessEqual(self._permute(sb, "two"), ['0','4','2','1','3'])
156 self.failUnlessEqual(self._permute(sb, "one"), [])
158 def test_versions(self):
159 basedir = "test_client.Basic.test_versions"
161 open(os.path.join(basedir, "introducer.furl"), "w").write("")
162 open(os.path.join(basedir, "vdrive.furl"), "w").write("")
163 c = client.Client(basedir)
164 ss = c.getServiceNamed("storage")
165 verdict = ss.remote_get_version()
166 self.failUnlessEqual(verdict["application-version"],
167 str(allmydata.__full_version__))
168 self.failIfEqual(str(allmydata.__version__), "unknown")
169 self.failUnless("." in str(allmydata.__full_version__),
170 "non-numeric version in '%s'" % allmydata.__version__)
171 all_versions = allmydata.get_package_versions_string()
172 self.failUnless("allmydata-tahoe" in all_versions)
173 log.msg("tahoe versions: %s" % all_versions)
175 stats = c.get_stats()
176 self.failUnless("node.uptime" in stats)
177 self.failUnless(isinstance(stats["node.uptime"], float))
179 def flush_but_dont_ignore(res):
180 d = flushEventualQueue()
186 class Run(unittest.TestCase, testutil.StallMixin):
189 self.sparent = service.MultiService()
190 self.sparent.startService()
192 d = self.sparent.stopService()
193 d.addBoth(flush_but_dont_ignore)
196 def test_loadable(self):
197 basedir = "test_client.Run.test_loadable"
199 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
200 open(os.path.join(basedir, "introducer.furl"), "w").write(dummy)
201 open(os.path.join(basedir, "suicide_prevention_hotline"), "w")
202 c = client.Client(basedir)
204 def test_reloadable(self):
205 basedir = "test_client.Run.test_reloadable"
207 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
208 open(os.path.join(basedir, "introducer.furl"), "w").write(dummy)
209 c1 = client.Client(basedir)
210 c1.setServiceParent(self.sparent)
212 # delay to let the service start up completely. I'm not entirely sure
214 d = self.stall(delay=2.0)
215 d.addCallback(lambda res: c1.disownServiceParent())
216 # the cygwin buildslave seems to need more time to let the old
217 # service completely shut down. When delay=0.1, I saw this test fail,
218 # probably due to the logport trying to reclaim the old socket
219 # number. This suggests that either we're dropping a Deferred
220 # somewhere in the shutdown sequence, or that cygwin is just cranky.
221 d.addCallback(self.stall, delay=2.0)
223 # TODO: pause for slightly over one second, to let
224 # Client._check_hotline poll the file once. That will exercise
225 # another few lines. Then add another test in which we don't
226 # update the file at all, and watch to see the node shutdown. (to
227 # do this, use a modified node which overrides Node.shutdown(),
228 # also change _check_hotline to use it instead of a raw
229 # reactor.stop, also instrument the shutdown event in an
230 # attribute that we can check)
231 c2 = client.Client(basedir)
232 c2.setServiceParent(self.sparent)
233 return c2.disownServiceParent()
234 d.addCallback(_restart)