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'])