internal/store/postgres/account_settings_integration_test.go

Rama main · réplica de lectura del repositorio original.

package postgres

import (
	"context"
	"testing"

	"telesrv/internal/domain"
)

// TestAccountSettingsRoundTripPostgres regression for migration 0004: account-level singleton settings are really persisted
// (globalPrivacy/accountTTL/sensitive/contactSignUp), including not-found -> default and upsert overwrite.
func TestAccountSettingsRoundTripPostgres(t *testing.T) {
	pool := testPool(t)
	ctx := context.Background()
	store := NewPasswordStore(pool)

	users := NewUserStore(pool)
	suffix := randomSuffix(t)
	u, err := users.Create(ctx, domain.User{AccessHash: 91, Phone: "+1664" + suffix + "01", FirstName: "AcctSettings"})
	if err != nil {
		t.Fatalf("create user: %v", err)
	}
	t.Cleanup(func() {
		_, _ = pool.Exec(ctx, "DELETE FROM account_settings WHERE user_id = $1", u.ID)
		_, _ = pool.Exec(ctx, "DELETE FROM users WHERE id = $1", u.ID)
	})

	// Not persisted yet: not found.
	if _, found, err := store.GetAccountSettings(ctx, u.ID); err != nil || found {
		t.Fatalf("get before save = found %v err %v, want not found", found, err)
	}

	want := domain.AccountSettings{
		GlobalPrivacy: domain.GlobalPrivacy{
			ArchiveAndMuteNewNoncontactPeers: true,
			HideReadMarks:                    true,
			DisplayGiftsButton:               true,
			NoncontactPeersPaidStars:         75,
			DisallowedGifts: domain.DisallowedGifts{
				UnlimitedStargifts: true,
				PremiumGifts:       true,
			},
		},
		AccountTTLDays:          30,
		SensitiveContentEnabled: true,
		ContactSignUpSilent:     true,
	}
	if err := store.SaveAccountSettings(ctx, u.ID, want); err != nil {
		t.Fatalf("save account settings: %v", err)
	}
	got, found, err := store.GetAccountSettings(ctx, u.ID)
	if err != nil || !found {
		t.Fatalf("get after save = found %v err %v", found, err)
	}
	if got != want {
		t.Fatalf("account settings round-trip = %+v, want %+v", got, want)
	}

	// upsert overwrite: change several fields and read back.
	want.GlobalPrivacy.HideReadMarks = false
	want.AccountTTLDays = 365
	want.SensitiveContentEnabled = false
	if err := store.SaveAccountSettings(ctx, u.ID, want); err != nil {
		t.Fatalf("re-save account settings: %v", err)
	}
	got, found, err = store.GetAccountSettings(ctx, u.ID)
	if err != nil || !found {
		t.Fatalf("get after re-save = found %v err %v", found, err)
	}
	if got != want {
		t.Fatalf("account settings after upsert = %+v, want %+v", got, want)
	}
}