WithRoles variadic args (#1395)
Co-authored-by: Ben Toogood <ben@micro.mu>
This commit is contained in:
parent
c1978265ab
commit
86272a3064
@ -85,7 +85,7 @@ func WithMetadata(md map[string]string) GenerateOption {
|
||||
}
|
||||
|
||||
// WithRoles for the generated account
|
||||
func WithRoles(rs []string) GenerateOption {
|
||||
func WithRoles(rs ...string) GenerateOption {
|
||||
return func(o *GenerateOptions) {
|
||||
o.Roles = rs
|
||||
}
|
||||
|
@ -68,7 +68,7 @@ func (s *Store) Generate(id string, opts ...auth.GenerateOption) (*auth.Account,
|
||||
secretOpts := []token.GenerateOption{
|
||||
token.WithExpiry(options.SecretExpiry),
|
||||
token.WithMetadata(options.Metadata),
|
||||
token.WithRoles(options.Roles),
|
||||
token.WithRoles(options.Roles...),
|
||||
}
|
||||
secret, err := s.secretProvider.Generate(id, secretOpts...)
|
||||
if err != nil {
|
||||
@ -166,6 +166,6 @@ func (s *Store) Refresh(secret string, opts ...auth.RefreshOption) (*auth.Token,
|
||||
return s.tokenProvider.Generate(sec.Subject,
|
||||
token.WithExpiry(options.TokenExpiry),
|
||||
token.WithMetadata(sec.Metadata),
|
||||
token.WithRoles(sec.Roles),
|
||||
token.WithRoles(sec.Roles...),
|
||||
)
|
||||
}
|
||||
|
@ -17,7 +17,7 @@ func TestGenerate(t *testing.T) {
|
||||
metadata := map[string]string{"foo": "bar"}
|
||||
|
||||
opts := []auth.GenerateOption{
|
||||
auth.WithRoles(roles),
|
||||
auth.WithRoles(roles...),
|
||||
auth.WithMetadata(metadata),
|
||||
}
|
||||
|
||||
@ -100,7 +100,7 @@ func TestInspect(t *testing.T) {
|
||||
metadata := map[string]string{"foo": "bar"}
|
||||
|
||||
opts := []auth.GenerateOption{
|
||||
auth.WithRoles(roles),
|
||||
auth.WithRoles(roles...),
|
||||
auth.WithMetadata(metadata),
|
||||
}
|
||||
|
||||
@ -146,7 +146,7 @@ func TestRefresh(t *testing.T) {
|
||||
metadata := map[string]string{"foo": "bar"}
|
||||
|
||||
opts := []auth.GenerateOption{
|
||||
auth.WithRoles(roles),
|
||||
auth.WithRoles(roles...),
|
||||
auth.WithMetadata(metadata),
|
||||
}
|
||||
|
||||
|
@ -37,7 +37,7 @@ func TestInspect(t *testing.T) {
|
||||
|
||||
opts := []token.GenerateOption{
|
||||
token.WithMetadata(md),
|
||||
token.WithRoles(roles),
|
||||
token.WithRoles(roles...),
|
||||
}
|
||||
|
||||
tok, err := b.Generate(subject, opts...)
|
||||
|
@ -46,7 +46,7 @@ func TestInspect(t *testing.T) {
|
||||
|
||||
opts := []token.GenerateOption{
|
||||
token.WithMetadata(md),
|
||||
token.WithRoles(roles),
|
||||
token.WithRoles(roles...),
|
||||
}
|
||||
|
||||
tok, err := j.Generate(subject, opts...)
|
||||
|
@ -76,7 +76,7 @@ func WithMetadata(md map[string]string) func(o *GenerateOptions) {
|
||||
}
|
||||
|
||||
// WithRoles for the token
|
||||
func WithRoles(rs []string) func(o *GenerateOptions) {
|
||||
func WithRoles(rs ...string) func(o *GenerateOptions) {
|
||||
return func(o *GenerateOptions) {
|
||||
o.Roles = rs
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user