mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 22:26:38 +08:00
ryan
75d1fcf345
- 修复 PostgreSQL/SQLite 日志库下首页「来源分布」卡片无数据：RegionCounts 对空 节点 ID 误拼 node_id = '' 恒空条件，改为空节点 ID 表示全节点聚合（对齐 CH 语义）， 并过滤空白归属地 - /access-logs?tab=list 新增状态码筛选：状态码下拉含常用 2xx/3xx/4xx/5xx 选项， 校验 100-599，非法值返回 400；ClickHouse 与 PostgreSQL/SQLite 日志库均支持 - 搜索框折叠：默认仅显示来源 IP 与状态码，节点 ID/访问域名/请求路径折叠进 「更多筛选」
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package observability
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestReadQueryStringArrayAcceptsHostsBracketForm(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
cases := []struct {
|
|
name string
|
|
url string
|
|
want []string
|
|
}{
|
|
{
|
|
name: "axios brackets form",
|
|
url: "/overview?hours=168&hosts%5B%5D=gist.arctel.de",
|
|
want: []string{"gist.arctel.de"},
|
|
},
|
|
{
|
|
name: "repeated hosts keys",
|
|
url: "/overview?hosts=a.example&hosts=b.example",
|
|
want: []string{"a.example", "b.example"},
|
|
},
|
|
{
|
|
name: "single hosts key",
|
|
url: "/overview?hosts=gist.arctel.de",
|
|
want: []string{"gist.arctel.de"},
|
|
},
|
|
{
|
|
name: "empty",
|
|
url: "/overview?hours=24",
|
|
want: nil,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
req, err := http.NewRequest(http.MethodGet, tc.url, nil)
|
|
require.NoError(t, err)
|
|
c.Request = req
|
|
|
|
got := readQueryStringArray(c, "hosts")
|
|
require.Equal(t, tc.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReadAccessLogQueryIncludesStatusCode(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
req, err := http.NewRequest(
|
|
http.MethodGet,
|
|
"/?node_id=n1&remote_addr=1.2.3.4&host=a.example&path=/api&status_code=404&p=2&page_size=50",
|
|
nil,
|
|
)
|
|
require.NoError(t, err)
|
|
c.Request = req
|
|
|
|
got, err := readAccessLogQuery(c)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "n1", got.NodeID)
|
|
require.Equal(t, "1.2.3.4", got.RemoteAddr)
|
|
require.Equal(t, "a.example", got.Host)
|
|
require.Equal(t, "/api", got.Path)
|
|
require.Equal(t, 404, got.StatusCode)
|
|
require.Equal(t, 2, got.Page)
|
|
require.Equal(t, 50, got.PageSize)
|
|
}
|
|
|
|
func TestReadAccessLogQueryRejectsInvalidStatusCode(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
for _, raw := range []string{"abc", "99", "600", "-1"} {
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
req, err := http.NewRequest(http.MethodGet, "/?status_code="+raw, nil)
|
|
require.NoError(t, err)
|
|
c.Request = req
|
|
|
|
_, err = readAccessLogQuery(c)
|
|
require.Error(t, err, "status_code=%s should be rejected", raw)
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
req, err := http.NewRequest(http.MethodGet, "/", nil)
|
|
require.NoError(t, err)
|
|
c.Request = req
|
|
got, err := readAccessLogQuery(c)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, got.StatusCode)
|
|
}
|