diff --git a/src/data/bigquery_builder.rs b/src/data/bigquery_builder.rs index 6d400da..b29ca9c 100644 --- a/src/data/bigquery_builder.rs +++ b/src/data/bigquery_builder.rs @@ -132,6 +132,13 @@ impl<'a, Table> BigQueryBuilder<'a, Table> //region add content + pub fn set_limit(self, limit: usize) -> Self { + Self { + limit: Some(limit), + ..self + } + } + fn set_select_fields(self, fields: Vec) -> Result { //TODO: this method probably does not work since the logic does // not work if (at least the required) fields are not selected diff --git a/tests/tests.rs b/tests/tests.rs index 51abb8f..80bb960 100644 --- a/tests/tests.rs +++ b/tests/tests.rs @@ -99,6 +99,15 @@ async fn simple_query() { last_num = line.row_id; } } +#[tokio::test] +async fn test_select_limit_1() { + init_logger(); + let client = get_test_client().await; + let q = DbInfos::query(&client) + .set_limit(1) + .run().await.unwrap(); + assert_eq!(q.len(), 1); +} fn init_logger() { let _ = env_logger::builder()