2 from twisted.trial import unittest
3 from twisted.application import service
6 from allmydata.node import OldConfigError
7 from allmydata import client
8 from allmydata.storage_client import StorageFarmBroker
9 from allmydata.util import base32, fileutil
10 from allmydata.interfaces import IFilesystemNode, IFileNode, \
11 IImmutableFileNode, IMutableFileNode, IDirectoryNode
12 from foolscap.api import flushEventualQueue
13 import allmydata.test.common_util as testutil
17 BASECONFIG = ("[client]\n"
18 "introducer.furl = \n"
21 BASECONFIG_I = ("[client]\n"
22 "introducer.furl = %s\n"
25 class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
26 def test_loadable(self):
27 basedir = "test_client.Basic.test_loadable"
29 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
31 client.Client(basedir)
33 @mock.patch('twisted.python.log.msg')
34 def test_error_on_old_config_files(self, mock_log_msg):
35 basedir = "test_client.Basic.test_error_on_old_config_files"
37 fileutil.write(os.path.join(basedir, "tahoe.cfg"),
41 "reserved_space = bogus\n")
42 fileutil.write(os.path.join(basedir, "introducer.furl"), "")
43 fileutil.write(os.path.join(basedir, "no_storage"), "")
44 fileutil.write(os.path.join(basedir, "readonly_storage"), "")
45 fileutil.write(os.path.join(basedir, "debug_discard_storage"), "")
47 e = self.failUnlessRaises(OldConfigError, client.Client, basedir)
48 self.failUnlessIn(os.path.abspath(os.path.join(basedir, "introducer.furl")), e.args[0])
49 self.failUnlessIn(os.path.abspath(os.path.join(basedir, "no_storage")), e.args[0])
50 self.failUnlessIn(os.path.abspath(os.path.join(basedir, "readonly_storage")), e.args[0])
51 self.failUnlessIn(os.path.abspath(os.path.join(basedir, "debug_discard_storage")), e.args[0])
53 for oldfile in ['introducer.furl', 'no_storage', 'readonly_storage',
54 'debug_discard_storage']:
55 logged = [ m for m in mock_log_msg.call_args_list if
56 ("Found pre-Tahoe-LAFS-v1.3 configuration file" in str(m[0][0]) and oldfile in str(m[0][0])) ]
57 self.failUnless(logged, (oldfile, mock_log_msg.call_args_list))
60 'nickname', 'webport', 'keepalive_timeout', 'log_gatherer.furl',
61 'disconnect_timeout', 'advertised_ip_addresses', 'helper.furl',
62 'key_generator.furl', 'stats_gatherer.furl', 'sizelimit',
64 logged = [ m for m in mock_log_msg.call_args_list if
65 ("Found pre-Tahoe-LAFS-v1.3 configuration file" in str(m[0][0]) and oldfile in str(m[0][0])) ]
66 self.failIf(logged, oldfile)
68 def test_secrets(self):
69 basedir = "test_client.Basic.test_secrets"
71 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
73 c = client.Client(basedir)
74 secret_fname = os.path.join(basedir, "private", "secret")
75 self.failUnless(os.path.exists(secret_fname), secret_fname)
76 renew_secret = c.get_renewal_secret()
77 self.failUnless(base32.b2a(renew_secret))
78 cancel_secret = c.get_cancel_secret()
79 self.failUnless(base32.b2a(cancel_secret))
81 def test_reserved_1(self):
82 basedir = "client.Basic.test_reserved_1"
84 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
87 "enabled = true\n" + \
88 "reserved_space = 1000\n")
89 c = client.Client(basedir)
90 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 1000)
92 def test_reserved_2(self):
93 basedir = "client.Basic.test_reserved_2"
95 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
98 "enabled = true\n" + \
99 "reserved_space = 10K\n")
100 c = client.Client(basedir)
101 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 10*1000)
103 def test_reserved_3(self):
104 basedir = "client.Basic.test_reserved_3"
106 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
109 "enabled = true\n" + \
110 "reserved_space = 5mB\n")
111 c = client.Client(basedir)
112 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
115 def test_reserved_4(self):
116 basedir = "client.Basic.test_reserved_4"
118 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
121 "enabled = true\n" + \
122 "reserved_space = 78Gb\n")
123 c = client.Client(basedir)
124 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
127 def test_reserved_bad(self):
128 basedir = "client.Basic.test_reserved_bad"
130 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
133 "enabled = true\n" + \
134 "reserved_space = bogus\n")
135 c = client.Client(basedir)
136 self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 0)
138 def _permute(self, sb, key):
139 return [ s.get_serverid() for s in sb.get_servers_for_psi(key) ]
141 def test_permute(self):
142 sb = StorageFarmBroker(None, True)
143 for k in ["%d" % i for i in range(5)]:
144 sb.test_add_rref(k, "rref")
146 self.failUnlessReallyEqual(self._permute(sb, "one"), ['3','1','0','4','2'])
147 self.failUnlessReallyEqual(self._permute(sb, "two"), ['0','4','2','1','3'])
149 self.failUnlessReallyEqual(self._permute(sb, "one"), [])
151 def test_versions(self):
152 basedir = "test_client.Basic.test_versions"
154 fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
158 c = client.Client(basedir)
159 ss = c.getServiceNamed("storage")
160 verdict = ss.remote_get_version()
161 self.failUnlessReallyEqual(verdict["application-version"],
162 str(allmydata.__full_version__))
163 self.failIfEqual(str(allmydata.__version__), "unknown")
164 self.failUnless("." in str(allmydata.__full_version__),
165 "non-numeric version in '%s'" % allmydata.__version__)
166 all_versions = allmydata.get_package_versions_string()
167 self.failUnless(allmydata.__appname__ in all_versions)
169 stats = c.get_stats()
170 self.failUnless("node.uptime" in stats)
171 self.failUnless(isinstance(stats["node.uptime"], float))
173 @mock.patch('allmydata.util.log.msg')
174 @mock.patch('allmydata.frontends.drop_upload.DropUploader')
175 def test_create_drop_uploader(self, mock_drop_uploader, mock_log_msg):
176 class MockDropUploader(service.MultiService):
179 def __init__(self, client, upload_dircap, local_dir_utf8, inotify=None):
180 service.MultiService.__init__(self)
182 self.upload_dircap = upload_dircap
183 self.local_dir_utf8 = local_dir_utf8
184 self.inotify = inotify
186 mock_drop_uploader.side_effect = MockDropUploader
188 upload_dircap = "URI:DIR2:blah"
189 local_dir_utf8 = u"loc\u0101l_dir".encode('utf-8')
190 config = (BASECONFIG +
192 "enabled = false\n" +
195 "upload.dircap = " + upload_dircap + "\n" +
196 "local.directory = " + local_dir_utf8 + "\n")
198 basedir1 = "test_client.Basic.test_create_drop_uploader1"
200 fileutil.write(os.path.join(basedir1, "tahoe.cfg"), config)
201 c1 = client.Client(basedir1)
202 uploader = c1.getServiceNamed('drop-upload')
203 self.failUnless(isinstance(uploader, MockDropUploader), uploader)
204 self.failUnlessReallyEqual(uploader.client, c1)
205 self.failUnlessReallyEqual(uploader.upload_dircap, upload_dircap)
206 self.failUnlessReallyEqual(uploader.local_dir_utf8, local_dir_utf8)
207 self.failUnless(uploader.inotify is None, uploader.inotify)
208 self.failUnless(uploader.running)
210 class Boom(Exception):
212 mock_drop_uploader.side_effect = Boom("something wrong")
214 basedir2 = "test_client.Basic.test_create_drop_uploader2"
216 fileutil.write(os.path.join(basedir2, "tahoe.cfg"),
220 c2 = client.Client(basedir2)
221 self.failUnlessRaises(KeyError, c2.getServiceNamed, 'drop-upload')
222 self.failIf([True for arg in mock_log_msg.call_args_list if "Boom" in repr(arg) or "something wrong" in repr(arg)],
223 mock_log_msg.call_args_list)
224 self.failUnless([True for arg in mock_log_msg.call_args_list if "upload.dircap or local.directory not specified" in repr(arg)],
225 mock_log_msg.call_args_list)
227 basedir3 = "test_client.Basic.test_create_drop_uploader3"
229 fileutil.write(os.path.join(basedir3, "tahoe.cfg"), config)
230 client.Client(basedir3)
231 self.failUnless([True for arg in mock_log_msg.call_args_list if "Boom" in repr(arg) and "something wrong" in repr(arg)],
232 mock_log_msg.call_args_list)
235 def flush_but_dont_ignore(res):
236 d = flushEventualQueue()
242 class Run(unittest.TestCase, testutil.StallMixin):
245 self.sparent = service.MultiService()
246 self.sparent.startService()
248 d = self.sparent.stopService()
249 d.addBoth(flush_but_dont_ignore)
252 def test_loadable(self):
253 basedir = "test_client.Run.test_loadable"
255 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
256 fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG_I % dummy)
257 fileutil.write(os.path.join(basedir, "suicide_prevention_hotline"), "")
258 client.Client(basedir)
260 def test_reloadable(self):
261 basedir = "test_client.Run.test_reloadable"
263 dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
264 fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG_I % dummy)
265 c1 = client.Client(basedir)
266 c1.setServiceParent(self.sparent)
268 # delay to let the service start up completely. I'm not entirely sure
270 d = self.stall(delay=2.0)
271 d.addCallback(lambda res: c1.disownServiceParent())
272 # the cygwin buildslave seems to need more time to let the old
273 # service completely shut down. When delay=0.1, I saw this test fail,
274 # probably due to the logport trying to reclaim the old socket
275 # number. This suggests that either we're dropping a Deferred
276 # somewhere in the shutdown sequence, or that cygwin is just cranky.
277 d.addCallback(self.stall, delay=2.0)
279 # TODO: pause for slightly over one second, to let
280 # Client._check_hotline poll the file once. That will exercise
281 # another few lines. Then add another test in which we don't
282 # update the file at all, and watch to see the node shutdown. (to
283 # do this, use a modified node which overrides Node.shutdown(),
284 # also change _check_hotline to use it instead of a raw
285 # reactor.stop, also instrument the shutdown event in an
286 # attribute that we can check)
287 c2 = client.Client(basedir)
288 c2.setServiceParent(self.sparent)
289 return c2.disownServiceParent()
290 d.addCallback(_restart)
293 class NodeMaker(testutil.ReallyEqualMixin, unittest.TestCase):
294 def test_maker(self):
295 basedir = "client/NodeMaker/maker"
296 fileutil.make_dirs(basedir)
297 fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG)
298 c = client.Client(basedir)
300 n = c.create_node_from_uri("URI:CHK:6nmrpsubgbe57udnexlkiwzmlu:bjt7j6hshrlmadjyr7otq3dc24end5meo5xcr5xe5r663po6itmq:3:10:7277")
301 self.failUnless(IFilesystemNode.providedBy(n))
302 self.failUnless(IFileNode.providedBy(n))
303 self.failUnless(IImmutableFileNode.providedBy(n))
304 self.failIf(IMutableFileNode.providedBy(n))
305 self.failIf(IDirectoryNode.providedBy(n))
306 self.failUnless(n.is_readonly())
307 self.failIf(n.is_mutable())
309 n = c.create_node_from_uri("URI:LIT:n5xgk")
310 self.failUnless(IFilesystemNode.providedBy(n))
311 self.failUnless(IFileNode.providedBy(n))
312 self.failUnless(IImmutableFileNode.providedBy(n))
313 self.failIf(IMutableFileNode.providedBy(n))
314 self.failIf(IDirectoryNode.providedBy(n))
315 self.failUnless(n.is_readonly())
316 self.failIf(n.is_mutable())
318 n = c.create_node_from_uri("URI:SSK:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
319 self.failUnless(IFilesystemNode.providedBy(n))
320 self.failUnless(IFileNode.providedBy(n))
321 self.failIf(IImmutableFileNode.providedBy(n))
322 self.failUnless(IMutableFileNode.providedBy(n))
323 self.failIf(IDirectoryNode.providedBy(n))
324 self.failIf(n.is_readonly())
325 self.failUnless(n.is_mutable())
327 n = c.create_node_from_uri("URI:SSK-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
328 self.failUnless(IFilesystemNode.providedBy(n))
329 self.failUnless(IFileNode.providedBy(n))
330 self.failIf(IImmutableFileNode.providedBy(n))
331 self.failUnless(IMutableFileNode.providedBy(n))
332 self.failIf(IDirectoryNode.providedBy(n))
333 self.failUnless(n.is_readonly())
334 self.failUnless(n.is_mutable())
336 n = c.create_node_from_uri("URI:DIR2:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
337 self.failUnless(IFilesystemNode.providedBy(n))
338 self.failIf(IFileNode.providedBy(n))
339 self.failIf(IImmutableFileNode.providedBy(n))
340 self.failIf(IMutableFileNode.providedBy(n))
341 self.failUnless(IDirectoryNode.providedBy(n))
342 self.failIf(n.is_readonly())
343 self.failUnless(n.is_mutable())
345 n = c.create_node_from_uri("URI:DIR2-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
346 self.failUnless(IFilesystemNode.providedBy(n))
347 self.failIf(IFileNode.providedBy(n))
348 self.failIf(IImmutableFileNode.providedBy(n))
349 self.failIf(IMutableFileNode.providedBy(n))
350 self.failUnless(IDirectoryNode.providedBy(n))
351 self.failUnless(n.is_readonly())
352 self.failUnless(n.is_mutable())
354 unknown_rw = "lafs://from_the_future"
355 unknown_ro = "lafs://readonly_from_the_future"
356 n = c.create_node_from_uri(unknown_rw, unknown_ro)
357 self.failUnless(IFilesystemNode.providedBy(n))
358 self.failIf(IFileNode.providedBy(n))
359 self.failIf(IImmutableFileNode.providedBy(n))
360 self.failIf(IMutableFileNode.providedBy(n))
361 self.failIf(IDirectoryNode.providedBy(n))
362 self.failUnless(n.is_unknown())
363 self.failUnlessReallyEqual(n.get_uri(), unknown_rw)
364 self.failUnlessReallyEqual(n.get_write_uri(), unknown_rw)
365 self.failUnlessReallyEqual(n.get_readonly_uri(), "ro." + unknown_ro)
367 # Note: it isn't that we *intend* to deploy non-ASCII caps in
368 # the future, it is that we want to make sure older Tahoe-LAFS
369 # versions wouldn't choke on them if we were to do so. See
370 # #1051 and wiki:NewCapDesign for details.
371 unknown_rw = u"lafs://from_the_future_rw_\u263A".encode('utf-8')
372 unknown_ro = u"lafs://readonly_from_the_future_ro_\u263A".encode('utf-8')
373 n = c.create_node_from_uri(unknown_rw, unknown_ro)
374 self.failUnless(IFilesystemNode.providedBy(n))
375 self.failIf(IFileNode.providedBy(n))
376 self.failIf(IImmutableFileNode.providedBy(n))
377 self.failIf(IMutableFileNode.providedBy(n))
378 self.failIf(IDirectoryNode.providedBy(n))
379 self.failUnless(n.is_unknown())
380 self.failUnlessReallyEqual(n.get_uri(), unknown_rw)
381 self.failUnlessReallyEqual(n.get_write_uri(), unknown_rw)
382 self.failUnlessReallyEqual(n.get_readonly_uri(), "ro." + unknown_ro)