From a93881039b22500fdd420779adceb44040a7a83c Mon Sep 17 00:00:00 2001 From: Takanori Hayashi Date: Sun, 22 Jun 2025 14:22:22 +0900 Subject: [PATCH] Update network driver test --- src/driver/network/driver.rs | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/driver/network/driver.rs b/src/driver/network/driver.rs index c8f19d3..e8c09ee 100644 --- a/src/driver/network/driver.rs +++ b/src/driver/network/driver.rs @@ -177,25 +177,32 @@ mod tests { statement.execute_update("insert into test (A, B) values (2, NULL)")?; statement.execute_update("insert into test (A, B) values (3, 'b')")?; - let mut result_set = statement.execute_query("select B from test")?; + let mut result_set = statement.execute_query("select A, B from test")?; let metadata = result_set.get_metadata()?; - assert_eq!(metadata.get_column_count()?, 1); - assert_eq!(metadata.get_column_name(0)?, "B"); + assert_eq!(metadata.get_column_count()?, 2); + assert_eq!(metadata.get_column_name(0)?, "A"); + assert_eq!(metadata.get_column_name(1)?, "B"); assert!(result_set.next()?); + assert_eq!(result_set.get_i32("A")?, Some(1)); assert_eq!(result_set.get_string("B")?, Some("a".to_string())); assert!(result_set.next()?); + assert_eq!(result_set.get_i32("A")?, Some(2)); assert_eq!(result_set.get_string("B")?, None); assert!(result_set.next()?); + assert_eq!(result_set.get_i32("A")?, Some(3)); assert_eq!(result_set.get_string("B")?, Some("b".to_string())); assert!(!result_set.next()?); result_set.after_last()?; assert!(result_set.previous()?); + assert_eq!(result_set.get_i32("A")?, Some(3)); assert_eq!(result_set.get_string("B")?, Some("b".to_string())); assert!(result_set.previous()?); + assert_eq!(result_set.get_i32("A")?, Some(2)); assert_eq!(result_set.get_string("B")?, None); assert!(result_set.previous()?); + assert_eq!(result_set.get_i32("A")?, Some(1)); assert_eq!(result_set.get_string("B")?, Some("a".to_string())); assert!(!result_set.previous()?); result_set.close()?;