use sql_cli::datatable::{DataTable, DataColumn, DataRow, DataValue};
use sql_cli::datatable_buffer::DataTableBuffer;
use sql_cli::buffer::BufferAPI;
fn main() {
let mut table = DataTable::new("test_table");
table.add_column(DataColumn::new("id"));
table.add_column(DataColumn::new("name"));
table.add_column(DataColumn::new("orderid"));
table.add_column(DataColumn::new("amount"));
table.add_row(DataRow::new(vec![
DataValue::Integer(1),
DataValue::String("Widget A".to_string()),
DataValue::String("ORD001".to_string()),
DataValue::Float(100.50),
]));
let buffer = DataTableBuffer::new(1, table);
let column_names = buffer.get_column_names();
println!("Column names from DataTableBuffer:");
for (idx, name) in column_names.iter().enumerate() {
println!(" Column {}: {}", idx, name);
}
assert_eq!(column_names.len(), 4);
assert_eq!(column_names[0], "id");
assert_eq!(column_names[1], "name");
assert_eq!(column_names[2], "orderid");
assert_eq!(column_names[3], "amount");
println!("\n✓ Column search fix verified!");
println!(" DataTableBuffer now correctly returns column names");
println!(" Column search should now work with CSV files");
}