使用Hoverfly进行Golang HTTP(S)代理录制与API模拟

Hoverfly是一个轻量级的服务虚拟化工具，可以用于录制和模拟HTTP(S)流量，特别适合REST/SOAP API的测试场景。下面我将介绍如何在Golang中使用Hoverfly。

Hoverfly核心功能

1. 流量录制：捕获真实API请求和响应
2. 流量模拟：根据录制的数据模拟API行为
3. 中间人代理：支持HTTPS流量拦截
4. 行为定制：可以修改响应延迟、错误率等

安装Hoverfly

首先需要下载Hoverfly二进制文件：

# Linux/macOS
curl -L https://github.com/SpectoLabs/hoverfly/releases/download/v1.6.0/hoverfly_bundle_OSX.zip -o hoverfly.zip
unzip hoverfly.zip
chmod +x hoverfly

# Windows
# 下载 https://github.com/SpectoLabs/hoverfly/releases

Golang中使用Hoverfly

1. 启动Hoverfly作为代理

package main

import (
	"fmt"
	"os/exec"
	"time"
)

func startHoverfly(adminPort, proxyPort int) (*exec.Cmd, error) {
	cmd := exec.Command(
		"./hoverfly",
		"-pp", fmt.Sprintf("%d", proxyPort), // 代理端口
		"-ap", fmt.Sprintf("%d", adminPort), // 管理端口
		"-db", "memory",
	)
	
	err := cmd.Start()
	if err != nil {
		return nil, err
	}
	
	// 等待Hoverfly启动
	time.Sleep(2 * time.Second)
	return cmd, nil
}

func main() {
	hoverfly, err := startHoverfly(8888, 8500)
	if err != nil {
		panic(err)
	}
	defer hoverfly.Process.Kill()
	
	fmt.Println("Hoverfly started on port 8500")
	// 在这里进行你的测试...
}

2. 录制API流量

import (
	"net/http"
	"net/url"
)

func recordMode(adminPort int) error {
	adminURL := fmt.Sprintf("http://localhost:%d/api/v2/hoverfly/mode", adminPort)
	
	req, err := http.NewRequest("PUT", adminURL, bytes.NewBufferString(`{"mode":"capture"}`))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json")
	
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	
	if resp.StatusCode != 200 {
		return fmt.Errorf("failed to set capture mode")
	}
	
	return nil
}

3. 使用代理发送请求

func makeRequestThroughProxy(targetURL string, proxyPort int) (string, error) {
	proxyURL, _ := url.Parse(fmt.Sprintf("http://localhost:%d", proxyPort))
	
	client := &http.Client{
		Transport: &http.Transport{
			Proxy: http.ProxyURL(proxyURL),
		},
	}
	
	resp, err := client.Get(targetURL)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()
	
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}
	
	return string(body), nil
}

4. 切换到模拟模式并使用录制的数据

func simulateMode(adminPort int) error {
	adminURL := fmt.Sprintf("http://localhost:%d/api/v2/hoverfly/mode", adminPort)
	
	req, err := http.NewRequest("PUT", adminURL, bytes.NewBufferString(`{"mode":"simulate"}`))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json")
	
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	
	if resp.StatusCode != 200 {
		return fmt.Errorf("failed to set simulate mode")
	}
	
	return nil
}

5. 导出和导入模拟数据

func exportSimulationData(adminPort int, filePath string) error {
	exportURL := fmt.Sprintf("http://localhost:%d/api/v2/simulation", adminPort)
	
	resp, err := http.Get(exportURL)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	
	data, err := io.ReadAll(resp.Body)
	if err != nil {
		return err
	}
	
	return os.WriteFile(filePath, data, 0644)
}

func importSimulationData(adminPort int, filePath string) error {
	data, err := os.ReadFile(filePath)
	if err != nil {
		return err
	}
	
	importURL := fmt.Sprintf("http://localhost:%d/api/v2/simulation", adminPort)
	req, err := http.NewRequest("PUT", importURL, bytes.NewBuffer(data))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json")
	
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	
	if resp.StatusCode != 200 {
		return fmt.Errorf("failed to import simulation data")
	}
	
	return nil
}

完整示例工作流

func main() {
	// 1. 启动Hoverfly
	hoverfly, err := startHoverfly(8888, 8500)
	if err != nil {
		panic(err)
	}
	defer hoverfly.Process.Kill()
	
	// 2. 设置为录制模式
	err = recordMode(8888)
	if err != nil {
		panic(err)
	}
	
	// 3. 通过代理发送请求(会被录制)
	response, err := makeRequestThroughProxy("https://api.example.com/data", 8500)
	if err != nil {
		panic(err)
	}
	fmt.Println("Recorded response:", response)
	
	// 4. 导出录制的数据
	err = exportSimulationData(8888, "simulation.json")
	if err != nil {
		panic(err)
	}
	
	// 5. 切换到模拟模式
	err = simulateMode(8888)
	if err != nil {
		panic(err)
	}
	
	// 6. 现在请求会返回录制的响应
	simulatedResponse, err := makeRequestThroughProxy("https://api.example.com/data", 8500)
	if err != nil {
		panic(err)
	}
	fmt.Println("Simulated response:", simulatedResponse)
}

高级功能

1. 延迟模拟：可以配置响应延迟
2. 故障注入：模拟错误响应
3. 请求修改：修改请求/响应头或体
4. 条件匹配：基于请求头、体等条件返回不同响应

Hoverfly提供了强大的API测试能力，特别适合以下场景：

• 开发时模拟依赖服务
• 自动化测试中的服务虚拟化
• 性能测试中的服务隔离
• API行为研究和分析

通过Golang与Hoverfly的结合，你可以构建强大的API测试和模拟框架。