74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
package config
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoadConfigYAML(t *testing.T) {
|
|
cfg, err := Load("../../configs/config.yaml")
|
|
if err != nil {
|
|
t.Fatalf("Load() error = %v", err)
|
|
}
|
|
|
|
if cfg.HTTPAddr != ":13100" {
|
|
t.Fatalf("HTTPAddr = %q", cfg.HTTPAddr)
|
|
}
|
|
if cfg.MySQLDSN != "hyapp:hyapp@tcp(127.0.0.1:23306)/hyapp_admin?parseTime=true&charset=utf8mb4&loc=UTC" {
|
|
t.Fatalf("MySQLDSN = %q", cfg.MySQLDSN)
|
|
}
|
|
if cfg.UserMySQLDSN != "hyapp:hyapp@tcp(127.0.0.1:23306)/hyapp_user?parseTime=true&charset=utf8mb4&loc=UTC" {
|
|
t.Fatalf("UserMySQLDSN = %q", cfg.UserMySQLDSN)
|
|
}
|
|
if cfg.AccessTokenTTL != 12*time.Hour {
|
|
t.Fatalf("AccessTokenTTL = %s", cfg.AccessTokenTTL)
|
|
}
|
|
if cfg.RefreshTokenTTL != 7*24*time.Hour {
|
|
t.Fatalf("RefreshTokenTTL = %s", cfg.RefreshTokenTTL)
|
|
}
|
|
if !cfg.Migrations.AllowChecksumRepair {
|
|
t.Fatalf("Migrations.AllowChecksumRepair = false, want true for local config")
|
|
}
|
|
if !cfg.TencentCOS.Enabled || cfg.TencentCOS.BucketName != "yumi-assets-1420526837" || cfg.TencentCOS.ObjectPrefix != "admin" {
|
|
t.Fatalf("TencentCOS = %#v", cfg.TencentCOS)
|
|
}
|
|
if !cfg.FinanceNotifications.DingTalk.Enabled {
|
|
t.Fatalf("finance dingtalk notification should be enabled by default")
|
|
}
|
|
if cfg.FinanceNotifications.DingTalk.WebhookURL == "" || cfg.FinanceNotifications.DingTalk.Secret == "" {
|
|
t.Fatalf("finance dingtalk notification config should be populated: %#v", cfg.FinanceNotifications.DingTalk)
|
|
}
|
|
}
|
|
|
|
func TestLoadEmptyPathUsesDefault(t *testing.T) {
|
|
cfg, err := Load("")
|
|
if err != nil {
|
|
t.Fatalf("Load() error = %v", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(cfg, Default()) {
|
|
t.Fatalf("Load(\"\") = %#v, want default", cfg)
|
|
}
|
|
}
|
|
|
|
func TestFinanceDingTalkEnvOverride(t *testing.T) {
|
|
t.Setenv("HYAPP_ADMIN_FINANCE_DINGTALK_WEBHOOK_URL", "https://oapi.dingtalk.com/robot/send?access_token=test")
|
|
t.Setenv("HYAPP_ADMIN_FINANCE_DINGTALK_SECRET", "SEC_TEST")
|
|
t.Setenv("HYAPP_ADMIN_FINANCE_DINGTALK_AT_MOBILES", "13800138000, 13900139000")
|
|
t.Setenv("HYAPP_ADMIN_FINANCE_DINGTALK_AT_ALL", "false")
|
|
|
|
cfg := Default()
|
|
cfg.Normalize()
|
|
|
|
if !cfg.FinanceNotifications.DingTalk.Enabled {
|
|
t.Fatal("DingTalk should be enabled when webhook env is provided")
|
|
}
|
|
if cfg.FinanceNotifications.DingTalk.Secret != "SEC_TEST" {
|
|
t.Fatalf("secret mismatch: %q", cfg.FinanceNotifications.DingTalk.Secret)
|
|
}
|
|
if len(cfg.FinanceNotifications.DingTalk.AtMobiles) != 2 {
|
|
t.Fatalf("at mobiles mismatch: %#v", cfg.FinanceNotifications.DingTalk.AtMobiles)
|
|
}
|
|
}
|