mrlee
278f8267b7
- 公网反代：Nginx 状态卡片（版本 / 运行状态 / 配置校验 / 端口监听） - 部署设置第一、二梯队（HTTPS 开关、安全头、限流、连接数、仅中国大陆 IP） - 隧道检测兼容 frps v0.68 / v0.71（V2 与旧版端点） - 恢复被清空的 internal/frp/store.go、internal/api/ddns_handler.go、web/src/views/DdnsView.vue
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/fonu/fonu/internal/db"
|
|
"github.com/fonu/fonu/internal/secret"
|
|
"github.com/fonu/fonu/internal/settings"
|
|
)
|
|
|
|
func TestStoreSaveTelegramPersistsType(t *testing.T) {
|
|
dir := t.TempDir()
|
|
conn, err := db.Open(filepath.Join(dir, "fonu.db"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer conn.Close()
|
|
if err := db.Migrate(context.Background(), conn, filepath.Join("..", "..", "migrations")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
settingsStore := settings.NewStore(conn)
|
|
box, err := secret.NewBox("test-secret-key-32bytes-long!!!")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store := NewStore(settingsStore, box)
|
|
if err := store.Save(context.Background(), SaveInput{
|
|
Type: NotifyTypeTelegram,
|
|
Telegram: TelegramConfig{
|
|
ChatID: "743955235",
|
|
ProxyURL: "http://127.0.0.1:7890",
|
|
HasBotToken: true,
|
|
},
|
|
TelegramToken: "123456:ABC-DEF",
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rawType, err := settingsStore.Get(context.Background(), settings.KeyNotifyType)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if rawType != "telegram" {
|
|
t.Fatalf("notify_type=%q, want telegram", rawType)
|
|
}
|
|
cfg, err := store.Load(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.Type != NotifyTypeTelegram || cfg.Telegram.ChatID != "743955235" {
|
|
t.Fatalf("unexpected loaded config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestStoreMigratesLegacyWebhookURL(t *testing.T) {
|
|
dir := t.TempDir()
|
|
conn, err := db.Open(filepath.Join(dir, "fonu.db"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer conn.Close()
|
|
if err := db.Migrate(context.Background(), conn, filepath.Join("..", "..", "migrations")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
settingsStore := settings.NewStore(conn)
|
|
if err := settingsStore.Set(context.Background(), settings.KeyNotifyWebhookURL, "https://example.com/hook"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
box, err := secret.NewBox("test-secret-key-32bytes-long!!!")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store := NewStore(settingsStore, box)
|
|
cfg, err := store.Load(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.Type != NotifyTypeWebhook || cfg.Webhook.Provider != WebhookCustom || cfg.Webhook.URL != "https://example.com/hook" {
|
|
t.Fatalf("unexpected migrated config: %+v", cfg)
|
|
}
|
|
}
|