add tests to check support for querying all columns including project_id column

Author: tonyalaribe

Cargo.lock

@@ -29,7 +29,9 @@ bytes = "1.4"
 tokio-rustls = "0.26.1"
 sled = "0.34.7"
 actix-web = "4.9.0"
-datafusion-postgres = { git = "https://github.com/apitoolkit/FusionGate.git" }
+# datafusion-postgres = { git = "https://github.com/apitoolkit/FusionGate.git" }
+datafusion-postgres = { git = "https://github.com/apitoolkit/datafusion-postgres.git", branch = "interpret-dictionary-types" }
+# datafusion-postgres = { path = "../datafusion-projects/datafusion-postgres/datafusion-postgres/" }
 datafusion-functions-json = "0.46.0"
 anyhow = "1.0.95"
 tokio-util = "0.7.13"

Cargo.toml

@@ -147,8 +147,11 @@ mod integration {
             let count_rows = client.query("SELECT COUNT(*) FROM otel_logs_and_spans WHERE project_id = $1", &[&"test_project"]).await?;
             assert_eq!(count_rows[0].get::<_, i64>(0), 6, "Should have a total of 6 records (1 initial + 5 batch)");
 
+            let count_rows = client.query("SELECT project_id FROM otel_logs_and_spans WHERE project_id = $1", &[&"test_project"]).await?;
+            assert_eq!(count_rows[0].get::<_, String>(0), "test_project", "project_id should match");
+
             let count_rows = client.query("SELECT * FROM otel_logs_and_spans WHERE project_id = $1", &[&"test_project"]).await?;
-            assert_eq!(count_rows[0].get::<_, i64>(0), 6, "Should have a total of 6 records (1 initial + 5 batch)");
+            assert_eq!(count_rows[0].columns().len(), 80, "Should return all 80 columns");
 
             Ok::<_, tokio_postgres::Error>(())
         }