26 lines
600 B
Go
26 lines
600 B
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
|
|
)
|
|
|
|
func TestRewardRemaining(t *testing.T) {
|
|
if got := rewardRemaining(2, 0); got != 2 {
|
|
t.Fatalf("未领取时剩余应为 2, got %d", got)
|
|
}
|
|
if got := rewardRemaining(2, 2); got != 0 {
|
|
t.Fatalf("已用满时剩余应为 0, got %d", got)
|
|
}
|
|
if got := rewardRemaining(1, 1); got != 0 {
|
|
t.Fatalf("vip_trial 已用完剩余应为 0, got %d", got)
|
|
}
|
|
}
|
|
|
|
func TestRewardQuota(t *testing.T) {
|
|
if got := rewardQuota(t.Context(), "effect_extra"); got != 2 {
|
|
t.Fatalf("默认每日 2 次, got %d", got)
|
|
}
|
|
}
|