Initial test refactoring
This commit is contained in:
31
tests/utils/tcpserver.py
Normal file
31
tests/utils/tcpserver.py
Normal file
@@ -0,0 +1,31 @@
|
||||
from multiprocessing import Process
|
||||
import socket
|
||||
|
||||
class TcpServer:
|
||||
def __init__(self,port):
|
||||
def _startServer(port):
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
sock.bind(('127.0.0.1', port))
|
||||
sock.listen(8)
|
||||
while True:
|
||||
connection,address = sock.accept()
|
||||
buf = connection.recv(4096)
|
||||
connection.send(buf)
|
||||
connection.close()
|
||||
self.port = port
|
||||
self.server = Process(target=_startServer,args=[port])
|
||||
|
||||
def start(self):
|
||||
self.server.start()
|
||||
|
||||
def stop(self):
|
||||
self.server.terminate()
|
||||
|
||||
def sendCheckData(self,data):
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
s.connect(('localhost', self.port))
|
||||
s.sendall(data)
|
||||
received_data = s.recv(4096)
|
||||
s.close()
|
||||
return received_data == data
|
||||
Reference in New Issue
Block a user