ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-03-21 03:15:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test_util: add full coverage for allmydata.util.deferredutil

Browse Source
This commit is contained in:
Brian Warner 2008-02-06 16:41:04 -07:00
parent b027fb5b9b
commit d87bfa9eb2
1 changed files with 32 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/allmydata/test/test_util.py
View File

@ -3,9 +3,11 @@ def foo(): pass # keep the line number constant
import os
from twisted.trial import unittest
from twisted.internet import defer
from twisted.python import failure
from allmydata.util import bencode, idlib, humanreadable, mathutil
from allmydata.util import assertutil, fileutil, testutil
from allmydata.util import assertutil, fileutil, testutil, deferredutil
class IDLib(unittest.TestCase):
@ -396,3 +398,32 @@ class PollMixinTests(unittest.TestCase):
return None # success
d.addCallbacks(_suc, _err)
return d
class DeferredUtilTests(unittest.TestCase):
def test_success(self):
d1, d2 = defer.Deferred(), defer.Deferred()
good = []
bad = []
dlss = deferredutil.DeferredListShouldSucceed([d1,d2])
dlss.addCallbacks(good.append, bad.append)
d1.callback(1)
d2.callback(2)
self.failUnlessEqual(good, [[1,2]])
self.failUnlessEqual(bad, [])
def test_failure(self):
d1, d2 = defer.Deferred(), defer.Deferred()
good = []
bad = []
dlss = deferredutil.DeferredListShouldSucceed([d1,d2])
dlss.addCallbacks(good.append, bad.append)
d1.addErrback(lambda _ignore: None)
d2.addErrback(lambda _ignore: None)
d1.callback(1)
d2.errback(RuntimeError())
self.failUnlessEqual(good, [])
self.failUnlessEqual(len(bad), 1)
f = bad[0]
self.failUnless(isinstance(f, failure.Failure))
self.failUnless(f.check(RuntimeError))