def _get_socket(self):
self.log.debug("connecting")
try:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(self.socket_timeout)
sock.connect(self.address)
return sock
except socket.error, e:
raise expected(SocketError(_socket_error_message(self.address, e)))
def _get_socket(self):
self.log.debug("connecting")
try:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(self.socket_timeout)
sock.connect(self.address)
return sock
except socket.error, e:
raise expected(SocketError(_socket_error_message(self.address, e)))
return input
assert_raises(GotSleep, run)
@parameterized([
("keyboard interrupt", KeyboardInterrupt()),
("system exit", SystemExit()),
("greenlet exit", GreenletExit()),
])
def test_expected_errors(self, name, exception):
@runloop(log)
def run():
raise exception
assert_raises(type(exception), run)
@parameterized([
("expected exc", expected(Exception("expected exception"))),
("unexpected exc", Exception("unexpected exception")),
("gevent.Timeout", gevent.Timeout()),
])
def test_restart_on_exception(self, name, exception):
@runloop(log, sleep=GotSleep.mock())
def run():
raise exception
assert_raises(GotSleep, run)
class TestAPIEdge(XXXTestBase):
def assert_edge_clean(self, edge):
assert_equal(edge._call_semaphore.counter,
edge.max_concurrent_calls)
def test_normal_return(self):
@runloop(log, sleep=GotSleep.mock())
def run(input):
return input
assert_raises(GotSleep, run)
def test_greenlet_exit(self):
@runloop(log)
def run():
raise GreenletExit()
assert_raises(GreenletExit, run)
@parameterized([
expected(Exception("expected exception")),
Exception("unexpected exception"),
])
def test_restart_on_exception(self, exception):
@runloop(log, sleep=GotSleep.mock())
def run():
raise exception
assert_raises(GotSleep, run)
class TestAPIEdge(XXXTestBase):
def assert_edge_clean(self, edge):
assert_equal(edge._call_semaphore.counter, edge.max_concurrent_calls)
def test_normal_call(self):