@@ -30,8 +30,6 @@ import (
3030 "github.com/spiffe/go-spiffe/v2/proto/spiffe/workload"
3131 "github.com/spiffe/go-spiffe/v2/svid/jwtsvid"
3232 "github.com/spiffe/go-spiffe/v2/svid/x509svid"
33- "github.com/stretchr/testify/assert"
34- "github.com/stretchr/testify/require"
3533 "github.com/tektoncd/pipeline/pkg/spire/test/pemutil"
3634 "github.com/tektoncd/pipeline/pkg/spire/test/x509util"
3735 "google.golang.org/grpc"
@@ -67,7 +65,9 @@ func New(tb testing.TB) *WorkloadAPI {
6765 }
6866
6967 listener , err := newListener ()
70- require .NoError (tb , err )
68+ if err != nil {
69+ tb .Fatalf ("failed to create listener: %v" , err )
70+ }
7171
7272 server := grpc .NewServer ()
7373 workload .RegisterSpiffeWorkloadAPIServer (server , & workloadAPIWrapper {w : w })
@@ -127,7 +127,9 @@ func (w *WorkloadAPI) SetJWTBundles(jwtBundles ...*jwtbundle.Bundle) {
127127 }
128128 for _ , bundle := range jwtBundles {
129129 bundleBytes , err := bundle .Marshal ()
130- assert .NoError (w .tb , err )
130+ if err != nil {
131+ w .tb .Fatalf ("failed to marshal JWT bundle: %v" , err )
132+ }
131133 resp .Bundles [bundle .TrustDomain ().String ()] = bundleBytes
132134 }
133135
@@ -151,9 +153,13 @@ func (w *WorkloadAPI) SetX509Bundles(x509Bundles ...*x509bundle.Bundle) {
151153 }
152154 for _ , bundle := range x509Bundles {
153155 bundleBytes , err := bundle .Marshal ()
154- assert .NoError (w .tb , err )
156+ if err != nil {
157+ w .tb .Fatalf ("failed to marshal X509 bundle: %v" , err )
158+ }
155159 bundlePem , err := pemutil .ParseCertificates (bundleBytes )
156- assert .NoError (w .tb , err )
160+ if err != nil {
161+ w .tb .Fatalf ("failed to parse certificates: %v" , err )
162+ }
157163
158164 var rawBytes []byte
159165 for _ , c := range bundlePem {
@@ -222,7 +228,9 @@ func (r *X509SVIDResponse) ToProto(tb testing.TB) *workload.X509SVIDResponse {
222228 if svid .PrivateKey != nil {
223229 var err error
224230 keyDER , err = x509 .MarshalPKCS8PrivateKey (svid .PrivateKey )
225- require .NoError (tb , err )
231+ if err != nil {
232+ tb .Fatalf ("failed to marshal private key: %v" , err )
233+ }
226234 }
227235 pb .Svids = append (pb .Svids , & workload.X509SVID {
228236 SpiffeId : svid .ID .String (),
@@ -381,7 +389,9 @@ func (w *WorkloadAPI) validateJWTSVID(_ context.Context, req *workload.ValidateJ
381389 return nil , status .Error (codes .InvalidArgument , err .Error ())
382390 }
383391 claims , err := structFromValues (jwtSvid .Claims )
384- require .NoError (w .tb , err )
392+ if err != nil {
393+ w .tb .Fatalf ("failed to convert claims to Struct: %v" , err )
394+ }
385395
386396 return & workload.ValidateJWTSVIDResponse {
387397 SpiffeId : jwtSvid .ID .String (),
0 commit comments