From: Brian Warner Date: Tue, 12 Jun 2012 06:19:30 +0000 (-0700) Subject: test_system.py: wait for the Helper connection properly before uploading X-Git-Url: https://git.rkrishnan.org/?a=commitdiff_plain;h=51a5aaa63948162091e959b002bd6eaa7bb50a32;p=tahoe-lafs%2Ftahoe-lafs.git test_system.py: wait for the Helper connection properly before uploading --- diff --git a/src/allmydata/test/test_system.py b/src/allmydata/test/test_system.py index 9eb8811a..f3d618d6 100644 --- a/src/allmydata/test/test_system.py +++ b/src/allmydata/test/test_system.py @@ -213,6 +213,12 @@ class SystemTest(SystemTestMixin, RunBinTahoeMixin, unittest.TestCase): self.extra_node.DEFAULT_ENCODING_PARAMETERS['happy'] = 5 d.addCallback(_added) + def _has_helper(): + uploader = self.extra_node.getServiceNamed("uploader") + furl, connected = uploader.get_helper_info() + return connected + d.addCallback(lambda ign: self.poll(_has_helper)) + HELPER_DATA = "Data that needs help to upload" * 1000 def _upload_with_helper(res): u = upload.Data(HELPER_DATA, convergence=convergence) @@ -304,11 +310,12 @@ class SystemTest(SystemTestMixin, RunBinTahoeMixin, unittest.TestCase): self.failIf(os.path.exists(incdir) and os.listdir(incdir)) d.addCallback(_disconnected) - # then we need to give the reconnector a chance to - # reestablish the connection to the helper. d.addCallback(lambda res: - log.msg("wait_for_connections", level=log.NOISY, + log.msg("wait_for_helper", level=log.NOISY, facility="tahoe.test.test_system")) + # then we need to wait for the extra node to reestablish its + # connection to the helper. + d.addCallback(lambda ign: self.poll(_has_helper)) d.addCallback(lambda res: log.msg("uploading again", level=log.NOISY,