mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
1ab6e799c1
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package grpc
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/hanzoai/cd/pkg/apiclient/account"
|
|
)
|
|
|
|
func Test_JSONLogging(t *testing.T) {
|
|
t.Parallel()
|
|
l := logrus.New()
|
|
l.SetFormatter(&logrus.JSONFormatter{})
|
|
var buf bytes.Buffer
|
|
l.SetOutput(&buf)
|
|
entry := logrus.NewEntry(l)
|
|
|
|
c := t.Context()
|
|
req := new(account.CreateTokenRequest)
|
|
req.Name = "create-token-name"
|
|
info := &grpc.UnaryServerInfo{}
|
|
handler := func(_ context.Context, _ any) (any, error) {
|
|
return nil, nil
|
|
}
|
|
decider := func(_ context.Context, _ interceptors.CallMeta) bool {
|
|
return true
|
|
}
|
|
interceptor := PayloadUnaryServerInterceptor(entry, false, decider)
|
|
_, err := interceptor(c, req, info, handler)
|
|
require.NoError(t, err)
|
|
|
|
out := buf.String()
|
|
assert.Contains(t, out, fmt.Sprintf(`"grpc.request.content":{"name":%q`, req.Name))
|
|
}
|
|
|
|
func Test_logRequest(t *testing.T) {
|
|
t.Parallel()
|
|
c := t.Context()
|
|
//nolint:staticcheck
|
|
c = context.WithValue(c, "claims", jwt.MapClaims{"groups": []string{"expected-group-claim"}})
|
|
req := new(account.CreateTokenRequest)
|
|
req.Name = "create-token-name"
|
|
info := &grpc.UnaryServerInfo{}
|
|
handler := func(_ context.Context, _ any) (any, error) {
|
|
return nil, nil
|
|
}
|
|
decider := func(_ context.Context, _ interceptors.CallMeta) bool {
|
|
return true
|
|
}
|
|
|
|
t.Run("with debug enabled, group claims are logged", func(t *testing.T) {
|
|
t.Parallel()
|
|
l := logrus.New()
|
|
l.SetFormatter(&logrus.JSONFormatter{})
|
|
var buf bytes.Buffer
|
|
l.SetOutput(&buf)
|
|
l.SetLevel(logrus.DebugLevel)
|
|
entry := logrus.NewEntry(l)
|
|
|
|
interceptor := PayloadUnaryServerInterceptor(entry, true, decider)
|
|
|
|
_, err := interceptor(c, req, info, handler)
|
|
require.NoError(t, err)
|
|
|
|
out := buf.String()
|
|
assert.Contains(t, out, "expected-group-claim")
|
|
})
|
|
|
|
t.Run("with debug not enabled, group claims aren't logged", func(t *testing.T) {
|
|
t.Parallel()
|
|
l := logrus.New()
|
|
l.SetFormatter(&logrus.JSONFormatter{})
|
|
var buf bytes.Buffer
|
|
l.SetOutput(&buf)
|
|
l.SetLevel(logrus.InfoLevel)
|
|
entry := logrus.NewEntry(l)
|
|
|
|
interceptor := PayloadUnaryServerInterceptor(entry, true, decider)
|
|
|
|
_, err := interceptor(c, req, info, handler)
|
|
require.NoError(t, err)
|
|
|
|
out := buf.String()
|
|
assert.NotContains(t, out, "expected-group-claim")
|
|
})
|
|
}
|