golib/net/formmailer/run_test.go

package formmailer

import (
	"context"
	"io"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"
)

// captureStdout runs fn while capturing everything written to os.Stdout,
// returning the captured text.
func captureStdout(t *testing.T, fn func()) string {
	t.Helper()
	orig := os.Stdout
	r, w, err := os.Pipe()
	if err != nil {
		t.Fatalf("pipe: %v", err)
	}
	os.Stdout = w
	done := make(chan string, 1)
	go func() {
		b, _ := io.ReadAll(r)
		done <- string(b)
	}()
	fn()
	_ = w.Close()
	os.Stdout = orig
	return <-done
}

// runOnce starts a FormMailer with a context that expires quickly so Run
// returns immediately after selecting and logging its GeoIP mode. It must not
// require network (BlocklistRepo is empty) and binds an ephemeral port.
func runOnce(t *testing.T, fm *FormMailer) string {
	t.Helper()
	fm.ListenAddr = "127.0.0.1:0"
	fm.RefreshInterval = time.Minute
	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	return captureStdout(t, func() {
		_ = fm.Run(ctx)
	})
}

// TestRunGeoModeSelection checks that formmailer only polls existing
// archives; downloading is owned by the external geoip-update command.
func TestRunGeoModeSelection(t *testing.T) {
	t.Run("poll", func(t *testing.T) {
		home := t.TempDir()
		t.Setenv("HOME", home)

		archive := filepath.Join(t.TempDir(), "maxmind")
		if err := os.MkdirAll(archive, 0o755); err != nil {
			t.Fatal(err)
		}
		// Any entry makes the dir "populated" -> poll mode.
		if err := os.WriteFile(filepath.Join(archive, "keep"), []byte("x"), 0o600); err != nil {
			t.Fatal(err)
		}

		out := runOnce(t, &FormMailer{
			CacheDir: home,
			GeoIPDir: archive,
			SMTPHost: "localhost:25",
			SMTPFrom: "a@example.com",
			SMTPTo:   []string{"b@example.com"},
		})
		if !strings.Contains(out, "GeoIP: poll mode") {
			t.Fatalf("missing poll mode line; stdout:\n%s", out)
		}
	})

	t.Run("disabled", func(t *testing.T) {
		home := t.TempDir()
		t.Setenv("HOME", home)
		empty := filepath.Join(home, "maxmind")
		if err := os.MkdirAll(empty, 0o755); err != nil {
			t.Fatal(err)
		}

		out := runOnce(t, &FormMailer{
			CacheDir: home,
			GeoIPDir: empty,
			SMTPHost: "localhost:25",
			SMTPFrom: "a@example.com",
			SMTPTo:   []string{"b@example.com"},
		})
		if strings.Contains(out, "GeoIP:") {
			t.Fatalf("expected GeoIP disabled; stdout:\n%s", out)
		}
	})
}