Skip to content

Commit 03c7e5e

Browse files
committed
fix #8775:Remove testify usage from pkg/spire/
1 parent e91bbdf commit 03c7e5e

File tree

2 files changed

+33
-14
lines changed

2 files changed

+33
-14
lines changed

pkg/spire/test/fakebundleendpoint/server.go

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ import (
2929
"github.com/spiffe/go-spiffe/v2/bundle/spiffebundle"
3030
"github.com/spiffe/go-spiffe/v2/spiffetls/tlsconfig"
3131
"github.com/spiffe/go-spiffe/v2/svid/x509svid"
32-
"github.com/stretchr/testify/assert"
3332
"github.com/tektoncd/pipeline/pkg/spire/test"
3433
"github.com/tektoncd/pipeline/pkg/spire/test/x509util"
3534
)
@@ -82,7 +81,9 @@ func New(tb testing.TB, option ...ServerOption) *Server {
8281

8382
func (s *Server) Shutdown() {
8483
err := s.httpServer.Shutdown(context.Background())
85-
assert.NoError(s.tb, err)
84+
if err!=nil {
85+
s.tb.Errorf("unexpected error: %v", err)
86+
}
8687
s.wg.Wait()
8788
}
8889

@@ -109,7 +110,9 @@ func (s *Server) start() error {
109110
s.wg.Add(1)
110111
go func() {
111112
err := s.httpServer.ServeTLS(ln, "", "")
112-
assert.EqualError(s.tb, err, http.ErrServerClosed.Error())
113+
if err != nil || err.Error()!=http.ErrServerClosed.Error(){
114+
s.tb.Errorf("expected error %q, got %v",http.ErrServerClosed.Error(),err)
115+
}
113116
s.wg.Done()
114117
ln.Close()
115118
}()
@@ -123,12 +126,18 @@ func (s *Server) testbundle(w http.ResponseWriter, r *http.Request) {
123126
}
124127

125128
bb, err := s.bundles[0].Marshal()
126-
assert.NoError(s.tb, err)
129+
if err != nil {
130+
s.tb.Errorf("unexpected error: %v", err)
131+
}
127132
s.bundles = s.bundles[1:]
128133
w.Header().Add("Content-Type", "application/json")
129134
b, err := w.Write(bb)
130-
assert.NoError(s.tb, err)
131-
assert.Equal(s.tb, len(bb), b)
135+
if err != nil {
136+
s.tb.Errorf("unexpected error: %v", err)
137+
}
138+
if len(bb) != b {
139+
s.tb.Errorf("expected written bytes %d, got %d", len(bb), b)
140+
}
132141
}
133142

134143
type serverOption func(*Server)

pkg/spire/test/fakeworkloadapi/workload_api.go

Lines changed: 18 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)