add POST query test

Author: ryansmith3136

src/main.rs

@@ -226,7 +226,9 @@ fn service(config: api::Config) -> IntoMakeServiceWithConnectInfo<Router, Socket
 
 #[cfg(test)]
 mod tests {
+    use alloy::primitives::{Bytes, B256, U64};
     use axum_test::TestServer;
+    use serde_json::json;
 
     use super::*;
     use crate::pg;
@@ -238,4 +240,37 @@ mod tests {
         let server = TestServer::new(service(config)).unwrap();
         server.get("/").await.assert_text_contains("hello");
     }
+
+    #[tokio::test]
+    async fn test_query_post_with_params() {
+        let (_pg_server, pool) = pg::test_utils::test_pg().await;
+        let pg = pool.get().await.expect("unable to get pg from pool");
+        pg::test_utils::insert(
+            &pg,
+            vec![pg::test_utils::Log {
+                chain: U64::from(1),
+                block_num: U64::from(1),
+                topics: vec![B256::with_last_byte(0x42)],
+                data: Bytes::from_static(&[0x42]),
+            }],
+        )
+        .await;
+        let config = api::Config::new(pool, None);
+        let server = TestServer::new(service(config)).unwrap();
+        let request = vec![api_sql::Request {
+            block_height: None,
+            event_signatures: vec![],
+            query: String::from("select block_num from logs"),
+        }];
+        server
+            .post("/query")
+            .add_query_param("api-key", "face")
+            .add_query_param("chain", "1")
+            .json(&request)
+            .await
+            .assert_json(&json!({
+                "block_height": 1,
+                "result": [[["block_num"],[1]]]
+            }));
+    }
 }

src/pg.rs

@@ -24,9 +24,34 @@ pub fn new_pool(url: &str, size: usize) -> Result<Pool> {
 
 #[cfg(test)]
 pub mod test_utils {
+    use alloy::primitives::{Bytes, B256, U64};
     use deadpool_postgres::Pool;
     use postgresql_embedded::{PostgreSQL, Settings, Version};
-    use tokio_postgres::NoTls;
+    use tokio_postgres::{Client, NoTls};
+
+    pub struct Log {
+        pub chain: U64,
+        pub block_num: U64,
+        pub topics: Vec<B256>,
+        pub data: Bytes,
+    }
+
+    pub async fn insert(pg: &Client, logs: Vec<Log>) {
+        for log in logs {
+            pg.execute(
+                r#"insert into blocks(chain, num, hash) values ($1, $2, '\xface')"#,
+                &[&log.chain, &log.block_num],
+            )
+            .await
+            .unwrap();
+            pg.execute(
+                "insert into logs(chain, block_num, topics, data) values ($1, $2, $3, $4)",
+                &[&log.chain, &log.block_num, &log.topics, &log.data.to_vec()],
+            )
+            .await
+            .unwrap();
+        }
+    }
 
     pub async fn test_pg() -> (PostgreSQL, Pool) {
         let pg_settings = Settings {