mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-28 18:47:46 +00:00
3549a8b64e
* move mgmt pkg * begin package restructure * rename auth package to authz * begin start api * move auth * move admin * fix merge * configs and interceptors * interceptor * revert generate-grpc.sh * some cleanups * console * move console * fix tests and merging * js linting * merge * merging and configs * change k8s base to current ports * fixes * cleanup * regenerate proto * remove unnecessary whitespace * missing param * go mod tidy * fix merging * move login pkg * cleanup * move api pkgs again * fix pkg naming * fix generate-static.sh for login * update workflow * fixes * logging * remove duplicate * comment for optional gateway interfaces * regenerate protos * fix proto imports for grpc web * protos * grpc web generate * grpc web generate * fix changes * add translation interceptor * fix merging * regenerate mgmt proto
93 lines
1.7 KiB
Go
93 lines
1.7 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
func TestGetHeader(t *testing.T) {
|
|
type args struct {
|
|
ctx context.Context
|
|
headername string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want string
|
|
}{
|
|
{
|
|
"empty context",
|
|
args{
|
|
ctx: context.Background(),
|
|
headername: "header",
|
|
},
|
|
"",
|
|
},
|
|
{
|
|
"context without header",
|
|
args{
|
|
ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs("header2", "value")),
|
|
headername: "header",
|
|
},
|
|
"",
|
|
},
|
|
{
|
|
"context with header",
|
|
args{
|
|
ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs("header", "value")),
|
|
headername: "header",
|
|
},
|
|
"value",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := GetHeader(tt.args.ctx, tt.args.headername); got != tt.want {
|
|
t.Errorf("GetHeader() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetAuthorizationHeader(t *testing.T) {
|
|
type args struct {
|
|
ctx context.Context
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want string
|
|
}{
|
|
{
|
|
"empty context",
|
|
args{
|
|
ctx: context.Background(),
|
|
},
|
|
"",
|
|
},
|
|
{
|
|
"context without header",
|
|
args{
|
|
ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs("header", "value")),
|
|
},
|
|
"",
|
|
},
|
|
{
|
|
"context with header",
|
|
args{
|
|
ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs("authorization", "value")),
|
|
},
|
|
"value",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := GetAuthorizationHeader(tt.args.ctx); got != tt.want {
|
|
t.Errorf("GetAuthorizationHeader() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|