67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package certificate
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/domain"
|
|
)
|
|
|
|
type renewalLister struct {
|
|
items []domain.Domain
|
|
err error
|
|
}
|
|
|
|
func (r renewalLister) List(context.Context) ([]domain.Domain, error) { return r.items, r.err }
|
|
|
|
type renewalService struct {
|
|
ids []string
|
|
err error
|
|
}
|
|
|
|
func (s *renewalService) Issue(_ context.Context, id string) (IssueResult, error) {
|
|
s.ids = append(s.ids, id)
|
|
return IssueResult{}, s.err
|
|
}
|
|
|
|
func TestRenewalSchedulerSelectsOnlyExpiringAutoRenewDomains(t *testing.T) {
|
|
now := time.Date(2026, 8, 26, 0, 0, 0, 0, time.UTC)
|
|
soon := now.Add(2 * 24 * time.Hour)
|
|
later := now.Add(20 * 24 * time.Hour)
|
|
items := []domain.Domain{
|
|
{ID: "soon", Name: "soon.example.com", AutoRenew: true, ExpiresAt: &soon},
|
|
{ID: "later", Name: "later.example.com", AutoRenew: true, ExpiresAt: &later},
|
|
{ID: "disabled", Name: "disabled.example.com", AutoRenew: false, ExpiresAt: &soon},
|
|
{ID: "missing", Name: "missing.example.com", AutoRenew: true},
|
|
}
|
|
service := &renewalService{}
|
|
scheduler := NewRenewalScheduler(renewalLister{items: items}, service, 6)
|
|
scheduler.now = func() time.Time { return now }
|
|
report, err := scheduler.RunOnce(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if report.Checked != 4 || report.Selected != 2 || report.Succeeded != 2 || report.Failed != 0 {
|
|
t.Fatalf("续期筛选结果错误：%+v", report)
|
|
}
|
|
if len(service.ids) != 2 || service.ids[0] != "soon" || service.ids[1] != "missing" {
|
|
t.Fatalf("续期执行域名错误：%v", service.ids)
|
|
}
|
|
}
|
|
|
|
func TestRenewalSchedulerReportsFailures(t *testing.T) {
|
|
now := time.Now().UTC()
|
|
expires := now.Add(time.Hour)
|
|
item := domain.Domain{ID: "one", Name: "one.example.com", AutoRenew: true, ExpiresAt: &expires}
|
|
scheduler := NewRenewalScheduler(renewalLister{items: []domain.Domain{item}}, &renewalService{err: errors.New("模拟续期失败")}, 6)
|
|
report, err := scheduler.RunOnce(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if report.Failed != 1 || len(report.Errors) != 1 {
|
|
t.Fatalf("续期失败结果错误：%+v", report)
|
|
}
|
|
}
|