mirror of
https://gitee.com/samwaf/SamWaf.git
synced 2026-11-03 03:31:58 +08:00
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package api
|
|
|
|
import (
|
|
"SamWaf/global"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
// 测试用例 ExportExcelApi
|
|
func TestExportExcelApi(t *testing.T) {
|
|
// 创建一个基于 Gin 的引擎
|
|
r := gin.Default()
|
|
|
|
global.GWAF_RELEASE_VERSION = "111"
|
|
r.GET("/samwaf/export", new(WafCommonApi).ExportExcelApi)
|
|
// 创建一个模拟的 HTTP 请求 WafCommonReq 对象
|
|
// 创建一个模拟的 HTTP 请求
|
|
queryParams := url.Values{}
|
|
queryParams.Set("table_name", "hosts")
|
|
|
|
req, err := http.NewRequest(http.MethodGet, "/samwaf/export?"+queryParams.Encode(), nil)
|
|
if err != nil {
|
|
t.Fatalf("无法创建请求：%v", err)
|
|
}
|
|
|
|
// 创建一个响应记录器
|
|
rec := httptest.NewRecorder()
|
|
|
|
// 将模拟的请求发送到测试的 API 路由
|
|
r.ServeHTTP(rec, req)
|
|
|
|
// 检查响应状态码是否为 200
|
|
if rec.Code != http.StatusOK {
|
|
t.Errorf("期望的状态码：%d，实际状态码：%d", http.StatusOK, rec.Code)
|
|
}
|
|
}
|
|
|
|
func TestGetColumnName(t *testing.T) {
|
|
colIdx := 500
|
|
colName := ""
|
|
for colIdx >= 0 {
|
|
colName = fmt.Sprintf("%c", 'A'+colIdx%26) + colName
|
|
colIdx = colIdx/26 - 1
|
|
}
|
|
fmt.Println(colName)
|
|
}
|