def test_close_ssl_exception(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMock(ssl.SSLSocket)
check.sock.unwrap().AndRaise(ssl.SSLError)
check.sock.close()
self.mox.ReplayAll()
check._close(True)
def test_close_ssl_exception(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMock(ssl.SSLSocket)
check.sock.unwrap().AndRaise(ssl.SSLError)
check.sock.close()
self.mox.ReplayAll()
check._close(True)
def test_close(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMock(socket.socket)
check.sock.close()
check.sock.close().AndRaise(socket.error)
self.mox.ReplayAll()
check._close(False)
check._close(False)
def test_get_banner_multiline(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('220-Part One\r\n')
self.mox.ReplayAll()
banner = check._get_banner()
self.assertIn('Banner-Elapsed', check.results)
self.assertEqual('220-Part One\r\n', banner)
def test_close_ssl(self):
check = SmtpHealthCheck()
orig_sock = self.mox.CreateMock(socket.socket)
check.sock = self.mox.CreateMock(ssl.SSLSocket)
check.sock.unwrap().AndReturn(orig_sock)
orig_sock.close()
self.mox.ReplayAll()
check._close(True)
def test_get_banner_multiline(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('220-Part One\r\n')
self.mox.ReplayAll()
banner = check._get_banner()
self.assertIn('Banner-Elapsed', check.results)
self.assertEqual('220-Part One\r\n', banner)
def test_close_ssl(self):
check = SmtpHealthCheck()
orig_sock = self.mox.CreateMock(socket.socket)
check.sock = self.mox.CreateMock(ssl.SSLSocket)
check.sock.unwrap().AndReturn(orig_sock)
orig_sock.close()
self.mox.ReplayAll()
check._close(True)
def test_close(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMock(socket.socket)
check.sock.close()
check.sock.close().AndRaise(socket.error)
self.mox.ReplayAll()
check._close(False)
check._close(False)
def test_get_banner_long(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('a' * 5120)
check.sock.recv(IsA(int)).AndReturn('a' * 5120)
check.sock.recv(IsA(int)).AndReturn('a')
self.mox.ReplayAll()
with self.assertRaises(BannerError):
check._get_banner()
def test_get_banner_long(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('a'*5120)
check.sock.recv(IsA(int)).AndReturn('a'*5120)
check.sock.recv(IsA(int)).AndReturn('a')
self.mox.ReplayAll()
with self.assertRaises(BannerError):
check._get_banner()
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(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep')
check._get_banner().AndReturn('beep beep')
check._check_banner('beep beep')
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13)
def test_run_exception(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep').AndRaise(Exception('test test'))
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13)
self.assertEqual('Exception', check.results['Exception-Type'])
self.assertEqual('test test', check.results['Exception-Value'])
def test_run(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep')
check._get_banner().AndReturn('beep beep')
check._check_banner('beep beep')
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13)
def test_run_exception(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
self.mox.StubOutWithMock(check, '_lookup')
self.mox.StubOutWithMock(check, '_connect')
self.mox.StubOutWithMock(check, '_get_banner')
self.mox.StubOutWithMock(check, '_check_banner')
check._lookup('test', 13).AndReturn('beep')
check._connect('beep').AndRaise(Exception('test test'))
check.sock.close()
self.mox.ReplayAll()
check.run('test', 13)
self.assertEqual('Exception', check.results['Exception-Type'])
self.assertEqual('test test', check.results['Exception-Value'])
def test_get_banner_slow(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('2')
check.sock.recv(IsA(int)).AndReturn('2')
check.sock.recv(IsA(int)).AndReturn('0')
check.sock.recv(IsA(int)).AndReturn(' ')
check.sock.recv(IsA(int)).AndReturn('O')
check.sock.recv(IsA(int)).AndReturn('k')
check.sock.recv(IsA(int)).AndReturn('\r')
check.sock.recv(IsA(int)).AndReturn('\n')
self.mox.ReplayAll()
banner = check._get_banner()
self.assertIn('Banner-Elapsed', check.results)
self.assertEqual('220 Ok\r\n', banner)
def test_get_banner_slow(self):
check = SmtpHealthCheck()
check.sock = self.mox.CreateMockAnything()
check.sock.recv(IsA(int)).AndReturn('2')
check.sock.recv(IsA(int)).AndReturn('2')
check.sock.recv(IsA(int)).AndReturn('0')
check.sock.recv(IsA(int)).AndReturn(' ')
check.sock.recv(IsA(int)).AndReturn('O')
check.sock.recv(IsA(int)).AndReturn('k')
check.sock.recv(IsA(int)).AndReturn('\r')
check.sock.recv(IsA(int)).AndReturn('\n')
self.mox.ReplayAll()
banner = check._get_banner()
self.assertIn('Banner-Elapsed', check.results)
self.assertEqual('220 Ok\r\n', banner)
