nexarch/cmd/account-api/dbstart_test.go

package main

import (
	"context"
	"errors"
	"testing"
	"time"

	"github.com/jackc/pgx/v5/pgconn"
)

type fakePinger struct {
	errs  []error
	calls int
}

func (f *fakePinger) Ping(context.Context) error {
	i := f.calls
	f.calls++
	if i < len(f.errs) {
		return f.errs[i]
	}
	return nil
}

func TestWaitForDB_PermanentAuthErrorStopsImmediately(t *testing.T) {
	p := &fakePinger{errs: []error{&pgconn.PgError{Code: "28P01"}, nil}}
	sleeps := 0
	err := waitForDB(context.Background(), p, time.Minute, func(time.Duration) { sleeps++ })
	if err == nil || !isPermanentDBError(err) {
		t.Fatalf("erwartet dauerhaften fehler, war %v", err)
	}
	if p.calls != 1 || sleeps != 0 {
		t.Fatalf("kein retry erwartet, calls=%d sleeps=%d", p.calls, sleeps)
	}
}

func TestWaitForDB_TransientErrorRetriesWithBackoff(t *testing.T) {
	p := &fakePinger{errs: []error{errors.New("connection refused"), errors.New("connection refused")}}
	var delays []time.Duration
	if err := waitForDB(context.Background(), p, time.Minute, func(d time.Duration) { delays = append(delays, d) }); err != nil {
		t.Fatal(err)
	}
	if p.calls != 3 || len(delays) != 2 || delays[1] <= delays[0] {
		t.Fatalf("calls=%d delays=%v", p.calls, delays)
	}
}

func TestWaitForDB_GivesUpAfterTimeout(t *testing.T) {
	p := &fakePinger{errs: []error{errors.New("x"), errors.New("x"), errors.New("x"), errors.New("x")}}
	err := waitForDB(context.Background(), p, 0, func(time.Duration) {})
	if err == nil || isPermanentDBError(err) {
		t.Fatalf("erwartet transienten timeout-fehler, war %v", err)
	}
}

func TestIsPermanentDBError(t *testing.T) {
	for code, want := range map[string]bool{"28P01": true, "28000": true, "3D000": true, "57P03": false} {
		if got := isPermanentDBError(&pgconn.PgError{Code: code}); got != want {
			t.Errorf("%s: got %v", code, got)
		}
	}
	if isPermanentDBError(errors.New("connection refused")) {
		t.Error("verbindungsfehler ist transient")
	}
}