Python网络测试入门：使用BaseHTTPServer的test()函数

Python的BaseHTTPServer模块是Python标准库中提供的一个HTTP服务器基类，可以用于创建自定义的HTTP服务器。其中，test()函数提供了一个简单的命令行接口，用于启动一个简单的HTTP服务器进行网络测试。

test()函数的基本用法如下：

from BaseHTTPServer import test
test([HandlerClass[, ServerClass[, port]]])

传入test()函数的参数可以分为三个部分：HandlerClass、ServerClass和port。其中，HandlerClass是处理客户端请求的类，默认为SimpleHTTPRequestHandler。ServerClass是HTTP服务器类，默认为BaseHTTPServer.HTTPServer。port是服务器监听的端口，默认为8000。

下面我们来看一个使用BaseHTTPServer的test()函数的例子：

from BaseHTTPServer import test, SimpleHTTPRequestHandler

class MyHandler(SimpleHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write('Hello, world!')
        return

test(MyHandler)

在这个例子中，我们自定义了一个处理请求的类MyHandler，继承自SimpleHTTPRequestHandler，并重写了do_GET()方法。在do_GET()方法中，我们发送了一个HTTP响应，内容为'Hello, world!'。

然后，我们通过调用test()函数，并传入自定义的HandlerClass参数MyHandler来启动HTTP服务器。这时，服务器会监听在默认的端口8000上。

要启动这个HTTP服务器，我们可以将以上代码保存为一个Python脚本，比如server.py，并在命令行中运行以下命令：

python server.py

然后，我们可以用浏览器或者curl命令来访问http://localhost:8000，就可以看到返回的'Hello, world!'了。

除了自定义HandlerClass，我们还可以使用其他的ServerClass和port参数来启动不同的HTTP服务器。比如，我们可以使用HTTPServer来启动一个支持SSL的HTTPS服务器：

from BaseHTTPServer import test, HTTPServer
from OpenSSL import SSL

class MyHandler(SimpleHTTPRequestHandler):
  def do_GET(self):
    self.send_response(200)
    self.send_header('Content-type', 'text/html')
    self.end_headers()
    self.wfile.write('Hello, world!')
    return

httpd = HTTPServer(('localhost', 8000), MyHandler)
httpd.socket = SSL.Context(SSL.SSLv23_METHOD)
httpd.socket.use_privatekey_file('path/to/private.key')
httpd.socket.use_certificate_file('path/to/certificate.crt')

test(HandlerClass=None, ServerClass=httpd)

在这个例子中，我们使用了OpenSSL库来提供SSL的支持。首先，我们创建了一个HTTPServer实例httpd，并绑定到地址('localhost', 8000)上。然后，我们使用SSL.Context()方法创建了一个SSL上下文，指定使用SSLv23_METHOD方法。接下来，我们使用use_privatekey_file()和use_certificate_file()方法加载私钥和证书，以启用HTTPS。最后，我们通过调用test()函数，并传入ServerClass参数httpd来启动HTTPS服务器。

这样，在运行server.py脚本之后，我们就可以通过https://localhost:8000来访问HTTPS服务器了。

总结起来，使用BaseHTTPServer的test()函数可以快速创建简单的HTTP服务器进行网络测试。通过自定义HandlerClass、ServerClass和port参数，我们可以创建各种不同的HTTP服务器，满足不同的测试需求。