test(bigquery/v2): update integration test with pagination

src/generated/cloud/bigquery/v2/src/builder.rs

@@ -558,9 +558,13 @@ pub mod dataset_service {
     /// # use google_cloud_bigquery_v2::builder;
     /// use builder::dataset_service::ListDatasets;
     /// # tokio_test::block_on(async {
+    /// use gax::paginator::ItemPaginator;
     ///
     /// let builder = prepare_request_builder();
-    /// let response = builder.send().await?;
+    /// let mut items = builder.by_item();
+    /// while let Some(result) = items.next().await {
+    ///   let item = result?;
+    /// }
     /// # gax::Result::<()>::Ok(()) });
     ///
     /// fn prepare_request_builder() -> ListDatasets {
@@ -598,6 +602,29 @@ pub mod dataset_service {
                 .map(gax::response::Response::into_body)
         }
 
+        /// Streams each page in the collection.
+        pub fn by_page(
+            self,
+        ) -> impl gax::paginator::Paginator<crate::model::DatasetList, gax::error::Error> {
+            use std::clone::Clone;
+            let token = self.0.request.page_token.clone();
+            let execute = move |token: String| {
+                let mut builder = self.clone();
+                builder.0.request = builder.0.request.set_page_token(token);
+                builder.send()
+            };
+            gax::paginator::internal::new_paginator(token, execute)
+        }
+
+        /// Streams each item in the collection.
+        pub fn by_item(
+            self,
+        ) -> impl gax::paginator::ItemPaginator<crate::model::DatasetList, gax::error::Error>
+        {
+            use gax::paginator::Paginator;
+            self.by_page().items()
+        }
+
         /// Sets the value of [project_id][crate::model::ListDatasetsRequest::project_id].
         ///
         /// This is a **required** field for requests.
@@ -1114,9 +1141,13 @@ pub mod job_service {
     /// # use google_cloud_bigquery_v2::builder;
     /// use builder::job_service::ListJobs;
     /// # tokio_test::block_on(async {
+    /// use gax::paginator::ItemPaginator;
     ///
     /// let builder = prepare_request_builder();
-    /// let response = builder.send().await?;
+    /// let mut items = builder.by_item();
+    /// while let Some(result) = items.next().await {
+    ///   let item = result?;
+    /// }
     /// # gax::Result::<()>::Ok(()) });
     ///
     /// fn prepare_request_builder() -> ListJobs {
@@ -1154,6 +1185,28 @@ pub mod job_service {
                 .map(gax::response::Response::into_body)
         }
 
+        /// Streams each page in the collection.
+        pub fn by_page(
+            self,
+        ) -> impl gax::paginator::Paginator<crate::model::JobList, gax::error::Error> {
+            use std::clone::Clone;
+            let token = self.0.request.page_token.clone();
+            let execute = move |token: String| {
+                let mut builder = self.clone();
+                builder.0.request = builder.0.request.set_page_token(token);
+                builder.send()
+            };
+            gax::paginator::internal::new_paginator(token, execute)
+        }
+
+        /// Streams each item in the collection.
+        pub fn by_item(
+            self,
+        ) -> impl gax::paginator::ItemPaginator<crate::model::JobList, gax::error::Error> {
+            use gax::paginator::Paginator;
+            self.by_page().items()
+        }
+
         /// Sets the value of [project_id][crate::model::ListJobsRequest::project_id].
         pub fn set_project_id<T: Into<std::string::String>>(mut self, v: T) -> Self {
             self.0.request.project_id = v.into();
@@ -1628,9 +1681,13 @@ pub mod model_service {
     /// # use google_cloud_bigquery_v2::builder;
     /// use builder::model_service::ListModels;
     /// # tokio_test::block_on(async {
+    /// use gax::paginator::ItemPaginator;
     ///
     /// let builder = prepare_request_builder();
-    /// let response = builder.send().await?;
+    /// let mut items = builder.by_item();
+    /// while let Some(result) = items.next().await {
+    ///   let item = result?;
+    /// }
     /// # gax::Result::<()>::Ok(()) });
     ///
     /// fn prepare_request_builder() -> ListModels {
@@ -1668,6 +1725,30 @@ pub mod model_service {
                 .map(gax::response::Response::into_body)
         }
 
+        /// Streams each page in the collection.
+        pub fn by_page(
+            self,
+        ) -> impl gax::paginator::Paginator<crate::model::ListModelsResponse, gax::error::Error>
+        {
+            use std::clone::Clone;
+            let token = self.0.request.page_token.clone();
+            let execute = move |token: String| {
+                let mut builder = self.clone();
+                builder.0.request = builder.0.request.set_page_token(token);
+                builder.send()
+            };
+            gax::paginator::internal::new_paginator(token, execute)
+        }
+
+        /// Streams each item in the collection.
+        pub fn by_item(
+            self,
+        ) -> impl gax::paginator::ItemPaginator<crate::model::ListModelsResponse, gax::error::Error>
+        {
+            use gax::paginator::Paginator;
+            self.by_page().items()
+        }
+
         /// Sets the value of [project_id][crate::model::ListModelsRequest::project_id].
         ///
         /// This is a **required** field for requests.
@@ -2437,9 +2518,13 @@ pub mod routine_service {
     /// # use google_cloud_bigquery_v2::builder;
     /// use builder::routine_service::ListRoutines;
     /// # tokio_test::block_on(async {
+    /// use gax::paginator::ItemPaginator;
     ///
     /// let builder = prepare_request_builder();
-    /// let response = builder.send().await?;
+    /// let mut items = builder.by_item();
+    /// while let Some(result) = items.next().await {
+    ///   let item = result?;
+    /// }
     /// # gax::Result::<()>::Ok(()) });
     ///
     /// fn prepare_request_builder() -> ListRoutines {
@@ -2477,6 +2562,30 @@ pub mod routine_service {
                 .map(gax::response::Response::into_body)
         }
 
+        /// Streams each page in the collection.
+        pub fn by_page(
+            self,
+        ) -> impl gax::paginator::Paginator<crate::model::ListRoutinesResponse, gax::error::Error>
+        {
+            use std::clone::Clone;
+            let token = self.0.request.page_token.clone();
+            let execute = move |token: String| {
+                let mut builder = self.clone();
+                builder.0.request = builder.0.request.set_page_token(token);
+                builder.send()
+            };
+            gax::paginator::internal::new_paginator(token, execute)
+        }
+
+        /// Streams each item in the collection.
+        pub fn by_item(
+            self,
+        ) -> impl gax::paginator::ItemPaginator<crate::model::ListRoutinesResponse, gax::error::Error>
+        {
+            use gax::paginator::Paginator;
+            self.by_page().items()
+        }
+
         /// Sets the value of [project_id][crate::model::ListRoutinesRequest::project_id].
         ///
         /// This is a **required** field for requests.
@@ -3793,9 +3902,13 @@ pub mod table_service {
     /// # use google_cloud_bigquery_v2::builder;
     /// use builder::table_service::ListTables;
     /// # tokio_test::block_on(async {
+    /// use gax::paginator::ItemPaginator;
     ///
     /// let builder = prepare_request_builder();
-    /// let response = builder.send().await?;
+    /// let mut items = builder.by_item();
+    /// while let Some(result) = items.next().await {
+    ///   let item = result?;
+    /// }
     /// # gax::Result::<()>::Ok(()) });
     ///
     /// fn prepare_request_builder() -> ListTables {
@@ -3833,6 +3946,29 @@ pub mod table_service {
                 .map(gax::response::Response::into_body)
         }
 
+        /// Streams each page in the collection.
+        pub fn by_page(
+            self,
+        ) -> impl gax::paginator::Paginator<crate::model::TableList, gax::error::Error> {
+            use std::clone::Clone;
+            let token = self.0.request.page_token.clone();
+            let execute = move |token: String| {
+                let mut builder = self.clone();
+                builder.0.request = builder.0.request.set_page_token(token);
+                builder.send()
+            };
+            gax::paginator::internal::new_paginator(token, execute)
+        }
+
+        /// Streams each item in the collection.
+        pub fn by_item(
+            self,
+        ) -> impl gax::paginator::ItemPaginator<crate::model::TableList, gax::error::Error>
+        {
+            use gax::paginator::Paginator;
+            self.by_page().items()
+        }
+
         /// Sets the value of [project_id][crate::model::ListTablesRequest::project_id].
         ///
         /// This is a **required** field for requests.

src/generated/cloud/bigquery/v2/src/model.rs

@@ -5930,6 +5930,20 @@ impl wkt::message::Message for DatasetList {
     }
 }
 
+#[doc(hidden)]
+impl gax::paginator::internal::PageableResponse for DatasetList {
+    type PageItem = crate::model::ListFormatDataset;
+
+    fn items(self) -> std::vec::Vec<Self::PageItem> {
+        self.datasets
+    }
+
+    fn next_page_token(&self) -> std::string::String {
+        use std::clone::Clone;
+        self.next_page_token.clone()
+    }
+}
+
 #[doc(hidden)]
 impl<'de> serde::de::Deserialize<'de> for DatasetList {
     fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
@@ -14465,6 +14479,20 @@ impl wkt::message::Message for JobList {
     }
 }
 
+#[doc(hidden)]
+impl gax::paginator::internal::PageableResponse for JobList {
+    type PageItem = crate::model::ListFormatJob;
+
+    fn items(self) -> std::vec::Vec<Self::PageItem> {
+        self.jobs
+    }
+
+    fn next_page_token(&self) -> std::string::String {
+        use std::clone::Clone;
+        self.next_page_token.clone()
+    }
+}
+
 #[doc(hidden)]
 impl<'de> serde::de::Deserialize<'de> for JobList {
     fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
@@ -65659,6 +65687,20 @@ impl wkt::message::Message for ListModelsResponse {
     }
 }
 
+#[doc(hidden)]
+impl gax::paginator::internal::PageableResponse for ListModelsResponse {
+    type PageItem = crate::model::Model;
+
+    fn items(self) -> std::vec::Vec<Self::PageItem> {
+        self.models
+    }
+
+    fn next_page_token(&self) -> std::string::String {
+        use std::clone::Clone;
+        self.next_page_token.clone()
+    }
+}
+
 #[doc(hidden)]
 impl<'de> serde::de::Deserialize<'de> for ListModelsResponse {
     fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
@@ -74653,6 +74695,20 @@ impl wkt::message::Message for ListRoutinesResponse {
     }
 }
 
+#[doc(hidden)]
+impl gax::paginator::internal::PageableResponse for ListRoutinesResponse {
+    type PageItem = crate::model::Routine;
+
+    fn items(self) -> std::vec::Vec<Self::PageItem> {
+        self.routines
+    }
+
+    fn next_page_token(&self) -> std::string::String {
+        use std::clone::Clone;
+        self.next_page_token.clone()
+    }
+}
+
 #[doc(hidden)]
 impl<'de> serde::de::Deserialize<'de> for ListRoutinesResponse {
     fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
@@ -85398,6 +85454,20 @@ impl wkt::message::Message for TableList {
     }
 }
 
+#[doc(hidden)]
+impl gax::paginator::internal::PageableResponse for TableList {
+    type PageItem = crate::model::ListFormatTable;
+
+    fn items(self) -> std::vec::Vec<Self::PageItem> {
+        self.tables
+    }
+
+    fn next_page_token(&self) -> std::string::String {
+        use std::clone::Clone;
+        self.next_page_token.clone()
+    }
+}
+
 #[doc(hidden)]
 impl<'de> serde::de::Deserialize<'de> for TableList {
     fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>