From 750be1e8f6bac2d48b0bec4eef518ba7822e7702 Mon Sep 17 00:00:00 2001 From: PIL Date: Tue, 8 Mar 2022 10:52:45 +0100 Subject: [PATCH] [securityOr] fix simpleTest --- operation.go | 5 +++-- operation_test.go | 16 ++++++++++++++++ 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/operation.go b/operation.go index 948d58bbd..ad3cb6a12 100644 --- a/operation.go +++ b/operation.go @@ -631,7 +631,9 @@ func (operation *Operation) ParseRouterComment(commentLine string) error { // ParseSecurityComment parses comment for given `security` comment string. func (operation *Operation) ParseSecurityComment(commentLine string) error { - var securityMap map[string][]string = map[string][]string{} + //var securityMap map[string][]string = map[string][]string{} + + var securityMap = make(map[string][]string) securitySource := commentLine[strings.Index(commentLine, "@Security")+1:] for _, securityOption := range strings.Split(securitySource, "||") { securityOption = strings.TrimSpace(securityOption) @@ -648,7 +650,6 @@ func (operation *Operation) ParseSecurityComment(commentLine string) error { } else { securityKey := strings.TrimSpace(securityOption) - securityMap := map[string][]string{} securityMap[securityKey] = []string{} } } diff --git a/operation_test.go b/operation_test.go index 6068f3d3e..45598e7db 100644 --- a/operation_test.go +++ b/operation_test.go @@ -2036,6 +2036,22 @@ func TestParseSecurityComment(t *testing.T) { }) } +func TestParseSecurityCommentSimple(t *testing.T) { + t.Parallel() + + comment := `@Security ApiKeyAuth` + operation := NewOperation(nil) + + err := operation.ParseComment(comment, nil) + assert.NoError(t, err) + + assert.Equal(t, operation.Security, []map[string][]string{ + { + "ApiKeyAuth": {}, + }, + }) +} + func TestParseSecurityCommentOr(t *testing.T) { t.Parallel()