diff --git a/pkg/acme/client/fake.go b/pkg/acme/client/fake.go index 92c487d6d..c9217d625 100644 --- a/pkg/acme/client/fake.go +++ b/pkg/acme/client/fake.go @@ -41,6 +41,7 @@ type FakeACME struct { FakeGetAccount func(ctx context.Context) (*acme.Account, error) FakeHTTP01ChallengeResponse func(token string) (string, error) FakeDNS01ChallengeRecord func(token string) (string, error) + FakeDiscover func(ctx context.Context) (acme.Directory, error) } func (f *FakeACME) CreateOrder(ctx context.Context, order *acme.Order) (*acme.Order, error) { @@ -133,3 +134,10 @@ func (f *FakeACME) DNS01ChallengeRecord(token string) (string, error) { } return "", fmt.Errorf("DNS01ChallengeRecord not implemented") } + +func (f *FakeACME) Discover(ctx context.Context) (acme.Directory, error) { + if f.FakeDiscover != nil { + return f.FakeDiscover(ctx) + } + return acme.Directory{}, fmt.Errorf("Discover not implemented") +} diff --git a/pkg/acme/client/interfaces.go b/pkg/acme/client/interfaces.go index 24dd19b54..1ad6bcede 100644 --- a/pkg/acme/client/interfaces.go +++ b/pkg/acme/client/interfaces.go @@ -36,6 +36,7 @@ type Interface interface { GetAccount(ctx context.Context) (*acme.Account, error) HTTP01ChallengeResponse(token string) (string, error) DNS01ChallengeRecord(token string) (string, error) + Discover(ctx context.Context) (acme.Directory, error) } var _ Interface = &acme.Client{} diff --git a/pkg/acme/client/middleware/logger.go b/pkg/acme/client/middleware/logger.go index c3ca8880e..3ba479e8b 100644 --- a/pkg/acme/client/middleware/logger.go +++ b/pkg/acme/client/middleware/logger.go @@ -98,3 +98,8 @@ func (l *Logger) DNS01ChallengeRecord(token string) (string, error) { glog.Infof("Calling DNS01ChallengeRecord") return l.baseCl.DNS01ChallengeRecord(token) } + +func (l *Logger) Discover(ctx context.Context) (acme.Directory, error) { + glog.Infof("Calling Discover") + return l.baseCl.Discover(ctx) +}