go: fix ignored page token bug in iterator (googleapis#2716)

busunkim96 · Nov 7, 2019 · 5addb73 · 5addb73
1 parent 35b2686
commit 5addb73
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/main/resources/com/google/api/codegen/go/main.snip b/src/main/resources/com/google/api/codegen/go/main.snip
@@ -270,6 +270,7 @@
         }
         it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)
         it.pageInfo.MaxSize = int(req.PageSize)
+        it.pageInfo.Token = req.PageToken
         return it
     }
 @end

diff --git a/src/test/java/com/google/api/codegen/gapic/testdata/go/go_library.baseline b/src/test/java/com/google/api/codegen/gapic/testdata/go/go_library.baseline
@@ -456,6 +456,7 @@ func (c *LibClient) ListShelves(ctx context.Context, req *librarypb.ListShelvesR
     }
     it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)
     it.pageInfo.MaxSize = int(req.PageSize)
+    it.pageInfo.Token = req.PageToken
     return it
 }
 
@@ -577,6 +578,7 @@ func (c *LibClient) ListBooks(ctx context.Context, req *librarypb.ListBooksReque
     }
     it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)
     it.pageInfo.MaxSize = int(req.PageSize)
+    it.pageInfo.Token = req.PageToken
     return it
 }
 
@@ -661,6 +663,7 @@ func (c *LibClient) ListStrings(ctx context.Context, req *librarypb.ListStringsR
     }
     it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)
     it.pageInfo.MaxSize = int(req.PageSize)
+    it.pageInfo.Token = req.PageToken
     return it
 }
 
@@ -841,6 +844,7 @@ func (c *LibClient) FindRelatedBooks(ctx context.Context, req *librarypb.FindRel
     }
     it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)
     it.pageInfo.MaxSize = int(req.PageSize)
+    it.pageInfo.Token = req.PageToken
     return it
 }