Andrew Louis
78000ba7a0
Fixes golang/go#37340 Change-Id: I19c4f150b8607ad4a1613cf97ad3362f4b779d7c GitHub-Last-Rev: 4215964b4a680b135301695ccd56cff88a8ffb26 GitHub-Pull-Request: golang/crypto#121 Reviewed-on: https://go-review.googlesource.com/c/crypto/+/220343 Reviewed-by: Filippo Valsorda <filippo@golang.org>
107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
// Copyright 2017 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package acme
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestRateLimit(t *testing.T) {
|
|
now := time.Date(2017, 04, 27, 10, 0, 0, 0, time.UTC)
|
|
f := timeNow
|
|
defer func() { timeNow = f }()
|
|
timeNow = func() time.Time { return now }
|
|
|
|
h120, hTime := http.Header{}, http.Header{}
|
|
h120.Set("Retry-After", "120")
|
|
hTime.Set("Retry-After", "Tue Apr 27 11:00:00 2017")
|
|
|
|
err1 := &Error{
|
|
ProblemType: "urn:ietf:params:acme:error:nolimit",
|
|
Header: h120,
|
|
}
|
|
err2 := &Error{
|
|
ProblemType: "urn:ietf:params:acme:error:rateLimited",
|
|
Header: h120,
|
|
}
|
|
err3 := &Error{
|
|
ProblemType: "urn:ietf:params:acme:error:rateLimited",
|
|
Header: nil,
|
|
}
|
|
err4 := &Error{
|
|
ProblemType: "urn:ietf:params:acme:error:rateLimited",
|
|
Header: hTime,
|
|
}
|
|
|
|
tt := []struct {
|
|
err error
|
|
res time.Duration
|
|
ok bool
|
|
}{
|
|
{nil, 0, false},
|
|
{errors.New("dummy"), 0, false},
|
|
{err1, 0, false},
|
|
{err2, 2 * time.Minute, true},
|
|
{err3, 0, true},
|
|
{err4, time.Hour, true},
|
|
}
|
|
for i, test := range tt {
|
|
res, ok := RateLimit(test.err)
|
|
if ok != test.ok {
|
|
t.Errorf("%d: RateLimit(%+v): ok = %v; want %v", i, test.err, ok, test.ok)
|
|
continue
|
|
}
|
|
if res != test.res {
|
|
t.Errorf("%d: RateLimit(%+v) = %v; want %v", i, test.err, res, test.res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAuthorizationError(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
err *AuthorizationError
|
|
msg string
|
|
}{
|
|
{
|
|
desc: "when auth error identifier is set",
|
|
err: &AuthorizationError{
|
|
Identifier: "domain.com",
|
|
Errors: []error{
|
|
(&wireError{
|
|
Status: 403,
|
|
Type: "urn:ietf:params:acme:error:caa",
|
|
Detail: "CAA record for domain.com prevents issuance",
|
|
}).error(nil),
|
|
},
|
|
},
|
|
msg: "acme: authorization error for domain.com: 403 urn:ietf:params:acme:error:caa: CAA record for domain.com prevents issuance",
|
|
},
|
|
|
|
{
|
|
desc: "when auth error identifier is unset",
|
|
err: &AuthorizationError{
|
|
Errors: []error{
|
|
(&wireError{
|
|
Status: 403,
|
|
Type: "urn:ietf:params:acme:error:caa",
|
|
Detail: "CAA record for domain.com prevents issuance",
|
|
}).error(nil),
|
|
},
|
|
},
|
|
msg: "acme: authorization error: 403 urn:ietf:params:acme:error:caa: CAA record for domain.com prevents issuance",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
if tt.err.Error() != tt.msg {
|
|
t.Errorf("got: %s\nwant: %s", tt.err, tt.msg)
|
|
}
|
|
}
|
|
}
|