接口测试tcp mock编写

最新推荐文章于 2024-06-27 16:22:03 发布

lyl-hello

最新推荐文章于 2024-06-27 16:22:03 发布

阅读量3.1k

点赞数

分类专栏： python 接口测试文章标签： tcp mock

本文链接：https://blog.csdn.net/qq_37023538/article/details/53957308

版权

本文介绍了如何编写TCP Mock代码进行接口测试，强调了代码的可扩展性，可以根据测试用例自定义返回值。此外，文章还提及可以采用相同思路模拟UDP、protobuf和Thrift协议的Mock。

摘要由CSDN通过智能技术生成

接口测试中经常会有下游的tcp服务，这样就需要些一个mock来模拟tcp。

import sys
import socket

class TcpServer:
    def __init__(self,_ip,_port):
        self.recv_addr = (_ip, _port)
        self.buffersize = 10240000
        self.comm_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.comm_socket.bind(self.recv_addr)
        self.comm_socket.listen(10)

    def send(self,string_data,addr):
        self.client_socket.send(string_data)

    def recv(self):
        self.client_sock,self.recv_from_addr = self.comm_socket.accept()
        self.client_socket = self.client_sock
        self.string_data = self.client_sock.recv(self.buffersize)
        return (self.string_data,self.recv_from_addr)
		
    def __del__(self):
        self.comm_socket.close()
        self.client_socket.close()

def mock():