diff --git a/internal/autocert/provider.go b/internal/autocert/provider.go index b526cc4..c598732 100644 --- a/internal/autocert/provider.go +++ b/internal/autocert/provider.go @@ -43,6 +43,12 @@ type ( var ErrGetCertFailure = errors.New("get certificate failed") +func TestProvider(cert *tls.Certificate) *Provider { + return &Provider{ + tlsCert: cert, + } +} + func (p *Provider) GetCert(_ *tls.ClientHelloInfo) (*tls.Certificate, error) { if p.tlsCert == nil { return nil, ErrGetCertFailure @@ -173,7 +179,7 @@ func (p *Provider) ScheduleRenewal(parent task.Parent) { timer := time.NewTimer(time.Until(renewalTime)) defer timer.Stop() - task := parent.Subtask("cert-renew-scheduler") + task := parent.Subtask("cert-renew-scheduler", true) defer task.Finish(nil) for { diff --git a/internal/utils/strutils/url_test.go b/internal/utils/strutils/url_test.go index 756418c..2cda448 100644 --- a/internal/utils/strutils/url_test.go +++ b/internal/utils/strutils/url_test.go @@ -1,8 +1,9 @@ -package strutils +package strutils_test import ( "testing" + . "github.com/yusing/go-proxy/internal/utils/strutils" expect "github.com/yusing/go-proxy/internal/utils/testing" )