def test_user_fd(self):
"""
Test that user's own file descriptor can be passed to init
"""
user_fd = StringIO()
receiver = StreamingBodyReceiver(user_fd)
self.assertIdentical(receiver._fd, user_fd)
def test_readback_mode_off(self):
"""
Test that when readback mode is off connectionLost() will simply
callback finished with the fd.
"""
receiver = StreamingBodyReceiver(readback=False)
d = Deferred()
receiver.finished = d
receiver.content_length = 5
fd = receiver._fd
receiver.dataReceived('hello')
why = Failure(ResponseDone('done'))
receiver.connectionLost(why)
self.assertIdentical(d.result, fd)
self.assertIdentical(receiver._fd, fd)
self.failIf(fd.closed)
def test_readback_mode_on(self):
"""
Test that when readback mode is on inside connectionLost() data will
be read back from the start of the file we're streaming and results
passed to finished callback.
"""
receiver = StreamingBodyReceiver()
d = Deferred()
receiver.finished = d
receiver.content_length = 5
fd = receiver._fd
receiver.dataReceived('hello')
why = Failure(ResponseDone('done'))
receiver.connectionLost(why)
self.assertEqual(d.result, 'hello')
self.assert_(fd.closed)
def test_readback_mode_off(self):
"""
Test that when readback mode is off connectionLost() will simply
callback finished with the fd.
"""
receiver = StreamingBodyReceiver(readback=False)
d = Deferred()
receiver.finished = d
receiver.content_length = 5
fd = receiver._fd
receiver.dataReceived('hello')
why = Failure(ResponseDone('done'))
receiver.connectionLost(why)
self.assertIdentical(d.result, fd)
self.assertIdentical(receiver._fd, fd)
self.failIf(fd.closed)
def test_readback_mode_on(self):
"""
Test that when readback mode is on inside connectionLost() data will
be read back from the start of the file we're streaming and results
passed to finished callback.
"""
receiver = StreamingBodyReceiver()
d = Deferred()
receiver.finished = d
receiver.content_length = 5
fd = receiver._fd
receiver.dataReceived('hello')
why = Failure(ResponseDone('done'))
receiver.connectionLost(why)
self.assertEqual(d.result, 'hello')
self.assert_(fd.closed)
def __init__(self):
StreamingBodyReceiver.__init__(self)
TestReceiverProtocol.used = True
def __init__(self):
StreamingBodyReceiver.__init__(self)
TestReceiverProtocol.used = True