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, fileutil
12 from allmydata.interfaces import IFilesystemNode, IFileNode, \
13 IImmutableFileNode, IMutableFileNode, IDirectoryNode
14 from foolscap.api import flushEventualQueue
15 import common_util as testutil
17 class FakeIntroducerClient(IntroducerClient):
19 self._connections = set()
20 def add_peer(self, nodeid):
21 entry = (nodeid, "storage", "rref")
22 self._connections.add(entry)
23 def remove_all_peers(self):
24 self._connections.clear()
26 BASECONFIG = ("[client]\n"
27 "introducer.furl = \n"
30 class Basic(unittest.TestCase):
31 def test_loadable(self):
32 basedir = "test_client.Basic.test_loadable"
34 open(os.path.join(basedir, "introducer.furl"), "w").write("")
35 client.Client(basedir)
37 def test_loadable_old_config_bits(self):
38 basedir = "test_client.Basic.test_loadable_old_config_bits"
40 open(os.path.join(basedir, "introducer.furl"), "w").write("")
41 open(os.path.join(basedir, "no_storage"), "w").write("")
42 open(os.path.join(basedir, "readonly_storage"), "w").write("")
43 open(os.path.join(basedir, "debug_discard_storage"), "w").write("")
44 c = client.Client(basedir)
46 c.getServiceNamed("storage")
47 self.fail("that was supposed to fail")
51 def test_loadable_old_storage_config_bits(self):
52 basedir = "test_client.Basic.test_loadable_old_storage_config_bits"
54 open(os.path.join(basedir, "introducer.furl"), "w").write("")
55 open(os.path.join(basedir, "readonly_storage"), "w").write("")
56 open(os.path.join(basedir, "debug_discard_storage"), "w").write("")
57 c = client.Client(basedir)
58 s = c.getServiceNamed("storage")
59 self.failUnless(s.no_storage)
60 self.failUnless(s.readonly_storage)
62 def test_secrets(self):
63 basedir = "test_client.Basic.test_secrets"
65 open(os.path.join(basedir, "introducer.furl"), "w").write("")
66 c = client.Client(basedir)
67 secret_fname = os.path.join(basedir, "private", "secret")
68 self.failUnless(os.path.exists(secret_fname), secret_fname)
69 renew_secret = c.get_renewal_secret()
70 self.failUnless(base32.b2a(renew_secret))
71 cancel_secret = c.get_cancel_secret()
72 self.failUnless(base32.b2a(cancel_secret))
74 def test_reserved_1(self):
75 basedir = "client.Basic.test_reserved_1"
77 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
79 f.write("[storage]\n")
80 f.write("enabled = true\n")
81 f.write("reserved_space = 1000\n")
83 c = client.Client(basedir)
84 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 1000)
86 def test_reserved_2(self):
87 basedir = "client.Basic.test_reserved_2"
89 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
91 f.write("[storage]\n")
92 f.write("enabled = true\n")
93 f.write("reserved_space = 10K\n")
95 c = client.Client(basedir)
96 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 10*1000)
98 def test_reserved_3(self):
99 basedir = "client.Basic.test_reserved_3"
101 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
103 f.write("[storage]\n")
104 f.write("enabled = true\n")
105 f.write("reserved_space = 5mB\n")
107 c = client.Client(basedir)
108 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
111 def test_reserved_4(self):
112 basedir = "client.Basic.test_reserved_4"
114 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
116 f.write("[storage]\n")
117 f.write("enabled = true\n")
118 f.write("reserved_space = 78Gb\n")
120 c = client.Client(basedir)
121 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
124 def test_reserved_bad(self):
125 basedir = "client.Basic.test_reserved_bad"
127 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
129 f.write("[storage]\n")
130 f.write("enabled = true\n")
131 f.write("reserved_space = bogus\n")
133 c = client.Client(basedir)
134 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 0)
136 def _permute(self, sb, key):
138 for (peerid,rref) in sb.get_servers_for_index(key) ]
140 def test_permute(self):
141 sb = StorageFarmBroker(None, True)
142 for k in ["%d" % i for i in range(5)]:
143 sb.test_add_server(k, None)
145 self.failUnlessEqual(self._permute(sb, "one"), ['3','1','0','4','2'])
146 self.failUnlessEqual(self._permute(sb, "two"), ['0','4','2','1','3'])
147 sb.test_servers.clear()
148 self.failUnlessEqual(self._permute(sb, "one"), [])
150 def test_versions(self):
151 basedir = "test_client.Basic.test_versions"
153 open(os.path.join(basedir, "introducer.furl"), "w").write("")
154 c = client.Client(basedir)
155 ss = c.getServiceNamed("storage")
156 verdict = ss.remote_get_version()
157 self.failUnlessEqual(verdict["application-version"],
158 str(allmydata.__full_version__))
159 self.failIfEqual(str(allmydata.__version__), "unknown")
160 self.failUnless("." in str(allmydata.__full_version__),
161 "non-numeric version in '%s'" % allmydata.__version__)
162 all_versions = allmydata.get_package_versions_string()
163 self.failUnless("allmydata-tahoe" in all_versions)
164 log.msg("tahoe versions: %s" % all_versions)
166 stats = c.get_stats()
167 self.failUnless("node.uptime" in stats)
168 self.failUnless(isinstance(stats["node.uptime"], float))
170 def flush_but_dont_ignore(res):
171 d = flushEventualQueue()
177 class Run(unittest.TestCase, testutil.StallMixin):
180 self.sparent = service.MultiService()
181 self.sparent.startService()
183 d = self.sparent.stopService()
184 d.addBoth(flush_but_dont_ignore)
187 def test_loadable(self):
188 basedir = "test_client.Run.test_loadable"
190 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
191 open(os.path.join(basedir, "introducer.furl"), "w").write(dummy)
192 open(os.path.join(basedir, "suicide_prevention_hotline"), "w")
193 client.Client(basedir)
195 def test_reloadable(self):
196 basedir = "test_client.Run.test_reloadable"
198 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
199 open(os.path.join(basedir, "introducer.furl"), "w").write(dummy)
200 c1 = client.Client(basedir)
201 c1.setServiceParent(self.sparent)
203 # delay to let the service start up completely. I'm not entirely sure
205 d = self.stall(delay=2.0)
206 d.addCallback(lambda res: c1.disownServiceParent())
207 # the cygwin buildslave seems to need more time to let the old
208 # service completely shut down. When delay=0.1, I saw this test fail,
209 # probably due to the logport trying to reclaim the old socket
210 # number. This suggests that either we're dropping a Deferred
211 # somewhere in the shutdown sequence, or that cygwin is just cranky.
212 d.addCallback(self.stall, delay=2.0)
214 # TODO: pause for slightly over one second, to let
215 # Client._check_hotline poll the file once. That will exercise
216 # another few lines. Then add another test in which we don't
217 # update the file at all, and watch to see the node shutdown. (to
218 # do this, use a modified node which overrides Node.shutdown(),
219 # also change _check_hotline to use it instead of a raw
220 # reactor.stop, also instrument the shutdown event in an
221 # attribute that we can check)
222 c2 = client.Client(basedir)
223 c2.setServiceParent(self.sparent)
224 return c2.disownServiceParent()
225 d.addCallback(_restart)
228 class NodeMaker(unittest.TestCase):
229 def test_maker(self):
230 basedir = "client/NodeMaker/maker"
231 fileutil.make_dirs(basedir)
232 f = open(os.path.join(basedir, "tahoe.cfg"), "w")
235 c = client.Client(basedir)
237 n = c.create_node_from_uri("URI:CHK:6nmrpsubgbe57udnexlkiwzmlu:bjt7j6hshrlmadjyr7otq3dc24end5meo5xcr5xe5r663po6itmq:3:10:7277")
238 self.failUnless(IFilesystemNode.providedBy(n))
239 self.failUnless(IFileNode.providedBy(n))
240 self.failUnless(IImmutableFileNode.providedBy(n))
241 self.failIf(IMutableFileNode.providedBy(n))
242 self.failIf(IDirectoryNode.providedBy(n))
243 self.failUnless(n.is_readonly())
244 self.failIf(n.is_mutable())
246 n = c.create_node_from_uri("URI:LIT:n5xgk")
247 self.failUnless(IFilesystemNode.providedBy(n))
248 self.failUnless(IFileNode.providedBy(n))
249 self.failUnless(IImmutableFileNode.providedBy(n))
250 self.failIf(IMutableFileNode.providedBy(n))
251 self.failIf(IDirectoryNode.providedBy(n))
252 self.failUnless(n.is_readonly())
253 self.failIf(n.is_mutable())
255 n = c.create_node_from_uri("URI:SSK:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
256 self.failUnless(IFilesystemNode.providedBy(n))
257 self.failUnless(IFileNode.providedBy(n))
258 self.failIf(IImmutableFileNode.providedBy(n))
259 self.failUnless(IMutableFileNode.providedBy(n))
260 self.failIf(IDirectoryNode.providedBy(n))
261 self.failIf(n.is_readonly())
262 self.failUnless(n.is_mutable())
264 n = c.create_node_from_uri("URI:SSK-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
265 self.failUnless(IFilesystemNode.providedBy(n))
266 self.failUnless(IFileNode.providedBy(n))
267 self.failIf(IImmutableFileNode.providedBy(n))
268 self.failUnless(IMutableFileNode.providedBy(n))
269 self.failIf(IDirectoryNode.providedBy(n))
270 self.failUnless(n.is_readonly())
271 self.failUnless(n.is_mutable())
273 n = c.create_node_from_uri("URI:DIR2:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
274 self.failUnless(IFilesystemNode.providedBy(n))
275 self.failIf(IFileNode.providedBy(n))
276 self.failIf(IImmutableFileNode.providedBy(n))
277 self.failIf(IMutableFileNode.providedBy(n))
278 self.failUnless(IDirectoryNode.providedBy(n))
279 self.failIf(n.is_readonly())
280 self.failUnless(n.is_mutable())
282 n = c.create_node_from_uri("URI:DIR2-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
283 self.failUnless(IFilesystemNode.providedBy(n))
284 self.failIf(IFileNode.providedBy(n))
285 self.failIf(IImmutableFileNode.providedBy(n))
286 self.failIf(IMutableFileNode.providedBy(n))
287 self.failUnless(IDirectoryNode.providedBy(n))
288 self.failUnless(n.is_readonly())
289 self.failUnless(n.is_mutable())
291 future = "x-tahoe-crazy://future_cap_format."
292 n = c.create_node_from_uri(future)
293 self.failUnless(IFilesystemNode.providedBy(n))
294 self.failIf(IFileNode.providedBy(n))
295 self.failIf(IImmutableFileNode.providedBy(n))
296 self.failIf(IMutableFileNode.providedBy(n))
297 self.failIf(IDirectoryNode.providedBy(n))
298 self.failUnlessEqual(n.get_uri(), future)