53 lines
1.6 KiB
Go
53 lines
1.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func TestResolveVisitorIPAndRegion(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
// 内网 IP
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request = httptest.NewRequest("GET", "/", nil)
|
|
c.Request.RemoteAddr = "192.168.1.105:12345"
|
|
ip := resolveVisitorIP(c)
|
|
if ip != "192.168.1.105" {
|
|
t.Fatalf("内网 IP 解析错误: %s", ip)
|
|
}
|
|
if region := resolveVisitorRegion(c, ip); region != "内网" {
|
|
t.Fatalf("内网地区应为内网, got %s", region)
|
|
}
|
|
|
|
// Caddy -> Nginx 链路：X-Real-IP 是上一跳代理，XFF 左侧才是访客。
|
|
c3, _ := gin.CreateTestContext(httptest.NewRecorder())
|
|
c3.Request = httptest.NewRequest("GET", "/", nil)
|
|
c3.Request.RemoteAddr = "172.18.0.3:42100"
|
|
c3.Request.Header.Set("X-Real-IP", "172.18.0.1")
|
|
c3.Request.Header.Set("X-Forwarded-For", "203.0.113.24, 172.18.0.1")
|
|
if ip3 := resolveVisitorIP(c3); ip3 != "203.0.113.24" {
|
|
t.Fatalf("代理链 IP 解析错误: %s", ip3)
|
|
}
|
|
|
|
// CF 国家头
|
|
c2, _ := gin.CreateTestContext(httptest.NewRecorder())
|
|
c2.Request = httptest.NewRequest("GET", "/", nil)
|
|
c2.Request.Header.Set("CF-Connecting-IP", "8.8.8.8")
|
|
c2.Request.Header.Set("CF-IPCountry", "US")
|
|
ip2 := resolveVisitorIP(c2)
|
|
if ip2 != "8.8.8.8" {
|
|
t.Fatalf("CF IP 解析错误: %s", ip2)
|
|
}
|
|
if region := resolveVisitorRegion(c2, ip2); region != "美国" {
|
|
t.Fatalf("期望美国, got %s", region)
|
|
}
|
|
|
|
if summarizeUserAgent("Mozilla/5.0 (Macintosh; Intel Mac OS X) Chrome/120.0.0.0") != "Chrome / macOS" {
|
|
t.Fatalf("UA 摘要错误")
|
|
}
|
|
}
|