Tests: waitforsocket() introduced.

This commit is contained in:
Andrei Zeliankou
2019-11-14 17:14:55 +03:00
parent 02e197e978
commit b5e3e22a46

View File

@@ -1,4 +1,5 @@
import re import re
import time
import socket import socket
import select import select
from unit.main import TestUnit from unit.main import TestUnit
@@ -178,3 +179,20 @@ class TestHTTP(TestUnit):
headers[m.group(1)] = [headers[m.group(1)], m.group(2)] headers[m.group(1)] = [headers[m.group(1)], m.group(2)]
return {'status': int(status), 'headers': headers, 'body': body} return {'status': int(status), 'headers': headers, 'body': body}
def waitforsocket(self, port):
ret = False
for i in range(50):
try:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', port))
ret = True
break
except:
sock.close()
time.sleep(0.1)
sock.close()
self.assertTrue(ret, 'socket connected')