李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
57 lines
2.0 KiB
Go
57 lines
2.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"entgo.io/ent/dialect"
|
|
entsql "entgo.io/ent/dialect/sql"
|
|
)
|
|
|
|
func newRedeemAdjustmentRepoMock(t *testing.T) (*userRepository, sqlmock.Sqlmock) {
|
|
t.Helper()
|
|
db, mock, err := sqlmock.New()
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = db.Close() })
|
|
driver := entsql.OpenDB(dialect.Postgres, db)
|
|
client := dbent.NewClient(dbent.Driver(driver))
|
|
t.Cleanup(func() { _ = client.Close() })
|
|
return newUserRepositoryWithSQL(client, db), mock
|
|
}
|
|
|
|
func TestApplyRedeemBalanceAdjustment_UsesAtomicFloor(t *testing.T) {
|
|
repo, mock := newRedeemAdjustmentRepoMock(t)
|
|
mock.ExpectExec(`UPDATE users SET balance = GREATEST\(balance \+ \$1, 0\), updated_at = NOW\(\) WHERE id = \$2 AND deleted_at IS NULL`).
|
|
WithArgs(-7.0, int64(42)).
|
|
WillReturnResult(sqlmock.NewResult(0, 1))
|
|
|
|
require.NoError(t, repo.ApplyRedeemBalanceAdjustment(context.Background(), 42, -7))
|
|
require.NoError(t, mock.ExpectationsWereMet())
|
|
}
|
|
|
|
func TestApplyRedeemConcurrencyAdjustment_UsesAtomicFloor(t *testing.T) {
|
|
repo, mock := newRedeemAdjustmentRepoMock(t)
|
|
mock.ExpectExec(`UPDATE users SET concurrency = GREATEST\(concurrency \+ \$1, 0\), updated_at = NOW\(\) WHERE id = \$2 AND deleted_at IS NULL`).
|
|
WithArgs(-7, int64(42)).
|
|
WillReturnResult(sqlmock.NewResult(0, 1))
|
|
|
|
require.NoError(t, repo.ApplyRedeemConcurrencyAdjustment(context.Background(), 42, -7))
|
|
require.NoError(t, mock.ExpectationsWereMet())
|
|
}
|
|
|
|
func TestApplyRedeemAdjustment_MissingUser(t *testing.T) {
|
|
repo, mock := newRedeemAdjustmentRepoMock(t)
|
|
mock.ExpectExec(`UPDATE users SET balance = GREATEST\(balance \+ \$1, 0\), updated_at = NOW\(\) WHERE id = \$2 AND deleted_at IS NULL`).
|
|
WithArgs(-1.0, int64(404)).
|
|
WillReturnResult(sqlmock.NewResult(0, 0))
|
|
|
|
err := repo.ApplyRedeemBalanceAdjustment(context.Background(), 404, -1)
|
|
require.ErrorIs(t, err, service.ErrUserNotFound)
|
|
require.NoError(t, mock.ExpectationsWereMet())
|
|
}
|