VpSanta33
115 lines
3.7 KiB
Go
115 lines
3.7 KiB
Go
//go:build integration
|
|
|
|
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func querySingleFloat(t *testing.T, ctx context.Context, client *dbent.Client, query string, args ...any) float64 {
|
|
t.Helper()
|
|
rows, err := client.QueryContext(ctx, query, args...)
|
|
require.NoError(t, err)
|
|
defer func() { _ = rows.Close() }()
|
|
|
|
require.True(t, rows.Next(), "expected one row")
|
|
var value float64
|
|
require.NoError(t, rows.Scan(&value))
|
|
require.NoError(t, rows.Err())
|
|
return value
|
|
}
|
|
|
|
func querySingleInt(t *testing.T, ctx context.Context, client *dbent.Client, query string, args ...any) int {
|
|
t.Helper()
|
|
rows, err := client.QueryContext(ctx, query, args...)
|
|
require.NoError(t, err)
|
|
defer func() { _ = rows.Close() }()
|
|
|
|
require.True(t, rows.Next(), "expected one row")
|
|
var value int
|
|
require.NoError(t, rows.Scan(&value))
|
|
require.NoError(t, rows.Err())
|
|
return value
|
|
}
|
|
|
|
func TestAffiliateRepository_TransferQuotaToBalance_UsesClaimedQuotaBeforeClear(t *testing.T) {
|
|
ctx := context.Background()
|
|
tx := testEntTx(t)
|
|
txCtx := dbent.NewTxContext(ctx, tx)
|
|
client := tx.Client()
|
|
|
|
repo := NewAffiliateRepository(client, integrationDB)
|
|
|
|
u := mustCreateUser(t, client, &service.User{
|
|
Email: fmt.Sprintf("affiliate-transfer-%d@example.com", time.Now().UnixNano()),
|
|
PasswordHash: "hash",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
Balance: 5.5,
|
|
Concurrency: 5,
|
|
})
|
|
|
|
affCode := fmt.Sprintf("AFF%09d", time.Now().UnixNano()%1_000_000_000)
|
|
_, err := client.ExecContext(txCtx, `
|
|
INSERT INTO user_affiliates (user_id, aff_code, aff_quota, aff_history_quota, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $3, NOW(), NOW())`, u.ID, affCode, 12.34)
|
|
require.NoError(t, err)
|
|
|
|
transferred, balance, err := repo.TransferQuotaToBalance(txCtx, u.ID)
|
|
require.NoError(t, err)
|
|
require.InDelta(t, 12.34, transferred, 1e-9)
|
|
require.InDelta(t, 17.84, balance, 1e-9)
|
|
|
|
affQuota := querySingleFloat(t, txCtx, client,
|
|
"SELECT aff_quota::double precision FROM user_affiliates WHERE user_id = $1", u.ID)
|
|
require.InDelta(t, 0.0, affQuota, 1e-9)
|
|
|
|
persistedBalance := querySingleFloat(t, txCtx, client,
|
|
"SELECT balance::double precision FROM users WHERE id = $1", u.ID)
|
|
require.InDelta(t, 17.84, persistedBalance, 1e-9)
|
|
|
|
ledgerCount := querySingleInt(t, txCtx, client,
|
|
"SELECT COUNT(*) FROM user_affiliate_ledger WHERE user_id = $1 AND action = 'transfer'", u.ID)
|
|
require.Equal(t, 1, ledgerCount)
|
|
}
|
|
|
|
func TestAffiliateRepository_TransferQuotaToBalance_EmptyQuota(t *testing.T) {
|
|
ctx := context.Background()
|
|
tx := testEntTx(t)
|
|
txCtx := dbent.NewTxContext(ctx, tx)
|
|
client := tx.Client()
|
|
|
|
repo := NewAffiliateRepository(client, integrationDB)
|
|
|
|
u := mustCreateUser(t, client, &service.User{
|
|
Email: fmt.Sprintf("affiliate-empty-%d@example.com", time.Now().UnixNano()),
|
|
PasswordHash: "hash",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
Balance: 3.21,
|
|
Concurrency: 5,
|
|
})
|
|
|
|
affCode := fmt.Sprintf("AFF%09d", time.Now().UnixNano()%1_000_000_000)
|
|
_, err := client.ExecContext(txCtx, `
|
|
INSERT INTO user_affiliates (user_id, aff_code, aff_quota, aff_history_quota, created_at, updated_at)
|
|
VALUES ($1, $2, 0, 0, NOW(), NOW())`, u.ID, affCode)
|
|
require.NoError(t, err)
|
|
|
|
transferred, balance, err := repo.TransferQuotaToBalance(txCtx, u.ID)
|
|
require.ErrorIs(t, err, service.ErrAffiliateQuotaEmpty)
|
|
require.InDelta(t, 0.0, transferred, 1e-9)
|
|
require.InDelta(t, 0.0, balance, 1e-9)
|
|
|
|
persistedBalance := querySingleFloat(t, txCtx, client,
|
|
"SELECT balance::double precision FROM users WHERE id = $1", u.ID)
|
|
require.InDelta(t, 3.21, persistedBalance, 1e-9)
|
|
}
|