TrueCloudLab/certificates
16
0
Fork 2
Code Issues 1 Pull requests 1 Releases Activity Actions

Use errs package for HTTP errors.

Browse source
This commit is contained in:
Mariano Cano 2020-01-03 17:41:16 -08:00
parent 50717b3ffa
commit 76c14560b0
1 changed files with 21 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
ca/client_test.go
View file

@ -16,6 +16,8 @@ import (
"testing" "testing"
"time" "time"
"github.com/smallstep/certificates/errs"
"github.com/smallstep/assert" "github.com/smallstep/assert"
"github.com/smallstep/certificates/api" "github.com/smallstep/certificates/api"
"github.com/smallstep/certificates/authority" "github.com/smallstep/certificates/authority"
@ -152,8 +154,8 @@ func equalJSON(t *testing.T, a interface{}, b interface{}) bool {
func TestClient_Version(t *testing.T) { func TestClient_Version(t *testing.T) {
ok := &api.VersionResponse{Version: "test"} ok := &api.VersionResponse{Version: "test"}
internal := api.InternalServerError(fmt.Errorf("Internal Server Error")) internal := errs.InternalServerError(fmt.Errorf("Internal Server Error"))
notFound := api.NotFound(fmt.Errorf("Not Found")) notFound := errs.NotFound(fmt.Errorf("Not Found"))
tests := []struct { tests := []struct {
name string name string
@ -207,7 +209,7 @@ func TestClient_Version(t *testing.T) {
func TestClient_Health(t *testing.T) { func TestClient_Health(t *testing.T) {
ok := &api.HealthResponse{Status: "ok"} ok := &api.HealthResponse{Status: "ok"}
nok := api.InternalServerError(fmt.Errorf("Internal Server Error")) nok := errs.InternalServerError(fmt.Errorf("Internal Server Error"))
tests := []struct { tests := []struct {
name string name string
@ -262,7 +264,7 @@ func TestClient_Root(t *testing.T) {
ok := &api.RootResponse{ ok := &api.RootResponse{
RootPEM: api.Certificate{Certificate: parseCertificate(rootPEM)}, RootPEM: api.Certificate{Certificate: parseCertificate(rootPEM)},
} }
notFound := api.NotFound(fmt.Errorf("Not Found")) notFound := errs.NotFound(fmt.Errorf("Not Found"))
tests := []struct { tests := []struct {
name string name string
@ -332,8 +334,8 @@ func TestClient_Sign(t *testing.T) {
NotBefore: api.NewTimeDuration(time.Now()), NotBefore: api.NewTimeDuration(time.Now()),
NotAfter: api.NewTimeDuration(time.Now().AddDate(0, 1, 0)), NotAfter: api.NewTimeDuration(time.Now().AddDate(0, 1, 0)),
} }
unauthorized := api.Unauthorized(fmt.Errorf("Unauthorized")) unauthorized := errs.Unauthorized(fmt.Errorf("Unauthorized"))
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string
@ -407,8 +409,8 @@ func TestClient_Revoke(t *testing.T) {
OTT: "the-ott", OTT: "the-ott",
ReasonCode: 4, ReasonCode: 4,
} }
unauthorized := api.Unauthorized(fmt.Errorf("Unauthorized")) unauthorized := errs.Unauthorized(fmt.Errorf("Unauthorized"))
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string
@ -483,8 +485,8 @@ func TestClient_Renew(t *testing.T) {
{Certificate: parseCertificate(rootPEM)}, {Certificate: parseCertificate(rootPEM)},
}, },
} }
unauthorized := api.Unauthorized(fmt.Errorf("Unauthorized")) unauthorized := errs.Unauthorized(fmt.Errorf("Unauthorized"))
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string
@ -541,7 +543,7 @@ func TestClient_Provisioners(t *testing.T) {
ok := &api.ProvisionersResponse{ ok := &api.ProvisionersResponse{
Provisioners: provisioner.List{}, Provisioners: provisioner.List{},
} }
internalServerError := api.InternalServerError(fmt.Errorf("Internal Server Error")) internalServerError := errs.InternalServerError(fmt.Errorf("Internal Server Error"))
tests := []struct { tests := []struct {
name string name string
@ -603,7 +605,7 @@ func TestClient_ProvisionerKey(t *testing.T) {
ok := &api.ProvisionerKeyResponse{ ok := &api.ProvisionerKeyResponse{
Key: "an encrypted key", Key: "an encrypted key",
} }
notFound := api.NotFound(fmt.Errorf("Not Found")) notFound := errs.NotFound(fmt.Errorf("Not Found"))
tests := []struct { tests := []struct {
name string name string
@ -664,8 +666,8 @@ func TestClient_Roots(t *testing.T) {
{Certificate: parseCertificate(rootPEM)}, {Certificate: parseCertificate(rootPEM)},
}, },
} }
unauthorized := api.Unauthorized(fmt.Errorf("Unauthorized")) unauthorized := errs.Unauthorized(fmt.Errorf("Unauthorized"))
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string
@ -724,8 +726,8 @@ func TestClient_Federation(t *testing.T) {
{Certificate: parseCertificate(rootPEM)}, {Certificate: parseCertificate(rootPEM)},
}, },
} }
unauthorized := api.Unauthorized(fmt.Errorf("Unauthorized")) unauthorized := errs.Unauthorized(fmt.Errorf("Unauthorized"))
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string
@ -788,7 +790,7 @@ func TestClient_SSHRoots(t *testing.T) {
HostKeys: []api.SSHPublicKey{{PublicKey: key}}, HostKeys: []api.SSHPublicKey{{PublicKey: key}},
UserKeys: []api.SSHPublicKey{{PublicKey: key}}, UserKeys: []api.SSHPublicKey{{PublicKey: key}},
} }
notFound := api.NotFound(fmt.Errorf("Not Found")) notFound := errs.NotFound(fmt.Errorf("Not Found"))
tests := []struct { tests := []struct {
name string name string
@ -879,7 +881,7 @@ func Test_parseEndpoint(t *testing.T) {
func TestClient_RootFingerprint(t *testing.T) { func TestClient_RootFingerprint(t *testing.T) {
ok := &api.HealthResponse{Status: "ok"} ok := &api.HealthResponse{Status: "ok"}
nok := api.InternalServerError(fmt.Errorf("Internal Server Error")) nok := errs.InternalServerError(fmt.Errorf("Internal Server Error"))
httpsServer := httptest.NewTLSServer(nil) httpsServer := httptest.NewTLSServer(nil)
defer httpsServer.Close() defer httpsServer.Close()
@ -946,7 +948,7 @@ func TestClient_SSHBastion(t *testing.T) {
Hostname: "bastion.local", Hostname: "bastion.local",
}, },
} }
badRequest := api.BadRequest(fmt.Errorf("Bad Request")) badRequest := errs.BadRequest(fmt.Errorf("Bad Request"))
tests := []struct { tests := []struct {
name string name string