Skip to content

Commit bcaba4f

Browse files
committed
Fix api tests.
1 parent d92a7f2 commit bcaba4f

File tree

1 file changed

+16
-15
lines changed

1 file changed

+16
-15
lines changed

api/api_test.go

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import (
2525

2626
"github.com/go-chi/chi"
2727
"github.com/smallstep/certificates/authority"
28+
"github.com/smallstep/certificates/authority/provisioner"
2829
"github.com/smallstep/certificates/logging"
2930
"github.com/smallstep/cli/crypto/tlsutil"
3031
"github.com/smallstep/cli/jose"
@@ -410,22 +411,22 @@ func TestSignRequest_Validate(t *testing.T) {
410411
type mockAuthority struct {
411412
ret1, ret2 interface{}
412413
err error
413-
authorize func(ott string) ([]interface{}, error)
414+
authorize func(ott string) ([]provisioner.SignOption, error)
414415
getTLSOptions func() *tlsutil.TLSOptions
415416
root func(shasum string) (*x509.Certificate, error)
416-
sign func(cr *x509.CertificateRequest, signOpts authority.SignOptions, extraOpts ...interface{}) (*x509.Certificate, *x509.Certificate, error)
417+
sign func(cr *x509.CertificateRequest, signOpts authority.SignOptions, extraOpts ...provisioner.SignOption) (*x509.Certificate, *x509.Certificate, error)
417418
renew func(cert *x509.Certificate) (*x509.Certificate, *x509.Certificate, error)
418-
getProvisioners func(nextCursor string, limit int) ([]*authority.Provisioner, string, error)
419+
getProvisioners func(nextCursor string, limit int) ([]*provisioner.Provisioner, string, error)
419420
getEncryptedKey func(kid string) (string, error)
420421
getRoots func() ([]*x509.Certificate, error)
421422
getFederation func() ([]*x509.Certificate, error)
422423
}
423424

424-
func (m *mockAuthority) Authorize(ott string) ([]interface{}, error) {
425+
func (m *mockAuthority) Authorize(ott string) ([]provisioner.SignOption, error) {
425426
if m.authorize != nil {
426427
return m.authorize(ott)
427428
}
428-
return m.ret1.([]interface{}), m.err
429+
return m.ret1.([]provisioner.SignOption), m.err
429430
}
430431

431432
func (m *mockAuthority) GetTLSOptions() *tlsutil.TLSOptions {
@@ -442,7 +443,7 @@ func (m *mockAuthority) Root(shasum string) (*x509.Certificate, error) {
442443
return m.ret1.(*x509.Certificate), m.err
443444
}
444445

445-
func (m *mockAuthority) Sign(cr *x509.CertificateRequest, signOpts authority.SignOptions, extraOpts ...interface{}) (*x509.Certificate, *x509.Certificate, error) {
446+
func (m *mockAuthority) Sign(cr *x509.CertificateRequest, signOpts authority.SignOptions, extraOpts ...provisioner.SignOption) (*x509.Certificate, *x509.Certificate, error) {
446447
if m.sign != nil {
447448
return m.sign(cr, signOpts, extraOpts...)
448449
}
@@ -456,11 +457,11 @@ func (m *mockAuthority) Renew(cert *x509.Certificate) (*x509.Certificate, *x509.
456457
return m.ret1.(*x509.Certificate), m.ret2.(*x509.Certificate), m.err
457458
}
458459

459-
func (m *mockAuthority) GetProvisioners(nextCursor string, limit int) ([]*authority.Provisioner, string, error) {
460+
func (m *mockAuthority) GetProvisioners(nextCursor string, limit int) ([]*provisioner.Provisioner, string, error) {
460461
if m.getProvisioners != nil {
461462
return m.getProvisioners(nextCursor, limit)
462463
}
463-
return m.ret1.([]*authority.Provisioner), m.ret2.(string), m.err
464+
return m.ret1.([]*provisioner.Provisioner), m.ret2.(string), m.err
464465
}
465466

466467
func (m *mockAuthority) GetEncryptedKey(kid string) (string, error) {
@@ -597,7 +598,7 @@ func Test_caHandler_Sign(t *testing.T) {
597598
tests := []struct {
598599
name string
599600
input string
600-
certAttrOpts []interface{}
601+
certAttrOpts []provisioner.SignOption
601602
autherr error
602603
cert *x509.Certificate
603604
root *x509.Certificate
@@ -617,7 +618,7 @@ func Test_caHandler_Sign(t *testing.T) {
617618
t.Run(tt.name, func(t *testing.T) {
618619
h := New(&mockAuthority{
619620
ret1: tt.cert, ret2: tt.root, err: tt.signErr,
620-
authorize: func(ott string) ([]interface{}, error) {
621+
authorize: func(ott string) ([]provisioner.SignOption, error) {
621622
return tt.certAttrOpts, tt.autherr
622623
},
623624
getTLSOptions: func() *tlsutil.TLSOptions {
@@ -723,19 +724,19 @@ func Test_caHandler_Provisioners(t *testing.T) {
723724
t.Fatal(err)
724725
}
725726

726-
p := []*authority.Provisioner{
727-
{
727+
p := []*provisioner.Provisioner{
728+
provisioner.New(&provisioner.JWK{
728729
Type: "JWK",
729730
Name: "max",
730731
EncryptedKey: "abc",
731732
Key: &key,
732-
},
733-
{
733+
}),
734+
provisioner.New(&provisioner.JWK{
734735
Type: "JWK",
735736
Name: "mariano",
736737
EncryptedKey: "def",
737738
Key: &key,
738-
},
739+
}),
739740
}
740741
pr := ProvisionersResponse{
741742
Provisioners: p,

0 commit comments

Comments
 (0)