mirror of
https://github.com/tnb-labs/panel.git
synced 2026-10-16 15:42:54 +08:00
耗子andClaude Opus 5
fe7cce6201
- 依赖注入换到 libtnb/wire v0.3.0，各层包导出 Module 并逐类型 Export， wire.Struct 改用 Struct[T]()，cleanup 签名改为 func() error - 测试断言换到 libtnb/assert 的 must/check，20 个 testify suite 拆成标准测试函数 - mock 改用 matryer 模板输出到 internal/mocks，wire 走 go.mod tool 指令， mockery 用 go run 调用，避免它的依赖树污染 go.mod - 引入 .golangci.yml 并修完全部告警，darwin 与 linux 双平台 0 issues - 工作流改用 go-version-file、govulncheck 官方 action， wire 校验移入 test.yml，mockery 由自动提交改为 PR 校验 顺带修复：pkg/db 四处漏查 rows.Err()（DatabaseExists 查询出错会被当成库不存在）、 websitestat 负值转 uint64 污染流量统计、pty 终端尺寸超 uint16 截断、 文件权限位超 32 位静默截断、面板与 ACME challenge 服务器缺 ReadHeaderTimeout、 GetDefault 丢失操作人、三处对结构体值恒成立的假断言、两处带空格而失效的 nolint Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package ipdb
|
|
|
|
import (
|
|
"io/fs"
|
|
"testing"
|
|
|
|
"github.com/libtnb/assert/check"
|
|
"github.com/libtnb/assert/must"
|
|
)
|
|
|
|
const testDBPath = "../geoip/ipipfree.ipdb"
|
|
|
|
func newReader(t *testing.T) *Reader {
|
|
t.Helper()
|
|
r, err := Open(testDBPath)
|
|
must.NoError(t, err)
|
|
t.Cleanup(func() { check.NoError(t, r.Close()) })
|
|
return r
|
|
}
|
|
|
|
// mustFind 查询并校验返回值个数与字段数一致——库内容会变，只有这个数量关系恒定
|
|
func mustFind(t *testing.T, r *Reader, ip string) []string {
|
|
t.Helper()
|
|
result, err := r.Find(ip, "CN")
|
|
must.NoError(t, err)
|
|
must.Len(t, result, len(r.Fields()))
|
|
return result
|
|
}
|
|
|
|
func TestOpen_InvalidPath(t *testing.T) {
|
|
_, err := Open("/nonexistent/path.ipdb")
|
|
check.ErrorIs(t, err, fs.ErrNotExist)
|
|
}
|
|
|
|
func TestFields(t *testing.T) {
|
|
fields := newReader(t).Fields()
|
|
check.NotEmpty(t, fields)
|
|
t.Logf("fields: %v", fields)
|
|
}
|
|
|
|
func TestFind_IPv4(t *testing.T) {
|
|
r := newReader(t)
|
|
for _, ip := range []string{"114.114.114.114", "8.8.8.8"} {
|
|
t.Run(ip, func(t *testing.T) {
|
|
result := mustFind(t, r, ip)
|
|
check.NotEmpty(t, result[0], check.Msgf("%s -> %v", ip, result))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFind_IPv6(t *testing.T) {
|
|
r := newReader(t)
|
|
// 库可能不含 IPv6 数据，允许返回错误，但不能 panic
|
|
check.NotPanics(t, func() { _, _ = r.Find("2001:4860:4860::8888", "CN") })
|
|
}
|
|
|
|
func TestFind_InvalidIP(t *testing.T) {
|
|
_, err := newReader(t).Find("not-an-ip", "CN")
|
|
check.ErrorIs(t, err, ErrInvalidIP)
|
|
}
|
|
|
|
func TestFind_InvalidLanguage(t *testing.T) {
|
|
_, err := newReader(t).Find("8.8.8.8", "INVALID")
|
|
check.ErrorIs(t, err, ErrNoLanguage)
|
|
}
|
|
|
|
func TestReload(t *testing.T) {
|
|
r := newReader(t)
|
|
must.NoError(t, r.Reload(testDBPath))
|
|
|
|
check.NotEmpty(t, mustFind(t, r, "114.114.114.114")[0])
|
|
}
|
|
|
|
func TestReload_InvalidPath(t *testing.T) {
|
|
r := newReader(t)
|
|
check.ErrorIs(t, r.Reload("/nonexistent/path.ipdb"), fs.ErrNotExist)
|
|
|
|
// 失败后原数据仍可用
|
|
check.NotEmpty(t, mustFind(t, r, "114.114.114.114")[0])
|
|
}
|
|
|
|
func TestClose_UseAfterClose(t *testing.T) {
|
|
r, err := Open(testDBPath)
|
|
must.NoError(t, err)
|
|
|
|
check.NoError(t, r.Close())
|
|
|
|
_, err = r.Find("8.8.8.8", "CN")
|
|
check.ErrorIs(t, err, ErrClosed)
|
|
}
|
|
|
|
func TestClose_DoubleClose(t *testing.T) {
|
|
r, err := Open(testDBPath)
|
|
must.NoError(t, err)
|
|
|
|
check.NoError(t, r.Close())
|
|
check.NoError(t, r.Close())
|
|
}
|
|
|
|
func TestClose_NilReceiver(t *testing.T) {
|
|
var r *Reader
|
|
check.NoError(t, r.Close())
|
|
}
|
|
|
|
func BenchmarkFind(b *testing.B) {
|
|
r, err := Open(testDBPath)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
defer func(r *Reader) { _ = r.Close() }(r)
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for b.Loop() {
|
|
_, _ = r.Find("114.114.114.114", "CN")
|
|
}
|
|
}
|