File tree Expand file tree Collapse file tree 4 files changed +9
-17
lines changed Expand file tree Collapse file tree 4 files changed +9
-17
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,6 @@ install:
12
12
- " pip install -e ."
13
13
- " pip install 'coverage>=3.7,<3.8' coveralls"
14
14
script :
15
- - " python ./setup.py nosetests"
15
+ - " PYTHONFAULTHANDLER=x timeout -sABRT 30s nosetests -vsd "
16
16
after_success :
17
17
- coveralls
Original file line number Diff line number Diff line change @@ -16,21 +16,22 @@ class MockRecvServer(threading.Thread):
16
16
"""
17
17
Single threaded server accepts one connection and recv until EOF.
18
18
"""
19
- def __init__ (self , host = 'localhost' , port = 24224 ):
19
+ def __init__ (self , host = 'localhost' , port = 0 ):
20
20
if host .startswith ('unix://' ):
21
21
self ._sock = socket .socket (socket .AF_UNIX , socket .SOCK_STREAM )
22
22
self ._sock .bind (host [len ('unix://' ):])
23
23
else :
24
24
self ._sock = socket .socket ()
25
25
self ._sock .bind ((host , port ))
26
+ self .port = self ._sock .getsockname ()[1 ]
27
+ self ._sock .listen (1 )
26
28
self ._buf = BytesIO ()
27
29
28
30
threading .Thread .__init__ (self )
29
31
self .start ()
30
32
31
33
def run (self ):
32
34
sock = self ._sock
33
- sock .listen (1 )
34
35
con , _ = sock .accept ()
35
36
while True :
36
37
data = con .recv (4096 )
Original file line number Diff line number Diff line change 11
11
class TestHandler (unittest .TestCase ):
12
12
def setUp (self ):
13
13
super (TestHandler , self ).setUp ()
14
- for port in range (10000 , 20000 ):
15
- try :
16
- self ._server = mockserver .MockRecvServer ('localhost' , port )
17
- self ._port = port
18
- break
19
- except IOError :
20
- pass
14
+ self ._server = mockserver .MockRecvServer ('localhost' )
15
+ self ._port = self ._server .port
21
16
22
17
def get_data (self ):
23
18
return self ._server .get_recieved ()
Original file line number Diff line number Diff line change @@ -41,13 +41,9 @@ def test_tolerant(self):
41
41
class TestSender (unittest .TestCase ):
42
42
def setUp (self ):
43
43
super (TestSender , self ).setUp ()
44
- for port in range (10000 , 20000 ):
45
- try :
46
- self ._server = mockserver .MockRecvServer ('localhost' , port )
47
- break
48
- except IOError as exc :
49
- print (exc )
50
- self ._sender = fluent .sender .FluentSender (tag = 'test' , port = port )
44
+ self ._server = mockserver .MockRecvServer ('localhost' )
45
+ self ._sender = fluent .sender .FluentSender (tag = 'test' ,
46
+ port = self ._server .port )
51
47
52
48
def get_data (self ):
53
49
return self ._server .get_recieved ()
You can’t perform that action at this time.
0 commit comments