mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-03 23:06:36 +08:00
ryan
2f60329886
代码质量全量清理，零行为变化：golangci 扩展集 13 类 linter（gosec/modernize/perfsprint/canonicalheader/usestdlibvars/wastedassign/intrange/errorlint/forcetypeassert/recvcheck/exhaustive/unparam）全量修复，测试代码质量（testifylint/thelper/usetesting）25→0，frpc 进程生命周期真 bug（进程组击杀）、全仓 go test -race 6 类数据竞争（含 1 个生产竞争）、SPDX license 头补齐 131 文件、前端测试套件 next-intl 迁移后 44 失败→全绿、过期 swagger 文档重新生成、pnpm-workspace 构建审批。 Experiments: #2-#17, #18, #20, #21, #23 Metric: total_issues 108 → 8 (-92.6%)
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package geoip
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
db "github.com/Rain-kl/Wavelet/internal/infra/persistence"
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestEnsureRuntimeProviderInitializesConfiguredProvider(t *testing.T) {
|
|
sqliteDB, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite: %v", err)
|
|
}
|
|
if err := sqliteDB.AutoMigrate(&model.SystemConfig{}); err != nil {
|
|
t.Fatalf("migrate: %v", err)
|
|
}
|
|
db.SetDB(sqliteDB)
|
|
t.Cleanup(func() {
|
|
db.SetDB(nil)
|
|
ResetRuntimeForTest()
|
|
})
|
|
|
|
ctx := context.Background()
|
|
ResetRuntimeForTest()
|
|
// 通过 SystemConfig 设置 GeoIPProvider 配置
|
|
if err := db.DB(ctx).Create(&model.SystemConfig{
|
|
Key: model.ConfigKeyGeoIPProvider,
|
|
Value: pkggeoip.ProviderIPInfo,
|
|
Type: "business",
|
|
Visibility: 0,
|
|
}).Error; err != nil {
|
|
t.Fatalf("create system config: %v", err)
|
|
}
|
|
|
|
if err := EnsureRuntimeProvider(ctx); err != nil {
|
|
t.Fatalf("EnsureRuntimeProvider error = %v", err)
|
|
}
|
|
if pkggeoip.CurrentProvider == nil || pkggeoip.CurrentProvider.Name() == "EmptyProvider" {
|
|
t.Fatalf("expected ipinfo provider, got %#v", pkggeoip.CurrentProvider)
|
|
}
|
|
}
|