Merge pull request #110 from cycloidio/fg-fix-pagination

Fix: aws pagination error
cycloidio · Jun 2, 2020 · bcbde16 · bcbde16
2 parents 86d8875 + d73ba63
commit bcbde16
Show file tree

Hide file tree

Showing 3 changed files with 174 additions and 0 deletions.
diff --git a/aws/cmd/template.go b/aws/cmd/template.go
@@ -80,6 +80,9 @@ const (
 				{{ if .HasNotPagination }}
 					hasNextToken = false
 				{{ else }}
+					if input == nil {
+						input = &{{.Input}}{}
+					}
 					input.{{.InputPaginationAttributeFn}} = o.{{.PaginationAttributeFn}}
 					hasNextToken = o.{{.PaginationAttributeFn}} != nil
 				{{ end }}

diff --git a/aws/cmd/template_test.go b/aws/cmd/template_test.go
@@ -160,6 +160,9 @@ func TestTemplateExecute(t *testing.T) {
 					if err != nil {
 						return nil, err
 					}
+					if input == nil {
+						input = &Service.PrefixEntitiesInput{}
+					}
 					input.NextToken = o.NextToken
 					hasNextToken = o.NextToken != nil
 
@@ -197,6 +200,9 @@ func TestTemplateExecute(t *testing.T) {
 					if err != nil {
 						return nil, err
 					}
+					if input == nil {
+						input = &Service.PrefixEntitiesInput{}
+					}
 					input.NextToken = o.NextToken
 					hasNextToken = o.NextToken != nil