def test_read_SSLError(self):
self.t.sock = Mock(name='SSLSocket')
self.t._quick_recv = Mock(name='recv', return_value='4')
self.t._quick_recv.side_effect = transport.SSLError('timed out')
self.t._read_buffer = MagicMock(return_value='AA')
with pytest.raises(socket.timeout):
self.t._read(64)
def test_read_frame__SSLError(self):
self.t._read = Mock()
self.t._read.side_effect = transport.SSLError('timed out')
with pytest.raises(socket.timeout):
self.t.read_frame()
