def test_body_collector_errs_on_PotentialDataLoss(self):
bc = BodyCollector()
d = bc.start()
bc.dataReceived('a')
bc.dataReceived('b')
bc.connectionLost(PotentialDataLoss())
d1 = self.failUnlessFailure(d, PotentialDataLoss)
return d1
def test_body_collector_collects_body(self):
bc = BodyCollector()
d = bc.start()
bc.dataReceived('a')
bc.dataReceived('b')
bc.connectionLost(ResponseDone())
def _check(res):
self.failUnless(res is bc, (res, bc))
self.failUnless(isinstance(res.reason, ResponseDone), (res.reason, type(res.reason)))
self.failUnless(res.bytes == 'ab', res.bytes)
d.addCallback(_check)
return d
