Files
email_agent/internal/config/config_test.go
Builder e5cce1c93a Phase 7 follow-up: 1-month body retention and actions_log cap (AUD-75)
Implements the AGENTS.md §7 data rule: keep message bodies for one month, and
size-cap the append-only action log.

- internal/store/retention.go: Store.Retain clears messages.body_text older
  than the window (metadata stays) and deletes actions_log rows beyond the
  newest N, in one transaction.
- internal/config: EMAIL_AGENT_RETENTION_DAYS (30) and
  EMAIL_AGENT_ACTIONS_LOG_MAX_ROWS (50000).
- cmd/email-agent: bounded -retain flag for the n8n schedule.
- Tests for body clearing, metadata retention, log cap, and disabled policy.
2026-10-13 10:41:01 +11:00

194 lines
5.6 KiB
Go

package config
import "testing"
func TestLoadDefaults(t *testing.T) {
// Ensure no ambient values leak into the test.
for _, k := range []string{
"EMAIL_AGENT_ENV", "EMAIL_AGENT_ADDR", "EMAIL_AGENT_PORT",
"EMAIL_AGENT_DB_PATH", "EMAIL_AGENT_LOG_LEVEL", "EMAIL_AGENT_PUBLIC_BASE_URL",
"GMAIL_USER", "GMAIL_APP_PASSWORD", "EMAIL_AGENT_IMAP_HOST", "EMAIL_AGENT_IMAP_PORT",
"EMAIL_AGENT_IMAP_LOOKBACK_DAYS", "EMAIL_AGENT_IMAP_MAX_MESSAGES",
"EMAIL_AGENT_RETENTION_DAYS", "EMAIL_AGENT_ACTIONS_LOG_MAX_ROWS",
} {
t.Setenv(k, "")
}
cfg, err := Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if cfg.Port != DefaultPort {
t.Errorf("Port = %d, want %d", cfg.Port, DefaultPort)
}
if cfg.Addr != DefaultAddr {
t.Errorf("Addr = %q, want %q", cfg.Addr, DefaultAddr)
}
if cfg.Env != DefaultEnv {
t.Errorf("Env = %q, want %q", cfg.Env, DefaultEnv)
}
if cfg.DBPath != DefaultDB {
t.Errorf("DBPath = %q, want %q", cfg.DBPath, DefaultDB)
}
if cfg.IMAPHost != DefaultIMAPHost {
t.Errorf("IMAPHost = %q, want %q", cfg.IMAPHost, DefaultIMAPHost)
}
if cfg.IMAPPort != DefaultIMAPPort {
t.Errorf("IMAPPort = %d, want %d", cfg.IMAPPort, DefaultIMAPPort)
}
if cfg.IMAPLookbackDays != DefaultIMAPLookbackDays {
t.Errorf("IMAPLookbackDays = %d, want %d", cfg.IMAPLookbackDays, DefaultIMAPLookbackDays)
}
if cfg.IMAPMaxMessages != DefaultIMAPMaxMessages {
t.Errorf("IMAPMaxMessages = %d, want %d", cfg.IMAPMaxMessages, DefaultIMAPMaxMessages)
}
if cfg.GmailConfigured() {
t.Error("GmailConfigured() = true with no credentials")
}
}
func TestRetentionDefaultsAndOverrides(t *testing.T) {
t.Setenv("EMAIL_AGENT_RETENTION_DAYS", "")
t.Setenv("EMAIL_AGENT_ACTIONS_LOG_MAX_ROWS", "")
cfg, err := Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if cfg.RetentionDays != DefaultRetentionDays {
t.Errorf("RetentionDays = %d, want %d", cfg.RetentionDays, DefaultRetentionDays)
}
if cfg.ActionsLogMaxRows != DefaultActionsLogMaxRows {
t.Errorf("ActionsLogMaxRows = %d, want %d", cfg.ActionsLogMaxRows, DefaultActionsLogMaxRows)
}
t.Setenv("EMAIL_AGENT_RETENTION_DAYS", "45")
t.Setenv("EMAIL_AGENT_ACTIONS_LOG_MAX_ROWS", "1234")
cfg, err = Load()
if err != nil {
t.Fatalf("Load() override error = %v", err)
}
if cfg.RetentionDays != 45 {
t.Errorf("RetentionDays = %d, want 45", cfg.RetentionDays)
}
if cfg.ActionsLogMaxRows != 1234 {
t.Errorf("ActionsLogMaxRows = %d, want 1234", cfg.ActionsLogMaxRows)
}
}
func TestGmailConfigured(t *testing.T) {
t.Setenv("GMAIL_USER", "samuelrolfe@gmail.com")
t.Setenv("GMAIL_APP_PASSWORD", "app-password")
cfg, err := Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if !cfg.GmailConfigured() {
t.Error("GmailConfigured() = false with credentials set")
}
t.Setenv("GMAIL_APP_PASSWORD", "")
cfg, err = Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if cfg.GmailConfigured() {
t.Error("GmailConfigured() = true with no app password")
}
}
func TestLoadOverrides(t *testing.T) {
t.Setenv("EMAIL_AGENT_ENV", "production")
t.Setenv("EMAIL_AGENT_PORT", "9000")
t.Setenv("EMAIL_AGENT_DB_PATH", "/tmp/custom.db")
t.Setenv("EMAIL_AGENT_LOG_LEVEL", "debug")
t.Setenv("EMAIL_AGENT_PUBLIC_BASE_URL", "https://example.test/")
t.Setenv("EMAIL_AGENT_ADDR", "")
cfg, err := Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if cfg.Port != 9000 {
t.Errorf("Port = %d, want 9000", cfg.Port)
}
if cfg.Addr != ":9000" {
t.Errorf("Addr = %q, want %q", cfg.Addr, ":9000")
}
if !cfg.IsProduction() {
t.Errorf("IsProduction() = false, want true")
}
if cfg.PublicBaseURL != "https://example.test" {
t.Errorf("PublicBaseURL = %q, want trimmed URL", cfg.PublicBaseURL)
}
}
func TestLoadInvalidPort(t *testing.T) {
t.Setenv("EMAIL_AGENT_PORT", "not-a-number")
t.Setenv("EMAIL_AGENT_ADDR", "")
if _, err := Load(); err == nil {
t.Fatal("Load() error = nil, want error for invalid port")
}
}
func TestLoadInvalidEnv(t *testing.T) {
t.Setenv("EMAIL_AGENT_ENV", "staging")
t.Setenv("EMAIL_AGENT_ADDR", "")
if _, err := Load(); err == nil {
t.Fatal("Load() error = nil, want error for invalid env")
}
}
func TestHubAndAppriseConfig(t *testing.T) {
for _, k := range []string{"HUB_TOKEN", "HUB_BASE_URL", "APPRISE_URL", "APPRISE_KEY", "APPRISE_URLS", "EMAIL_AGENT_SUMMARY_TO", "HUB_NOTIFY_TIMEOUT"} {
t.Setenv(k, "")
}
cfg, err := Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if cfg.HubConfigured() {
t.Error("HubConfigured() = true with no HUB_TOKEN")
}
if cfg.HubBaseURL != DefaultHubBaseURL {
t.Errorf("HubBaseURL = %q, want %q", cfg.HubBaseURL, DefaultHubBaseURL)
}
if cfg.AppriseURL != DefaultAppriseURL {
t.Errorf("AppriseURL = %q, want %q", cfg.AppriseURL, DefaultAppriseURL)
}
if cfg.SummaryTo != DefaultSummaryTo {
t.Errorf("SummaryTo = %q, want %q", cfg.SummaryTo, DefaultSummaryTo)
}
if cfg.AppriseConfigured() {
t.Error("AppriseConfigured() = true with no key or URLs")
}
t.Setenv("HUB_TOKEN", "hub-secret")
t.Setenv("APPRISE_KEY", "email-summary")
cfg, err = Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if !cfg.HubConfigured() {
t.Error("HubConfigured() = false with HUB_TOKEN set")
}
if !cfg.AppriseConfigured() {
t.Error("AppriseConfigured() = false with a key set")
}
// A stateless URL list also configures Apprise.
t.Setenv("APPRISE_KEY", "")
t.Setenv("APPRISE_URLS", "mailto://a:b@example.com, ntfy://topic")
cfg, err = Load()
if err != nil {
t.Fatalf("Load() error = %v", err)
}
if !cfg.AppriseConfigured() {
t.Error("AppriseConfigured() = false with APPRISE_URLS set")
}
if len(cfg.AppriseURLs) != 2 {
t.Errorf("AppriseURLs = %v, want 2 entries", cfg.AppriseURLs)
}
}