def __init__(self, *args, **keys):
if keys.pop("verify_cert", True):
raise socket.sslerror("module 'ssl' required for " +
"certificate verification")
keys.pop("ca_certs", None)
httplib.HTTPSConnection.__init__(self, *args, **keys)
def recv(self, amt=-1):
try:
if self._fail_recv:
raise socket.sslerror(socket.SSL_ERROR_WANT_READ)
return self._sock.recv(amt=amt)
finally:
self._fail_recv = not self._fail_recv
def recv(self, *args, **kwds):
import gnutls.errors
try:
s = self.session.recv(*args, **kwds)
except gnutls.errors.GNUTLSError:
raise socket.sslerror(socket.SSL_ERROR_EOF)
return s
def recv(self, *args, **kwds):
import gnutls.errors
try:
s = self.session.recv(*args, **kwds)
except gnutls.errors.GNUTLSError:
raise socket.sslerror(socket.SSL_ERROR_EOF)
return s
def recv(self, amt=-1):
try:
if self._fail_recv:
raise socket.sslerror(socket.SSL_ERROR_WANT_READ)
return self._sock.recv(amt=amt)
finally:
self._fail_recv = not self._fail_recv
def __init__(self, *args, **keys):
if keys.pop("verify_cert", True):
raise socket.sslerror("module 'ssl' required for " +
"certificate verification")
keys.pop("ca_certs", None)
httplib.HTTPSConnection.__init__(self, *args, **keys)
def test_send(self, SMTP):
client = SMTP.return_value = Mock()
mailer = Mailer(use_ssl=False, use_tls=False)
mailer._send(msg)
client.sendmail.assert_called_With(msg.sender, msg.to, str(msg))
client.quit.side_effect = socket.sslerror()
mailer._send(msg)
client.close.assert_called_with()
def testRetrying(self):
"""We test that conary retries in a bunch of cases"""
class successException(Exception):
'''This exception will be raised in place of actually calling
the actual function'''
pass
self.mockMax = 0
self.theException = None
self.theReturn = socket.socket()
def mockCountdown(*args, **kwargs):
if self.mockMax <= 1:
raise successException
else:
self.mockMax -= 1
raise self.theException
try:
# setup
self.openRepository()
client = conaryclient.ConaryClient(self.cfg)
self.mock(transport.URLOpener, 'createConnection',
mockCountdown)
# test ssl errors
self.theException = socket.sslerror(socket.SSL_ERROR_EOF,
'A particular ssl error')
self.mockMax = 2
try:
client.repos.c['localhost'].checkVersion()
except errors.OpenError, e:
if 'successException' not in e.args[0]:
raise
self.mockMax = 6
try:
client.repos.c['localhost'].checkVersion()
except errors.OpenError, e:
if 'A particular ssl error' not in e.args[0]:
raise
def testRetrying(self):
"""We test that conary retries in a bunch of cases"""
class successException(Exception):
'''This exception will be raised in place of actually calling
the actual function'''
pass
self.mockMax = 0
self.theException = None
self.theReturn = socket.socket()
def mockCountdown(*args, **kwargs):
if self.mockMax <= 1:
raise successException
else:
self.mockMax -= 1
raise self.theException
try:
# setup
self.openRepository()
client = conaryclient.ConaryClient(self.cfg)
self.mock(transport.URLOpener, 'createConnection', mockCountdown)
# test ssl errors
self.theException = socket.sslerror(socket.SSL_ERROR_EOF,
'A particular ssl error')
self.mockMax = 2
try:
client.repos.c['localhost'].checkVersion()
except errors.OpenError, e:
if 'successException' not in e.args[0]:
raise
self.mockMax = 6
try:
client.repos.c['localhost'].checkVersion()
except errors.OpenError, e:
if 'A particular ssl error' not in e.args[0]:
raise
def quit(self):
if self.fail_on_quit:
raise socket.sslerror("dang")
self.quitted = True
self.close()
def quit(self):
if self.fail_on_quit:
raise socket.sslerror("dang")
self.quitted = True
self.close()
def write(self, data):
try:
return self.sendall(data)
except util.SSL.Error, ex:
raise socket.sslerror(str(ex))
