src/allmydata/test/test_node.py

   1
   2 from twisted.trial import unittest
   3 from twisted.internet import defer
   4 from twisted.python import log
   5
   6 from foolscap.eventual import flushEventualQueue
   7 from twisted.application import service
   8 from allmydata.node import Node
   9 from allmydata.util import testutil
  10
  11 class LoggingMultiService(service.MultiService):
  12     def log(self, msg):
  13         pass
  14
  15 class TestNode(Node):
  16     CERTFILE='DEFAULT_CERTFILE_BLANK'
  17     PORTNUMFILE='DEFAULT_PORTNUMFILE_BLANK'
  18
  19 class TestCase(unittest.TestCase, testutil.SignalMixin):
  20     def setUp(self):
  21         self.parent = LoggingMultiService()
  22         self.parent.startService()
  23     def tearDown(self):
  24         log.msg("%s.tearDown" % self.__class__.__name__)
  25         d = defer.succeed(None)
  26         d.addCallback(lambda res: self.parent.stopService())
  27         d.addCallback(flushEventualQueue)
  28         return d
  29
  30     def test_advertised_ip_addresses(self):
  31         open('advertised_ip_addresses','w').write('1.2.3.4:5')
  32
  33         n = TestNode()
  34         n.setServiceParent(self.parent)
  35         d = n.when_tub_ready()
  36
  37         def _check_addresses(ignored_result):
  38             self.failUnless("1.2.3.4:5" in n.tub.registerReference(n), n.tub.registerReference(n))
  39
  40         d.addCallback(_check_addresses)
  41         return d
  42