diff --git a/src/data/bigquery_builder.rs b/src/data/bigquery_builder.rs index 54840f4..6d400da 100644 --- a/src/data/bigquery_builder.rs +++ b/src/data/bigquery_builder.rs @@ -76,14 +76,14 @@ impl<'a, Table> BigQueryBuilder<'a, Table> } let query_res = query_res.rows.unwrap(); - println!("query_res: {:?}", query_res); + log::debug!("query_res: {:?}", query_res); let mut result: Vec = Vec::new(); for row in query_res { let row = row.f.unwrap(); let mut row_data: HashMap = HashMap::new(); for (i, field) in row.into_iter().enumerate() { let field = field.v.unwrap_or(Value::Null); - println!("{}: {}", fields[i], field); + log::debug!("{}: {}", fields[i], field); row_data.insert(fields[i].clone(), field); } let data = Table::new_from_query_result_row(client, &row_data)?; @@ -98,7 +98,6 @@ impl<'a, Table> BigQueryBuilder<'a, Table> query: Some(self.build_query_string()), query_parameters: Some(self.required_params), use_legacy_sql: Some(false), - /*TODO: is this line needed?: use_legacy_sql: Some(false),*/ ..Default::default() } } @@ -202,7 +201,7 @@ impl<'a, Table> BigQueryBuilder<'a, Table> .collect() } }; - println!("fields: {:?}", fields); + log::debug!("fields: {:?}", fields); fields.sort(); fields } diff --git a/tests/tests.rs b/tests/tests.rs index 6d3fc1b..51abb8f 100644 --- a/tests/tests.rs +++ b/tests/tests.rs @@ -22,17 +22,17 @@ pub struct DbInfos<'a> { #[tokio::test] async fn test_get_table_name() { - init_logger(LevelFilter::Debug); + init_logger(); let pk = DbInfos::get_table_name(); - println!("table name: {}", pk); + log::debug!("table name: {}", pk); assert_eq!("Infos", pk, "table name is not correct") } #[tokio::test] async fn test_get_query_fields() { - init_logger(LevelFilter::Debug); + init_logger(); let fields = DbInfos::get_query_fields(true); - println!("fields: {:?}", fields); + log::debug!("fields: {:?}", fields); assert_eq!(6, fields.len(), "fields length is not correct"); assert_eq!("Id", fields.get("row_id").unwrap(), ); assert_eq!("info1", fields.get("info1").unwrap(), ); @@ -44,7 +44,7 @@ async fn test_get_query_fields() { #[tokio::test] async fn test_query_builder_1() { - init_logger(LevelFilter::Debug); + init_logger(); let client = get_test_client().await; let query_builder: BigQueryBuilder = DbInfos::query(&client); let query_builder: BigQueryBuilder = query_builder @@ -60,9 +60,9 @@ async fn test_query_builder_1() { WHERE info1 is NULL AND info3 = @__PARAM_0 \ ORDER BY info ASC LIMIT 1000", ); - println!("query : {}", query_string); - println!("expected: {}", expected_query_string); - println!("request: {:?}", query_builder.clone().build_query_request()); + log::debug!("query : {}", query_string); + log::debug!("expected: {}", expected_query_string); + log::debug!("request: {:?}", query_builder.clone().build_query_request()); assert_eq!(query_string, expected_query_string); assert_eq!( @@ -75,7 +75,7 @@ async fn test_query_builder_1() { 1 ); let res = query_builder.clone().run().await.unwrap(); - println!("res: {:?}", res); + log::debug!("res: {:?}", res); } async fn get_test_client() -> BigqueryClient { @@ -86,7 +86,7 @@ async fn get_test_client() -> BigqueryClient { #[tokio::test] async fn simple_query() { - init_logger(LevelFilter::Info); + init_logger(); let client = get_test_client().await; let q = DbInfos::query(&client) .add_order_by(name_of!(row_id in DbInfos), OrderDirection::Descending) @@ -100,9 +100,12 @@ async fn simple_query() { } } -fn init_logger(level: LevelFilter) { +fn init_logger() { let _ = env_logger::builder() .is_test(true) - .filter_level(level) + .filter_level(LevelFilter::Info) + .filter_module("google_bigquery_v2", LevelFilter::Trace) + .filter_module("google_bigquery_v2_derive", LevelFilter::Trace) + .filter_module("tests", LevelFilter::Trace) .try_init(); }