src/allmydata/test/test_client.py

   1
   2 import os
   3 from twisted.trial import unittest
   4
   5 from allmydata import client, introducer
   6
   7 class MyIntroducerClient(introducer.IntroducerClient):
   8     def __init__(self):
   9         self.connections = {}
  10
  11 def permute(c, key):
  12     return [ y for x, y, z in c.get_permuted_peers(key) ]
  13
  14 class Basic(unittest.TestCase):
  15     def test_loadable(self):
  16         basedir = "test_client.Basic.test_loadable"
  17         os.mkdir(basedir)
  18         open(os.path.join(basedir, "introducer.furl"), "w").write("")
  19         open(os.path.join(basedir, "vdrive.furl"), "w").write("")
  20         c = client.Client(basedir)
  21
  22     def test_permute(self):
  23         basedir = "test_client.Basic.test_permute"
  24         os.mkdir(basedir)
  25         open(os.path.join(basedir, "introducer.furl"), "w").write("")
  26         open(os.path.join(basedir, "vdrive.furl"), "w").write("")
  27         c = client.Client(basedir)
  28         c.introducer_client = MyIntroducerClient()
  29         for k in ["%d" % i for i in range(5)]:
  30             c.introducer_client.connections[k] = None
  31         self.failUnlessEqual(permute(c, "one"), ['3','1','0','4','2'])
  32         self.failUnlessEqual(permute(c, "two"), ['0','4','2','1','3'])
  33         c.introducer_client.connections.clear()
  34         self.failUnlessEqual(permute(c, "one"), [])
  35
  36         c2 = client.Client(basedir)
  37         c2.introducer_client = MyIntroducerClient()
  38         for k in ["%d" % i for i in range(5)]:
  39             c2.introducer_client.connections[k] = None
  40         self.failUnlessEqual(permute(c2, "one"), ['3','1','0','4','2'])
  41