Simplify binary engine runner

prisma · Dec 15, 2022 · 7c8aaac · 7c8aaac
1 parent de57ec0
commit 7c8aaac
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/query-engine/connector-test-kit-rs/query-tests-setup/src/runner/mod.rs b/query-engine/connector-test-kit-rs/query-tests-setup/src/runner/mod.rs
@@ -2,6 +2,8 @@ mod binary;
 mod direct;
 mod node_api;
 
+use std::collections::HashMap;
+
 pub use binary::*;
 pub use direct::*;
 use hyper::HeaderMap;
@@ -163,16 +165,27 @@ impl Runner {
         }
     }
 
-    pub fn set_headers(&mut self, headers: Option<HeaderMap>) {
+    pub fn set_headers(&mut self, headers: HashMap<String, String>) {
         match &mut self.inner {
             RunnerType::Direct(_) => todo!(),
             RunnerType::NodeApi(_) => todo!(),
-            RunnerType::Binary(r) => r.set_additional_headers(headers),
+            RunnerType::Binary(r) => {
+                let map = Some(HeaderMap::from_iter(
+                    headers
+                        .into_iter()
+                        .map(|(k, v)| (k.parse().unwrap(), v.parse().unwrap())),
+                ));
+                r.set_additional_headers(map)
+            }
         }
     }
 
     pub fn clear_headers(&mut self) {
-        self.set_headers(None)
+        match &mut self.inner {
+            RunnerType::Direct(_) => todo!(),
+            RunnerType::NodeApi(_) => todo!(),
+            RunnerType::Binary(r) => r.set_additional_headers(None),
+        }
     }
 
     async fn direct(datamodel: String, connector_tag: ConnectorTag, metrics: MetricRegistry) -> TestResult<RunnerType> {

diff --git a/query-engine/query-engine/src/server/mod.rs b/query-engine/query-engine/src/server/mod.rs
@@ -143,6 +143,7 @@ async fn graphql_handler(state: State, req: Request<Body>) -> Result<Response<Bo
                         "result": result,
                         "traces": traces,
                     });
+                    dbg!(&json);
                     serde_json::to_vec(&json).unwrap()
                 } else {
                     serde_json::to_vec(&result).unwrap()