63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
package db
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/jackc/pgx/v5/pgxpool"
|
||
)
|
||
|
||
// applyPoolEnv читает env: пустые переменные не трогают дефолты pgxpool,
|
||
// заполненные — применяются, кривые — ошибка на старте.
|
||
func TestApplyPoolEnv(t *testing.T) {
|
||
// Дефолт: ни одной переменной — конфиг не меняется.
|
||
cfg := &pgxpool.Config{}
|
||
before := *cfg
|
||
if err := applyPoolEnv(cfg); err != nil {
|
||
t.Fatalf("empty env: %v", err)
|
||
}
|
||
if cfg.MaxConns != before.MaxConns || cfg.MinConns != before.MinConns {
|
||
t.Fatalf("empty env must not touch config: %+v", cfg)
|
||
}
|
||
|
||
// Заданные значения применяются как есть.
|
||
t.Setenv("DB_MAX_CONNS", "20")
|
||
t.Setenv("DB_MIN_CONNS", "2")
|
||
t.Setenv("DB_MAX_CONN_LIFETIME", "1h")
|
||
t.Setenv("DB_MAX_CONN_IDLE_TIME", "30m")
|
||
cfg = &pgxpool.Config{}
|
||
if err := applyPoolEnv(cfg); err != nil {
|
||
t.Fatalf("valid env: %v", err)
|
||
}
|
||
if cfg.MaxConns != 20 || cfg.MinConns != 2 {
|
||
t.Fatalf("conns not applied: max=%d min=%d", cfg.MaxConns, cfg.MinConns)
|
||
}
|
||
if cfg.MaxConnLifetime.Hours() != 1 {
|
||
t.Fatalf("lifetime not applied: %v", cfg.MaxConnLifetime)
|
||
}
|
||
if cfg.MaxConnIdleTime.Minutes() != 30 {
|
||
t.Fatalf("idle time not applied: %v", cfg.MaxConnIdleTime)
|
||
}
|
||
}
|
||
|
||
// Кривые значения — фейл-фаст, а не молчаливый дефолт.
|
||
func TestApplyPoolEnvInvalid(t *testing.T) {
|
||
cases := []struct {
|
||
env string
|
||
val string
|
||
}{
|
||
{"DB_MAX_CONNS", "ten"},
|
||
{"DB_MAX_CONNS", "0"},
|
||
{"DB_MIN_CONNS", "-1"},
|
||
{"DB_MAX_CONN_LIFETIME", "1h30"},
|
||
{"DB_MAX_CONN_IDLE_TIME", "0s"},
|
||
}
|
||
for _, c := range cases {
|
||
t.Run(c.env+"="+c.val, func(t *testing.T) {
|
||
t.Setenv(c.env, c.val)
|
||
if err := applyPoolEnv(&pgxpool.Config{}); err == nil {
|
||
t.Fatalf("%s=%s must fail", c.env, c.val)
|
||
}
|
||
})
|
||
}
|
||
}
|