def test_concurrent_receivers(self):
wrappers = []
wrapped = mock.Mock(Protocol)
unwrapped = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(wrapped)
br.deliverBody(unwrapped)
response.deliverBody.assert_called_once_with(wrappers[0])
wrappers[0].dataReceived(b"foo")
wrapped.dataReceived.assert_called_once_with(b"foo")
self.assertEqual(unwrapped.dataReceived.call_count, 0)
class TestResponseDone(Exception):
pass
done = Failure(TestResponseDone())
wrappers[0].connectionLost(done)
wrapped.connectionLost.assert_called_once_with(done)
unwrapped.dataReceived.assert_called_once_with(b"foo")
unwrapped.connectionLost.assert_called_once_with(done)
def test_concurrent_receivers(self):
wrappers = []
wrapped = mock.Mock(Protocol)
unwrapped = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(wrapped)
br.deliverBody(unwrapped)
response.deliverBody.assert_called_once_with(wrappers[0])
wrappers[0].dataReceived(b"foo")
wrapped.dataReceived.assert_called_once_with(b"foo")
self.assertEqual(unwrapped.dataReceived.call_count, 0)
class TestResponseDone(Exception):
pass
done = Failure(TestResponseDone())
wrappers[0].connectionLost(done)
wrapped.connectionLost.assert_called_once_with(done)
unwrapped.dataReceived.assert_called_once_with(b"foo")
unwrapped.connectionLost.assert_called_once_with(done)
def setUp(self):
self.response = mock.Mock()
self.protocol = None
def deliverBody(protocol):
self.protocol = protocol
self.response.deliverBody.side_effect = deliverBody
self.response = _BufferedResponse(self.response)
def setUp(self):
self.response = mock.Mock()
self.protocol = None
def deliverBody(protocol):
self.protocol = protocol
self.response.deliverBody.side_effect = deliverBody
self.response = _BufferedResponse(self.response)
def test_wraps_protocol(self):
wrappers = []
wrapped = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(wrapped)
response.deliverBody.assert_called_once_with(wrappers[0])
self.assertNotEqual(wrapped, wrappers[0])
def test_wraps_protocol(self):
wrappers = []
wrapped = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(wrapped)
response.deliverBody.assert_called_once_with(wrappers[0])
self.assertNotEqual(wrapped, wrappers[0])
def test_receiver_after_finished(self):
wrappers = []
finished = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(mock.Mock(Protocol))
wrappers[0].dataReceived(b"foo")
class TestResponseDone(Exception):
pass
done = Failure(TestResponseDone())
wrappers[0].connectionLost(done)
br.deliverBody(finished)
finished.dataReceived.assert_called_once_with(b"foo")
finished.connectionLost.assert_called_once_with(done)
def test_receiver_after_finished(self):
wrappers = []
finished = mock.Mock(Protocol)
response = mock.Mock(deliverBody=mock.Mock(wraps=wrappers.append))
br = _BufferedResponse(response)
br.deliverBody(mock.Mock(Protocol))
wrappers[0].dataReceived(b"foo")
class TestResponseDone(Exception):
pass
done = Failure(TestResponseDone())
wrappers[0].connectionLost(done)
br.deliverBody(finished)
finished.dataReceived.assert_called_once_with(b"foo")
finished.connectionLost.assert_called_once_with(done)
