mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-05-05 05:30:44 +08:00
29 lines
911 B
Go
29 lines
911 B
Go
package repository
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAffiliateUserOverviewSQLIncludesMaturedFrozenQuota(t *testing.T) {
|
|
query := strings.Join(strings.Fields(affiliateUserOverviewSQL), " ")
|
|
|
|
require.Contains(t, query, "ua.aff_quota + COALESCE(matured.matured_frozen_quota, 0)")
|
|
require.Contains(t, query, "frozen_until <= NOW()")
|
|
}
|
|
|
|
func TestAffiliateRecordQueriesUseLedgerAuditFields(t *testing.T) {
|
|
source, err := os.ReadFile("affiliate_repo.go")
|
|
require.NoError(t, err)
|
|
content := string(source)
|
|
|
|
require.Contains(t, content, "JOIN payment_orders po ON po.id = ual.source_order_id")
|
|
require.Contains(t, content, "ual.amount::double precision")
|
|
require.Contains(t, content, "ual.balance_after::double precision")
|
|
require.NotContains(t, content, "parseAffiliateRebateAmount")
|
|
require.NotContains(t, content, `"current_balance": "u.balance"`)
|
|
}
|