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_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_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_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_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)