102 lines
3.7 KiB
Go
102 lines
3.7 KiB
Go
package certificate
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/domain"
|
|
)
|
|
|
|
type managementRenewalFunc func(context.Context, RenewalPolicy, time.Time) (RenewalReport, error)
|
|
|
|
func (f managementRenewalFunc) RunRenewal(ctx context.Context, policy RenewalPolicy, now time.Time) (RenewalReport, error) {
|
|
return f(ctx, policy, now)
|
|
}
|
|
|
|
func TestRenewalSchedulerChecksManagementWithoutBusinessDomains(t *testing.T) {
|
|
calls := 0
|
|
management := managementRenewalFunc(func(_ context.Context, policy RenewalPolicy, now time.Time) (RenewalReport, error) {
|
|
calls++
|
|
if policy.MaxRetries != 3 || now.IsZero() {
|
|
t.Fatal("管理续期未收到当前策略和时间")
|
|
}
|
|
return RenewalReport{Checked: 1, Selected: 1, Succeeded: 1}, nil
|
|
})
|
|
s := NewRenewalSchedulerWithManagement(renewalLister{}, &renewalService{}, RenewalPolicy{}, management)
|
|
report, err := s.RunOnce(t.Context())
|
|
if err != nil || calls != 1 || report.Checked != 1 || report.Succeeded != 1 {
|
|
t.Fatalf("空业务列表漏掉管理证书：%+v %v", report, err)
|
|
}
|
|
}
|
|
|
|
func TestRenewalSchedulerKeepsBusinessRenewalAfterManagementFailure(t *testing.T) {
|
|
managementErr := errors.New("模拟管理证书读取失败")
|
|
management := managementRenewalFunc(func(context.Context, RenewalPolicy, time.Time) (RenewalReport, error) {
|
|
return RenewalReport{Checked: 1, Selected: 1, Failed: 1}, managementErr
|
|
})
|
|
issuer := &renewalService{}
|
|
s := NewRenewalSchedulerWithManagement(renewalLister{items: []domain.Domain{{ID: "business", AutoRenew: true}}}, issuer, RenewalPolicy{}, management)
|
|
report, err := s.RunOnce(t.Context())
|
|
if !errors.Is(err, managementErr) || report.Checked != 2 || report.Succeeded != 1 || report.Failed != 1 || len(issuer.ids) != 1 {
|
|
t.Fatalf("管理错误阻断业务续期：%+v %v", report, err)
|
|
}
|
|
}
|
|
|
|
func TestRenewalSchedulerChecksManagementBeforeFailedBusinessQuery(t *testing.T) {
|
|
management := managementRenewalFunc(func(context.Context, RenewalPolicy, time.Time) (RenewalReport, error) {
|
|
return RenewalReport{Checked: 1, Selected: 1, Succeeded: 1}, nil
|
|
})
|
|
s := NewRenewalSchedulerWithManagement(renewalLister{err: errors.New("模拟业务查询失败")}, &renewalService{}, RenewalPolicy{}, management)
|
|
report, err := s.RunOnce(t.Context())
|
|
if err == nil || report.Succeeded != 1 {
|
|
t.Fatalf("业务查询失败导致遗漏管理续期：%+v %v", report, err)
|
|
}
|
|
}
|
|
|
|
func TestRenewalSchedulerStopsManagementBeforeClosing(t *testing.T) {
|
|
started := make(chan struct{})
|
|
stopped := make(chan struct{})
|
|
management := managementRenewalFunc(func(ctx context.Context, _ RenewalPolicy, _ time.Time) (RenewalReport, error) {
|
|
close(started)
|
|
<-ctx.Done()
|
|
close(stopped)
|
|
return RenewalReport{}, ctx.Err()
|
|
})
|
|
s := NewRenewalSchedulerWithManagement(renewalLister{}, &renewalService{}, RenewalPolicy{}, management)
|
|
s.Start(t.Context())
|
|
select {
|
|
case <-started:
|
|
case <-time.After(time.Second):
|
|
t.Fatal("管理调度未启动")
|
|
}
|
|
s.Stop()
|
|
select {
|
|
case <-stopped:
|
|
default:
|
|
t.Fatal("停止调度未等待管理任务退出")
|
|
}
|
|
}
|
|
|
|
func TestRenewalSchedulerDoesNotLengthenShortCheckInterval(t *testing.T) {
|
|
calls := make(chan struct{}, 2)
|
|
management := managementRenewalFunc(func(context.Context, RenewalPolicy, time.Time) (RenewalReport, error) {
|
|
select {
|
|
case calls <- struct{}{}:
|
|
default:
|
|
}
|
|
return RenewalReport{Checked: 1, NearExpiry: 1}, nil
|
|
})
|
|
s := NewRenewalSchedulerWithManagement(renewalLister{}, &renewalService{}, RenewalPolicy{CheckInterval: 10 * time.Millisecond}, management)
|
|
s.Start(t.Context())
|
|
t.Cleanup(s.Stop)
|
|
for i := 0; i < 2; i++ {
|
|
select {
|
|
case <-calls:
|
|
case <-time.After(time.Second):
|
|
t.Fatal("临期规则不应将较短检查间隔延长为一天")
|
|
}
|
|
}
|
|
}
|