src/allmydata/test/test_client.py

   1
   2 import os
   3 from twisted.trial import unittest
   4 from twisted.application import service
   5 from twisted.python import log
   6
   7 import allmydata
   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
  16
  17 class FakeIntroducerClient(IntroducerClient):
  18     def __init__(self):
  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()
  25
  26 BASECONFIG = ("[client]\n"
  27               "introducer.furl = \n"
  28               )
  29
  30 class Basic(unittest.TestCase):
  31     def test_loadable(self):
  32         basedir = "test_client.Basic.test_loadable"
  33         os.mkdir(basedir)
  34         open(os.path.join(basedir, "introducer.furl"), "w").write("")
  35         client.Client(basedir)
  36
  37     def test_loadable_old_config_bits(self):
  38         basedir = "test_client.Basic.test_loadable_old_config_bits"
  39         os.mkdir(basedir)
  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)
  45         try:
  46             c.getServiceNamed("storage")
  47             self.fail("that was supposed to fail")
  48         except KeyError:
  49             pass
  50
  51     def test_loadable_old_storage_config_bits(self):
  52         basedir = "test_client.Basic.test_loadable_old_storage_config_bits"
  53         os.mkdir(basedir)
  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)
  61
  62     def test_secrets(self):
  63         basedir = "test_client.Basic.test_secrets"
  64         os.mkdir(basedir)
  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))
  73
  74     def test_reserved_1(self):
  75         basedir = "client.Basic.test_reserved_1"
  76         os.mkdir(basedir)
  77         f = open(os.path.join(basedir, "tahoe.cfg"), "w")
  78         f.write(BASECONFIG)
  79         f.write("[storage]\n")
  80         f.write("enabled = true\n")
  81         f.write("reserved_space = 1000\n")
  82         f.close()
  83         c = client.Client(basedir)
  84         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 1000)
  85
  86     def test_reserved_2(self):
  87         basedir = "client.Basic.test_reserved_2"
  88         os.mkdir(basedir)
  89         f = open(os.path.join(basedir, "tahoe.cfg"), "w")
  90         f.write(BASECONFIG)
  91         f.write("[storage]\n")
  92         f.write("enabled = true\n")
  93         f.write("reserved_space = 10K\n")
  94         f.close()
  95         c = client.Client(basedir)
  96         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 10*1000)
  97
  98     def test_reserved_3(self):
  99         basedir = "client.Basic.test_reserved_3"
 100         os.mkdir(basedir)
 101         f = open(os.path.join(basedir, "tahoe.cfg"), "w")
 102         f.write(BASECONFIG)
 103         f.write("[storage]\n")
 104         f.write("enabled = true\n")
 105         f.write("reserved_space = 5mB\n")
 106         f.close()
 107         c = client.Client(basedir)
 108         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
 109                              5*1000*1000)
 110
 111     def test_reserved_4(self):
 112         basedir = "client.Basic.test_reserved_4"
 113         os.mkdir(basedir)
 114         f = open(os.path.join(basedir, "tahoe.cfg"), "w")
 115         f.write(BASECONFIG)
 116         f.write("[storage]\n")
 117         f.write("enabled = true\n")
 118         f.write("reserved_space = 78Gb\n")
 119         f.close()
 120         c = client.Client(basedir)
 121         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space,
 122                              78*1000*1000*1000)
 123
 124     def test_reserved_bad(self):
 125         basedir = "client.Basic.test_reserved_bad"
 126         os.mkdir(basedir)
 127         f = open(os.path.join(basedir, "tahoe.cfg"), "w")
 128         f.write(BASECONFIG)
 129         f.write("[storage]\n")
 130         f.write("enabled = true\n")
 131         f.write("reserved_space = bogus\n")
 132         f.close()
 133         c = client.Client(basedir)
 134         self.failUnlessEqual(c.getServiceNamed("storage").reserved_space, 0)
 135
 136     def _permute(self, sb, key):
 137         return [ peerid
 138                  for (peerid,rref) in sb.get_servers_for_index(key) ]
 139
 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)
 144
 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"), [])
 149
 150     def test_versions(self):
 151         basedir = "test_client.Basic.test_versions"
 152         os.mkdir(basedir)
 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)
 165         # also test stats
 166         stats = c.get_stats()
 167         self.failUnless("node.uptime" in stats)
 168         self.failUnless(isinstance(stats["node.uptime"], float))
 169
 170 def flush_but_dont_ignore(res):
 171     d = flushEventualQueue()
 172     def _done(ignored):
 173         return res
 174     d.addCallback(_done)
 175     return d
 176
 177 class Run(unittest.TestCase, testutil.StallMixin):
 178
 179     def setUp(self):
 180         self.sparent = service.MultiService()
 181         self.sparent.startService()
 182     def tearDown(self):
 183         d = self.sparent.stopService()
 184         d.addBoth(flush_but_dont_ignore)
 185         return d
 186
 187     def test_loadable(self):
 188         basedir = "test_client.Run.test_loadable"
 189         os.mkdir(basedir)
 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)
 194
 195     def test_reloadable(self):
 196         basedir = "test_client.Run.test_reloadable"
 197         os.mkdir(basedir)
 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)
 202
 203         # delay to let the service start up completely. I'm not entirely sure
 204         # this is necessary.
 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)
 213         def _restart(res):
 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)
 226         return d
 227
 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")
 233         f.write(BASECONFIG)
 234         f.close()
 235         c = client.Client(basedir)
 236
 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())
 245
 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())
 254
 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())
 263
 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())
 272
 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())
 281
 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())
 290
 291         unknown_rw = "lafs://from_the_future"
 292         unknown_ro = "lafs://readonly_from_the_future"
 293         n = c.create_node_from_uri(unknown_rw, unknown_ro)
 294         self.failUnless(IFilesystemNode.providedBy(n))
 295         self.failIf(IFileNode.providedBy(n))
 296         self.failIf(IImmutableFileNode.providedBy(n))
 297         self.failIf(IMutableFileNode.providedBy(n))
 298         self.failIf(IDirectoryNode.providedBy(n))
 299         self.failUnless(n.is_unknown())
 300         self.failUnlessEqual(n.get_uri(), unknown_rw)
 301         self.failUnlessEqual(n.get_write_uri(), unknown_rw)
 302         self.failUnlessEqual(n.get_readonly_uri(), "ro." + unknown_ro)