一.TCP服务器交互检查
import socket import time def TCP_Server_Interaction(host,port): MaxBytes=1024*1024 try: client = socket.socket(socket.AF_INET,socket.SOCK_STREAM) client.settimeout(30) client.connect((host,port)) check_time = time.strftime('%Y-%m-%d_%H_%M_%S', time.localtime(time.time())) inputData= 'CheckDataCommunication-{}'.format(check_time) sendBytes = client.send(inputData.encode()) print('sendBytes : {}'.format(sendBytes)) recvData = client.recv(MaxBytes) client.close() if recvData.decode()==inputData: print('TCP Communication normal') return 'PASS' else: return 'TCP server connect fail' except: return 'TCP server connect fail' if __name__=="__main__": host = '0.0.0.0' port =xxxx TCP_Server_Interaction(host, port)二.UDP服务器交互检查
import socket import time def UDP_Server_Interaction(host,port): addr = (host, port) server_conn = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) print('server_conn res:{}'.format(server_conn)) check_time = time.strftime('%Y-%m-%d_%H_%M_%S', time.localtime(time.time())) send_data = 'CheckDataCommunication-{}'.format(check_time) UDP_send = server_conn.sendto(send_data.encode(), addr) print('UDP_send_res : {}'.format(UDP_send)) UDP_response, addr = server_conn.recvfrom(2048) print('UDP_response : {}'.format(UDP_response)) server_conn.close() if UDP_response.decode() == send_data: print('UDP Communication normal') return 'PASS' else: return 'UDP server connect fail' if __name__=="__main__": host = '0.0.0.0' port = xxxx print(UDP_Server_Interaction(host, port))