You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
5 years ago
|
import gevent
|
||
|
|
from gevent import socket
|
||
|
|
from gevent import server
|
||
|
|
import gevent.testing as greentest
|
||
|
|
|
||
|
|
# XXX also test: send, sendall, recvfrom, recvfrom_into, sendto
|
||
|
|
|
||
|
|
|
||
|
|
def readall(sock, _):
|
||
|
|
while sock.recv(1024):
|
||
|
|
pass # pragma: no cover we never actually send the data
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
|
||
|
|
class Test(greentest.TestCase):
|
||
|
|
|
||
|
|
error_fatal = False
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.server = server.StreamServer(greentest.DEFAULT_BIND_ADDR_TUPLE, readall)
|
||
|
|
self.server.start()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
self.server.stop()
|
||
|
|
|
||
|
|
def test_recv_closed(self):
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
sock.connect((greentest.DEFAULT_CONNECT_HOST, self.server.server_port))
|
||
|
|
receiver = gevent.spawn(sock.recv, 25)
|
||
|
|
try:
|
||
|
|
gevent.sleep(0.001)
|
||
|
|
sock.close()
|
||
|
|
receiver.join(timeout=0.1)
|
||
|
|
self.assertTrue(receiver.ready(), receiver)
|
||
|
|
self.assertEqual(receiver.value, None)
|
||
|
|
self.assertIsInstance(receiver.exception, socket.error)
|
||
|
|
self.assertEqual(receiver.exception.errno, socket.EBADF)
|
||
|
|
finally:
|
||
|
|
receiver.kill()
|
||
|
|
|
||
|
|
# XXX: This is possibly due to the bad behaviour of small sleeps?
|
||
|
|
# The timeout is the global test timeout, 10s
|
||
|
|
@greentest.skipOnLibuvOnCI("Sometimes randomly times out")
|
||
|
|
def test_recv_twice(self):
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
sock.connect((greentest.DEFAULT_CONNECT_HOST, self.server.server_port))
|
||
|
|
receiver = gevent.spawn(sock.recv, 25)
|
||
|
|
try:
|
||
|
|
gevent.sleep(0.001)
|
||
|
|
self.assertRaises(AssertionError, sock.recv, 25)
|
||
|
|
self.assertRaises(AssertionError, sock.recv, 25)
|
||
|
|
finally:
|
||
|
|
receiver.kill()
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
greentest.main()
|