def test_wrap_ssl(self):
check = SmtpHealthCheck()
check.sock = 13
ssl_sock = self.mox.CreateMock(ssl.SSLSocket)
self.mox.StubOutWithMock(ssl, 'wrap_socket')
ssl.wrap_socket(13).AndReturn(ssl_sock)
ssl_sock.do_handshake()
self.mox.ReplayAll()
check._wrap_ssl()
self.assertEqual(ssl_sock, check.sock)
def test_wrap_ssl(self):
check = SmtpHealthCheck()
check.sock = 13
ssl_sock = self.mox.CreateMock(ssl.SSLSocket)
self.mox.StubOutWithMock(ssl, 'wrap_socket')
ssl.wrap_socket(13).AndReturn(ssl_sock)
ssl_sock.do_handshake()
self.mox.ReplayAll()
check._wrap_ssl()
self.assertEqual(ssl_sock, check.sock)
def test_run_ssl(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_wrap_ssl')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep')
check._wrap_ssl()
check._get_banner().AndReturn('beep beep')
check._check_banner('beep beep')
check.sock.unwrap().AndReturn(check.sock)
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13, with_ssl=True)
def test_run_ssl(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_wrap_ssl')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep')
check._wrap_ssl()
check._get_banner().AndReturn('beep beep')
check._check_banner('beep beep')
check.sock.unwrap().AndReturn(check.sock)
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13, with_ssl=True)
