src/allmydata/test/test_client.py

   1 import os
   2 from twisted.trial import unittest
   3 from twisted.application import service
   4
   5 import allmydata
   6 from allmydata.node import OldConfigError, OldConfigOptionError, MissingConfigEntry
   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
  14
  15 import mock
  16
  17 BASECONFIG = ("[client]\n"
  18               "introducer.furl = \n"
  19               )
  20
  21 BASECONFIG_I = ("[client]\n"
  22               "introducer.furl = %s\n"
  23               )
  24
  25 class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
  26     def test_loadable(self):
  27         basedir = "test_client.Basic.test_loadable"
  28         os.mkdir(basedir)
  29         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
  30                            BASECONFIG)
  31         client.Client(basedir)
  32
  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"
  36         os.mkdir(basedir)
  37         fileutil.write(os.path.join(basedir, "tahoe.cfg"),
  38                        BASECONFIG +
  39                        "[storage]\n" +
  40                        "enabled = false\n" +
  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"), "")
  46
  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])
  52
  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))
  58
  59         for oldfile in [
  60             'nickname', 'webport', 'keepalive_timeout', 'log_gatherer.furl',
  61             'disconnect_timeout', 'advertised_ip_addresses', 'helper.furl',
  62             'key_generator.furl', 'stats_gatherer.furl', 'sizelimit',
  63             'run_helper']:
  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)
  67
  68     def test_secrets(self):
  69         basedir = "test_client.Basic.test_secrets"
  70         os.mkdir(basedir)
  71         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
  72                            BASECONFIG)
  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))
  80
  81     def test_reserved_1(self):
  82         basedir = "client.Basic.test_reserved_1"
  83         os.mkdir(basedir)
  84         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
  85                            BASECONFIG + \
  86                            "[storage]\n" + \
  87                            "enabled = true\n" + \
  88                            "reserved_space = 1000\n")
  89         c = client.Client(basedir)
  90         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 1000)
  91
  92     def test_reserved_2(self):
  93         basedir = "client.Basic.test_reserved_2"
  94         os.mkdir(basedir)
  95         fileutil.write(os.path.join(basedir, "tahoe.cfg"),  \
  96                            BASECONFIG + \
  97                            "[storage]\n" + \
  98                            "enabled = true\n" + \
  99                            "reserved_space = 10K\n")
 100         c = client.Client(basedir)
 101         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 10*1000)
 102
 103     def test_reserved_3(self):
 104         basedir = "client.Basic.test_reserved_3"
 105         os.mkdir(basedir)
 106         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
 107                            BASECONFIG + \
 108                            "[storage]\n" + \
 109                            "enabled = true\n" + \
 110                            "reserved_space = 5mB\n")
 111         c = client.Client(basedir)
 112         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
 113                              5*1000*1000)
 114
 115     def test_reserved_4(self):
 116         basedir = "client.Basic.test_reserved_4"
 117         os.mkdir(basedir)
 118         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
 119                            BASECONFIG + \
 120                            "[storage]\n" + \
 121                            "enabled = true\n" + \
 122                            "reserved_space = 78Gb\n")
 123         c = client.Client(basedir)
 124         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
 125                              78*1000*1000*1000)
 126
 127     def test_reserved_bad(self):
 128         basedir = "client.Basic.test_reserved_bad"
 129         os.mkdir(basedir)
 130         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
 131                            BASECONFIG + \
 132                            "[storage]\n" + \
 133                            "enabled = true\n" + \
 134                            "reserved_space = bogus\n")
 135         c = client.Client(basedir)
 136         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 0)
 137
 138     def _permute(self, sb, key):
 139         return [ s.get_longname() for s in sb.get_servers_for_psi(key) ]
 140
 141     def test_permute(self):
 142         sb = StorageFarmBroker(None, True)
 143         for k in ["%d" % i for i in range(5)]:
 144             ann = {"anonymous-storage-FURL": "pb://abcde@nowhere/fake",
 145                    "permutation-seed-base32": base32.b2a(k) }
 146             sb.test_add_rref(k, "rref", ann)
 147
 148         self.failUnlessReallyEqual(self._permute(sb, "one"), ['3','1','0','4','2'])
 149         self.failUnlessReallyEqual(self._permute(sb, "two"), ['0','4','2','1','3'])
 150         sb.servers.clear()
 151         self.failUnlessReallyEqual(self._permute(sb, "one"), [])
 152
 153     def test_versions(self):
 154         basedir = "test_client.Basic.test_versions"
 155         os.mkdir(basedir)
 156         fileutil.write(os.path.join(basedir, "tahoe.cfg"), \
 157                            BASECONFIG + \
 158                            "[storage]\n" + \
 159                            "enabled = true\n")
 160         c = client.Client(basedir)
 161         ss = c.getServiceNamed("storage")
 162         verdict = ss.remote_get_version()
 163         self.failUnlessReallyEqual(verdict["application-version"],
 164                                    str(allmydata.__full_version__))
 165         self.failIfEqual(str(allmydata.__version__), "unknown")
 166         self.failUnless("." in str(allmydata.__full_version__),
 167                         "non-numeric version in '%s'" % allmydata.__version__)
 168         all_versions = allmydata.get_package_versions_string()
 169         self.failUnless(allmydata.__appname__ in all_versions)
 170         # also test stats
 171         stats = c.get_stats()
 172         self.failUnless("node.uptime" in stats)
 173         self.failUnless(isinstance(stats["node.uptime"], float))
 174
 175     @mock.patch('allmydata.util.log.msg')
 176     @mock.patch('allmydata.frontends.drop_upload.DropUploader')
 177     def test_create_drop_uploader(self, mock_drop_uploader, mock_log_msg):
 178         class MockDropUploader(service.MultiService):
 179             name = 'drop-upload'
 180
 181             def __init__(self, client, upload_dircap, local_dir_utf8, inotify=None):
 182                 service.MultiService.__init__(self)
 183                 self.client = client
 184                 self.upload_dircap = upload_dircap
 185                 self.local_dir_utf8 = local_dir_utf8
 186                 self.inotify = inotify
 187
 188         mock_drop_uploader.side_effect = MockDropUploader
 189
 190         upload_dircap = "URI:DIR2:blah"
 191         local_dir_utf8 = u"loc\u0101l_dir".encode('utf-8')
 192         config = (BASECONFIG +
 193                   "[storage]\n" +
 194                   "enabled = false\n" +
 195                   "[drop_upload]\n" +
 196                   "enabled = true\n")
 197
 198         basedir1 = "test_client.Basic.test_create_drop_uploader1"
 199         os.mkdir(basedir1)
 200         fileutil.write(os.path.join(basedir1, "tahoe.cfg"),
 201                        config + "local.directory = " + local_dir_utf8 + "\n")
 202         self.failUnlessRaises(MissingConfigEntry, client.Client, basedir1)
 203
 204         fileutil.write(os.path.join(basedir1, "tahoe.cfg"), config)
 205         fileutil.write(os.path.join(basedir1, "private", "drop_upload_dircap"), "URI:DIR2:blah")
 206         self.failUnlessRaises(MissingConfigEntry, client.Client, basedir1)
 207
 208         fileutil.write(os.path.join(basedir1, "tahoe.cfg"),
 209                        config + "upload.dircap = " + upload_dircap + "\n")
 210         self.failUnlessRaises(OldConfigOptionError, client.Client, basedir1)
 211
 212         fileutil.write(os.path.join(basedir1, "tahoe.cfg"),
 213                        config + "local.directory = " + local_dir_utf8 + "\n")
 214         c1 = client.Client(basedir1)
 215         uploader = c1.getServiceNamed('drop-upload')
 216         self.failUnless(isinstance(uploader, MockDropUploader), uploader)
 217         self.failUnlessReallyEqual(uploader.client, c1)
 218         self.failUnlessReallyEqual(uploader.upload_dircap, upload_dircap)
 219         self.failUnlessReallyEqual(uploader.local_dir_utf8, local_dir_utf8)
 220         self.failUnless(uploader.inotify is None, uploader.inotify)
 221         self.failUnless(uploader.running)
 222
 223         class Boom(Exception):
 224             pass
 225         mock_drop_uploader.side_effect = Boom()
 226
 227         basedir2 = "test_client.Basic.test_create_drop_uploader2"
 228         os.mkdir(basedir2)
 229         os.mkdir(os.path.join(basedir2, "private"))
 230         fileutil.write(os.path.join(basedir2, "tahoe.cfg"),
 231                        BASECONFIG +
 232                        "[drop_upload]\n" +
 233                        "enabled = true\n" +
 234                        "local.directory = " + local_dir_utf8 + "\n")
 235         fileutil.write(os.path.join(basedir2, "private", "drop_upload_dircap"), "URI:DIR2:blah")
 236         c2 = client.Client(basedir2)
 237         self.failUnlessRaises(KeyError, c2.getServiceNamed, 'drop-upload')
 238         self.failUnless([True for arg in mock_log_msg.call_args_list if "Boom" in repr(arg)],
 239                         mock_log_msg.call_args_list)
 240
 241
 242 def flush_but_dont_ignore(res):
 243     d = flushEventualQueue()
 244     def _done(ignored):
 245         return res
 246     d.addCallback(_done)
 247     return d
 248
 249 class Run(unittest.TestCase, testutil.StallMixin):
 250
 251     def setUp(self):
 252         self.sparent = service.MultiService()
 253         self.sparent.startService()
 254     def tearDown(self):
 255         d = self.sparent.stopService()
 256         d.addBoth(flush_but_dont_ignore)
 257         return d
 258
 259     def test_loadable(self):
 260         basedir = "test_client.Run.test_loadable"
 261         os.mkdir(basedir)
 262         dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
 263         fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG_I % dummy)
 264         fileutil.write(os.path.join(basedir, "suicide_prevention_hotline"), "")
 265         client.Client(basedir)
 266
 267     def test_reloadable(self):
 268         basedir = "test_client.Run.test_reloadable"
 269         os.mkdir(basedir)
 270         dummy = "pb://wl74cyahejagspqgy4x5ukrvfnevlknt@127.0.0.1:58889/bogus"
 271         fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG_I % dummy)
 272         c1 = client.Client(basedir)
 273         c1.setServiceParent(self.sparent)
 274
 275         # delay to let the service start up completely. I'm not entirely sure
 276         # this is necessary.
 277         d = self.stall(delay=2.0)
 278         d.addCallback(lambda res: c1.disownServiceParent())
 279         # the cygwin buildslave seems to need more time to let the old
 280         # service completely shut down. When delay=0.1, I saw this test fail,
 281         # probably due to the logport trying to reclaim the old socket
 282         # number. This suggests that either we're dropping a Deferred
 283         # somewhere in the shutdown sequence, or that cygwin is just cranky.
 284         d.addCallback(self.stall, delay=2.0)
 285         def _restart(res):
 286             # TODO: pause for slightly over one second, to let
 287             # Client._check_hotline poll the file once. That will exercise
 288             # another few lines. Then add another test in which we don't
 289             # update the file at all, and watch to see the node shutdown. (to
 290             # do this, use a modified node which overrides Node.shutdown(),
 291             # also change _check_hotline to use it instead of a raw
 292             # reactor.stop, also instrument the shutdown event in an
 293             # attribute that we can check)
 294             c2 = client.Client(basedir)
 295             c2.setServiceParent(self.sparent)
 296             return c2.disownServiceParent()
 297         d.addCallback(_restart)
 298         return d
 299
 300 class NodeMaker(testutil.ReallyEqualMixin, unittest.TestCase):
 301     def test_maker(self):
 302         basedir = "client/NodeMaker/maker"
 303         fileutil.make_dirs(basedir)
 304         fileutil.write(os.path.join(basedir, "tahoe.cfg"), BASECONFIG)
 305         c = client.Client(basedir)
 306
 307         n = c.create_node_from_uri("URI:CHK:6nmrpsubgbe57udnexlkiwzmlu:bjt7j6hshrlmadjyr7otq3dc24end5meo5xcr5xe5r663po6itmq:3:10:7277")
 308         self.failUnless(IFilesystemNode.providedBy(n))
 309         self.failUnless(IFileNode.providedBy(n))
 310         self.failUnless(IImmutableFileNode.providedBy(n))
 311         self.failIf(IMutableFileNode.providedBy(n))
 312         self.failIf(IDirectoryNode.providedBy(n))
 313         self.failUnless(n.is_readonly())
 314         self.failIf(n.is_mutable())
 315
 316         n = c.create_node_from_uri("URI:LIT:n5xgk")
 317         self.failUnless(IFilesystemNode.providedBy(n))
 318         self.failUnless(IFileNode.providedBy(n))
 319         self.failUnless(IImmutableFileNode.providedBy(n))
 320         self.failIf(IMutableFileNode.providedBy(n))
 321         self.failIf(IDirectoryNode.providedBy(n))
 322         self.failUnless(n.is_readonly())
 323         self.failIf(n.is_mutable())
 324
 325         n = c.create_node_from_uri("URI:SSK:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
 326         self.failUnless(IFilesystemNode.providedBy(n))
 327         self.failUnless(IFileNode.providedBy(n))
 328         self.failIf(IImmutableFileNode.providedBy(n))
 329         self.failUnless(IMutableFileNode.providedBy(n))
 330         self.failIf(IDirectoryNode.providedBy(n))
 331         self.failIf(n.is_readonly())
 332         self.failUnless(n.is_mutable())
 333
 334         n = c.create_node_from_uri("URI:SSK-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
 335         self.failUnless(IFilesystemNode.providedBy(n))
 336         self.failUnless(IFileNode.providedBy(n))
 337         self.failIf(IImmutableFileNode.providedBy(n))
 338         self.failUnless(IMutableFileNode.providedBy(n))
 339         self.failIf(IDirectoryNode.providedBy(n))
 340         self.failUnless(n.is_readonly())
 341         self.failUnless(n.is_mutable())
 342
 343         n = c.create_node_from_uri("URI:DIR2:n6x24zd3seu725yluj75q5boaa:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
 344         self.failUnless(IFilesystemNode.providedBy(n))
 345         self.failIf(IFileNode.providedBy(n))
 346         self.failIf(IImmutableFileNode.providedBy(n))
 347         self.failIf(IMutableFileNode.providedBy(n))
 348         self.failUnless(IDirectoryNode.providedBy(n))
 349         self.failIf(n.is_readonly())
 350         self.failUnless(n.is_mutable())
 351
 352         n = c.create_node_from_uri("URI:DIR2-RO:b7sr5qsifnicca7cbk3rhrhbvq:mm6yoqjhl6ueh7iereldqxue4nene4wl7rqfjfybqrehdqmqskvq")
 353         self.failUnless(IFilesystemNode.providedBy(n))
 354         self.failIf(IFileNode.providedBy(n))
 355         self.failIf(IImmutableFileNode.providedBy(n))
 356         self.failIf(IMutableFileNode.providedBy(n))
 357         self.failUnless(IDirectoryNode.providedBy(n))
 358         self.failUnless(n.is_readonly())
 359         self.failUnless(n.is_mutable())
 360
 361         unknown_rw = "lafs://from_the_future"
 362         unknown_ro = "lafs://readonly_from_the_future"
 363         n = c.create_node_from_uri(unknown_rw, unknown_ro)
 364         self.failUnless(IFilesystemNode.providedBy(n))
 365         self.failIf(IFileNode.providedBy(n))
 366         self.failIf(IImmutableFileNode.providedBy(n))
 367         self.failIf(IMutableFileNode.providedBy(n))
 368         self.failIf(IDirectoryNode.providedBy(n))
 369         self.failUnless(n.is_unknown())
 370         self.failUnlessReallyEqual(n.get_uri(), unknown_rw)
 371         self.failUnlessReallyEqual(n.get_write_uri(), unknown_rw)
 372         self.failUnlessReallyEqual(n.get_readonly_uri(), "ro." + unknown_ro)
 373
 374         # Note: it isn't that we *intend* to deploy non-ASCII caps in
 375         # the future, it is that we want to make sure older Tahoe-LAFS
 376         # versions wouldn't choke on them if we were to do so. See
 377         # #1051 and wiki:NewCapDesign for details.
 378         unknown_rw = u"lafs://from_the_future_rw_\u263A".encode('utf-8')
 379         unknown_ro = u"lafs://readonly_from_the_future_ro_\u263A".encode('utf-8')
 380         n = c.create_node_from_uri(unknown_rw, unknown_ro)
 381         self.failUnless(IFilesystemNode.providedBy(n))
 382         self.failIf(IFileNode.providedBy(n))
 383         self.failIf(IImmutableFileNode.providedBy(n))
 384         self.failIf(IMutableFileNode.providedBy(n))
 385         self.failIf(IDirectoryNode.providedBy(n))
 386         self.failUnless(n.is_unknown())
 387         self.failUnlessReallyEqual(n.get_uri(), unknown_rw)
 388         self.failUnlessReallyEqual(n.get_write_uri(), unknown_rw)
 389         self.failUnlessReallyEqual(n.get_readonly_uri(), "ro." + unknown_ro)