mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 22:26:38 +08:00
ryan
498a9ed3ff
- 启动时按 of_options.GeoIPProvider 初始化 pkg/geoip（bootstrap + runtime） - mmdb 模式从内置 GeoLite2 种子到 data/；保存归属方式后热刷新 Provider - Agent/Relay 心跳在服务端根据 IP 写入 geo 字段，尊重 geo_manual_override - ipinfo 归属名称改为 City, Region, Country 可读格式
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package geoip
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
"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.OpenFlareOption{}); err != nil {
|
|
t.Fatalf("migrate: %v", err)
|
|
}
|
|
db.SetDB(sqliteDB)
|
|
t.Cleanup(func() {
|
|
db.SetDB(nil)
|
|
model.ResetOptionMapForTest()
|
|
ResetRuntimeForTest()
|
|
})
|
|
|
|
ctx := context.Background()
|
|
model.ResetOptionMapForTest()
|
|
ResetRuntimeForTest()
|
|
if err := model.UpdateOpenFlareOption(ctx, "GeoIPProvider", pkggeoip.ProviderIPInfo); err != nil {
|
|
t.Fatalf("update option: %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)
|
|
}
|
|
}
|