#!/bin/bash

# 修复测试文件中的编译错误
cd "$(dirname "$0")"

# 修复未使用变量问题
for file in tests/contract/test_*.go; do
    sed -i 's/req, _ := http\.NewRequest(/req, _ := http.NewRequest(/g' "$file"
    sed -i 's/w := httptest\.NewRecorder()/w := httptest.NewRecorder()/g'
    sed -i '/req, _ := http\.NewRequest(/a\			_ = req // 避免未使用变量错误' "$file"
    sed -i '/w := httptest\.NewRecorder()/a\			_ = w   // 避免未使用变量错误' "$file"
    sed -i '/var response map\[string\]interface{}/a\			_ = response // 避免未使用变量错误' "$file"
done

# 修复未使用导入问题
for file in tests/contract/test_*.go tests/integration/test_*.go; do
    if [ -f "$file" ]; then
        sed -i 's/"encoding\/json"/_ "encoding\/json"/g' "$file"
        sed -i 's/"bytes"/_ "bytes"/g' "$file"
        sed -i 's/"net\/http"/_ "net\/http"/g' "$file"
        sed -i 's/"net\/http\/httptest"/_ "net\/http\/httptest"/g' "$file"
        sed -i 's/"os\/exec"/_ "os\/exec"/g' "$file"
        sed -i 's/"strings"/_ "strings"/g' "$file"
    fi
done

echo "测试文件修复完成"