Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- diff --git a/src/allmydata/test/test_node.py b/src/allmydata/test/test_node.py
- index 24ec571..f3d9f6c 100644
- --- a/src/allmydata/test/test_node.py
- +++ b/src/allmydata/test/test_node.py
- @@ -7,7 +7,7 @@ from twisted.python import log
- from foolscap.api import flushEventualQueue
- from twisted.application import service
- from allmydata.node import Node, formatTimeTahoeStyle
- -from allmydata.util import fileutil
- +from allmydata.util import fileutil, iputil
- import allmydata.test.common_util as testutil
- class LoggingMultiService(service.MultiService):
- @@ -31,6 +31,36 @@ class TestCase(testutil.SignalMixin, unittest.TestCase):
- d.addCallback(flushEventualQueue)
- return d
- + def _mock_get_local_addresses_async(self):
- + self._get_local_addresses_async = iputil.get_local_addresses_async
- + iputil.get_local_addresses_async = lambda : ["1.2.3.4"]
- +
- + def _unmock_get_local_addresses_async(self):
- + iputil.get_local_addresses_async = self._get_local_addresses_async
- +
- + def test_anonymous_location(self):
- + self._mock_get_local_addresses_async()
- +
- + basedir = "test_node/test_anonymous_location"
- + fileutil.make_dirs(basedir)
- + f = open(os.path.join(basedir, 'tahoe.cfg'), 'wt')
- + f.write("[node]\n")
- + f.write("tub.location = \n")
- + f.close()
- +
- + n = TestNode(basedir)
- + n.setServiceParent(self.parent)
- + d = n.when_tub_ready()
- +
- + def _check_addresses(ignored_result):
- + furl = n.tub.registerReference(n)
- + self.failIf("1.2.3.4" in furl, furl)
- + self.failUnless("127.0.0.1" in furl, furl)
- +
- + d.addCallback(_check_addresses)
- + d.addCallback(lambda res: self._unmock_get_local_addresses_async())
- + return d
- +
- def test_location(self):
- basedir = "test_node/test_location"
- fileutil.make_dirs(basedir)
Add Comment
Please, Sign In to add comment