使用HttpMockSequence()模拟HTTP请求和响应的几个示例

HttpMockSequence()函数是Python中模拟HTTP请求和响应的一个强大工具。它可以用于测试代码中使用HTTP请求的部分，以及构建和验证网络爬虫。下面我将为你提供几个使用HttpMockSequence()函数的示例，并进行详细解释。

示例一：模拟GET请求

from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.request import urlopen
from httpdomock import HttpMockSequence

# 定义模拟请求响应
mock = HttpMockSequence([
    ({'url': 'http://example.com'}, {'status_code': 200, 'content': b'Hello, World!'}),
])

# 启动一个本地HTTP服务器来处理请求
class RequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        response = mock.get_response(self.path)
        self.send_response(response['status_code'])
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write(response['content'])

httpd = HTTPServer(('localhost', 8000), RequestHandler)
httpd.handle_request()

# 发起一个GET请求并打印结果
response = urlopen('http://localhost:8000')
print(response.read().decode())

在上面的示例中，我们使用HttpMockSequence()函数创建了一个模拟请求响应。对于请求URL http://example.com，我们指定了响应的状态码（200）和内容（b'Hello, World!'）。然后，我们启动了一个本地HTTP服务器来处理请求，并在处理请求时调用模拟响应。最后，我们使用urlopen()函数向服务器发起了一个GET请求，并打印了响应的内容。

示例二：模拟POST请求

from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.request import urlopen
from httpdomock import HttpMockSequence

# 定义模拟请求响应
mock = HttpMockSequence([
    ({'url': 'http://example.com', 'method': 'POST', 'data': b'username=test'}, {'status_code': 200, 'content': b'Success!'}),
])

# 启动一个本地HTTP服务器来处理请求
class RequestHandler(BaseHTTPRequestHandler):
    def do_POST(self):
        content_length = int(self.headers['Content-Length'])
        body = self.rfile.read(content_length)
        
        response = mock.get_response(self.path, method='POST', data=body)
        self.send_response(response['status_code'])
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write(response['content'])

httpd = HTTPServer(('localhost', 8000), RequestHandler)
httpd.handle_request()

# 发起一个POST请求并打印结果
data = b'username=test'
req = urlopen('http://localhost:8000', data=data)
response = req.read().decode()
print(response)

在上面的示例中，我们使用HttpMockSequence()函数创建了一个模拟请求响应。对于请求URL http://example.com，我们指定了请求方法为POST，并且提交了一个username=test的数据。我们还定义了响应的状态码（200）和内容（b'Success!'）。然后，我们启动了一个本地HTTP服务器来处理请求，并在处理请求时调用模拟响应。最后，我们使用urlopen()函数向服务器发起了一个POST请求，并打印了响应的内容。

示例三：模拟多个不同请求

from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.request import urlopen
from httpdomock import HttpMockSequence

# 定义模拟请求响应
mock = HttpMockSequence([
    ({'url': 'http://example.com'}, {'status_code': 200, 'content': b'Hello, World!'}),
    ({'url': 'http://example.com/api', 'method': 'POST', 'data': b'username=test'}, {'status_code': 201, 'content': b'Success!'}),
    ({'url': 'http://example.com/about'}, {'status_code': 404, 'content': b'Page not found!'}),
])

# 启动一个本地HTTP服务器来处理请求
class RequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        response = mock.get_response(self.path)
        self.send_response(response['status_code'])
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write(response['content'])
        
    def do_POST(self):
        content_length = int(self.headers['Content-Length'])
        body = self.rfile.read(content_length)
        
        response = mock.get_response(self.path, method='POST', data=body)
        self.send_response(response['status_code'])
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write(response['content'])

httpd = HTTPServer(('localhost', 8000), RequestHandler)
httpd.handle_request()

# 发起多个不同的请求并打印结果
response1 = urlopen('http://localhost:8000')
response2 = urlopen('http://localhost:8000/api', data=b'username=test')
response3 = urlopen('http://localhost:8000/about')

print(response1.read().decode())
print(response2.read().decode())
print(response3.read().decode())

在上面的示例中，我们使用HttpMockSequence()函数创建了一个模拟请求响应。我们模拟了三个不同的请求，对于请求URL http://example.com，我们指定了响应的状态码（200）和内容（b'Hello, World!'）；对于请求URL http://example.com/api，我们指定了请求方法为POST，并且提交了一个username=test的数据，同时指定了响应的状态码（201）和内容（b'Success!'）；对于请求URL http://example.com/about，我们指定了响应的状态码（404）和内容（b'Page not found!'）。然后，我们启动了一个本地HTTP服务器来处理请求，并在处理请求时调用模拟响应。最后，我们使用urlopen()函数向服务器分别发起了三个不同的请求，并打印了每个响应的内容。

这些示例展示了如何使用HttpMockSequence()函数模拟HTTP请求和响应。你可以根据自己的需求，定制不同的请求和响应，以验证你的代码或者构建和验证网络爬虫。希望以上内容对你有帮助！