From b5e3e22a46df6700415583002a15cc15eaac8514 Mon Sep 17 00:00:00 2001 From: Andrei Zeliankou Date: Thu, 14 Nov 2019 17:14:55 +0300 Subject: Tests: waitforsocket() introduced. --- test/unit/http.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'test/unit/http.py') diff --git a/test/unit/http.py b/test/unit/http.py index 1a68ca3b..c7e3e36d 100644 --- a/test/unit/http.py +++ b/test/unit/http.py @@ -1,4 +1,5 @@ import re +import time import socket import select from unit.main import TestUnit @@ -178,3 +179,20 @@ class TestHTTP(TestUnit): headers[m.group(1)] = [headers[m.group(1)], m.group(2)] 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') -- cgit