101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
|
package usecase
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"haixun-backend/internal/model/publish_guard/domain/entity"
|
||
|
|
domusecase "haixun-backend/internal/model/publish_guard/domain/usecase"
|
||
|
|
)
|
||
|
|
|
||
|
|
type guardRepoStub struct {
|
||
|
|
policy *entity.Policy
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *guardRepoStub) EnsureIndexes(ctx context.Context) error { return nil }
|
||
|
|
func (r *guardRepoStub) Get(ctx context.Context, tenantID, ownerUID, accountID string) (*entity.Policy, error) {
|
||
|
|
if r.policy == nil {
|
||
|
|
r.policy = defaultPolicy(tenantID, ownerUID, accountID)
|
||
|
|
}
|
||
|
|
return r.policy, nil
|
||
|
|
}
|
||
|
|
func (r *guardRepoStub) Upsert(ctx context.Context, policy *entity.Policy) (*entity.Policy, error) {
|
||
|
|
r.policy = policy
|
||
|
|
return policy, nil
|
||
|
|
}
|
||
|
|
func (r *guardRepoStub) DeleteByAccount(ctx context.Context, tenantID, ownerUID, accountID string) error {
|
||
|
|
r.policy = nil
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGuardCheckBlocksPausedPolicy(t *testing.T) {
|
||
|
|
repo := &guardRepoStub{policy: defaultPolicy("t", "u", "a")}
|
||
|
|
repo.policy.Paused = true
|
||
|
|
repo.policy.PausedReason = "token error"
|
||
|
|
uc := NewUseCase(repo, nil)
|
||
|
|
|
||
|
|
result, err := uc.Check(context.Background(), checkReq())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if result.Allowed || result.Reason != "token error" {
|
||
|
|
t.Fatalf("expected paused block, got %#v", result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGuardCheckDelaysWhenDailyLimitReached(t *testing.T) {
|
||
|
|
repo := &guardRepoStub{policy: defaultPolicy("t", "u", "a")}
|
||
|
|
repo.policy.MaxDailyPosts = 1
|
||
|
|
uc := NewUseCase(repo, nil)
|
||
|
|
|
||
|
|
result, err := uc.Check(context.Background(), checkReqWith(func(req *CheckRequestBuilder) {
|
||
|
|
req.publishedToday = 1
|
||
|
|
}))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if result.Allowed || result.DelayUntil == 0 {
|
||
|
|
t.Fatalf("expected delayed daily limit, got %#v", result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGuardCheckDelaysWhenIntervalNotReached(t *testing.T) {
|
||
|
|
repo := &guardRepoStub{policy: defaultPolicy("t", "u", "a")}
|
||
|
|
repo.policy.MinIntervalMinutes = 90
|
||
|
|
uc := NewUseCase(repo, nil)
|
||
|
|
|
||
|
|
now := time.Date(2026, 6, 30, 8, 0, 0, 0, time.UTC).UnixNano()
|
||
|
|
result, err := uc.Check(context.Background(), checkReqWith(func(req *CheckRequestBuilder) {
|
||
|
|
req.now = now
|
||
|
|
req.lastPublishedAt = now - int64(30*time.Minute)
|
||
|
|
}))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if result.Allowed || result.DelayUntil <= now {
|
||
|
|
t.Fatalf("expected interval delay, got %#v", result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type CheckRequestBuilder struct {
|
||
|
|
now int64
|
||
|
|
publishedToday int64
|
||
|
|
lastPublishedAt int64
|
||
|
|
}
|
||
|
|
|
||
|
|
func checkReq() domusecase.CheckRequest {
|
||
|
|
return checkReqWith(nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func checkReqWith(update func(*CheckRequestBuilder)) domusecase.CheckRequest {
|
||
|
|
builder := &CheckRequestBuilder{now: time.Date(2026, 6, 30, 8, 0, 0, 0, time.UTC).UnixNano()}
|
||
|
|
if update != nil {
|
||
|
|
update(builder)
|
||
|
|
}
|
||
|
|
return domusecase.CheckRequest{
|
||
|
|
TenantID: "t", OwnerUID: "u", AccountID: "a",
|
||
|
|
Now: builder.now, PublishedToday: builder.publishedToday, LastPublishedAt: builder.lastPublishedAt,
|
||
|
|
}
|
||
|
|
}
|