golang动态生成自签名证书和证书颁发机构的测试插件库testcerts的使用

testcerts是一个用于在测试中动态生成自签名证书和证书颁发机构的Golang库，可以避免在代码仓库中保存测试证书。

基本用法

func TestFunc(t *testing.T) {
    // 创建自签名证书和密钥并写入临时文件
    cert, key, err := testcerts.GenerateToTempFile("/tmp/")
    if err != nil {
        // 处理错误
    }
    defer os.Remove(key)
    defer os.Remove(cert)

    // 使用测试证书启动HTTP监听器
    err = http.ListenAndServeTLS("127.0.0.1:443", cert, key, someHandler)
    if err != nil {
        // 处理错误
    }
}

高级用法：创建证书颁发机构

对于更复杂的测试，可以使用该包创建证书颁发机构(CA)，并为任何测试域创建由该CA签名的密钥对。

func TestFunc(t *testing.T) {
    // 生成证书颁发机构
    ca := testcerts.NewCA()

    go func() {
        // 为"localhost"创建签名的证书和密钥
        certs, err := ca.NewKeyPair("localhost")
        if err != nil {
            // 处理错误
        }

        // 将证书写入文件
        err = certs.ToFile("/tmp/cert", "/tmp/key")
        if err != nil {
            // 处理错误
        }

        // 启动HTTP监听器
        err = http.ListenAndServeTLS("localhost:443", "/tmp/cert", "/tmp/key", someHandler)
        if err != nil {
            // 处理错误
        }
    }()

    // 使用自签名CA创建客户端
    client := &http.Client{
        Transport: &http.Transport{
            TLSClientConfig: certs.ConfigureTLSConfig(ca.GenerateTLSConfig()),
        },
    }

    // 发起HTTPS请求
    r, _ := client.Get("https://localhost")
}

完整示例

下面是一个完整的测试示例，展示了如何使用testcerts创建CA并测试HTTPS服务：

package main

import (
    "net/http"
    "os"
    "testing"
    
    "github.com/madflojo/testcerts"
)

func TestHTTPS(t *testing.T) {
    // 1. 创建证书颁发机构
    ca := testcerts.NewCA()
    
    // 2. 为测试域名创建证书
    certs, err := ca.NewKeyPair("example.com")
    if err != nil {
        t.Fatalf("Failed to create key pair: %v", err)
    }
    
    // 3. 将证书写入临时文件
    certFile := "/tmp/test-cert.pem"
    keyFile := "/tmp/test-key.pem"
    defer os.Remove(certFile)
    defer os.Remove(keyFile)
    
    if err := certs.ToFile(certFile, keyFile); err != nil {
        t.Fatalf("Failed to write cert files: %v", err)
    }
    
    // 4. 启动测试HTTPS服务器
    go func() {
        mux := http.NewServeMux()
        mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
            w.Write([]byte("Hello, TLS!"))
        })
        
        if err := http.ListenAndServeTLS(":8443", certFile, keyFile, mux); err != nil {
            t.Logf("Server error: %v", err)
        }
    }()
    
    // 5. 创建信任CA的HTTP客户端
    client := &http.Client{
        Transport: &http.Transport{
            TLSClientConfig: certs.ConfigureTLSConfig(ca.GenerateTLSConfig()),
        },
    }
    
    // 6. 测试HTTPS请求
    resp, err := client.Get("https://example.com:8443")
    if err != nil {
        t.Fatalf("HTTPS request failed: %v", err)
    }
    defer resp.Body.Close()
    
    // 验证响应
    if resp.StatusCode != http.StatusOK {
        t.Errorf("Expected status 200, got %d", resp.StatusCode)
    }
}

贡献

如果发现bug或有功能建议，请提交issue或pull request。

许可证

testcerts使用MIT许可证发布。