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