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(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_check_banner_failure(self):
check = SmtpHealthCheck()
with self.assertRaises(BannerError):
check._check_banner('520 No!\r\n')
self.assertEquals('CRITICAL', check.results['Status'])
def test_check_banner_invalid(self):
check = SmtpHealthCheck()
with self.assertRaises(BannerSyntaxError):
check._check_banner('asdf\r\n')
self.assertEquals('CRITICAL', check.results['Status'])
def test_check_banner(self):
check = SmtpHealthCheck()
check._check_banner('220 Ok\r\n')
self.assertEqual('220', check.results['Banner-Code'])
self.assertEqual('Ok', check.results['Banner-Message'])
self.assertEqual('OK', check.results['Status'])
def test_check_banner_failure(self):
check = SmtpHealthCheck()
with self.assertRaises(BannerError):
check._check_banner('520 No!\r\n')
self.assertEquals('CRITICAL', check.results['Status'])
def test_check_banner_invalid(self):
check = SmtpHealthCheck()
with self.assertRaises(BannerSyntaxError):
check._check_banner('asdf\r\n')
self.assertEquals('CRITICAL', check.results['Status'])
def test_check_banner(self):
check = SmtpHealthCheck()
check._check_banner('220 Ok\r\n')
self.assertEqual('220', check.results['Banner-Code'])
self.assertEqual('Ok', check.results['Banner-Message'])
self.assertEqual('OK', check.results['Status'])
