use the http::HeaderMap instead of dealing with string silliness

MystenLabs · Nov 22, 2024 · 2f0e074 · 2f0e074
1 parent dc45d8c
commit 2f0e074
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/sui-graphql-e2e-tests/tests/tests.rs b/crates/sui-graphql-e2e-tests/tests/tests.rs
@@ -50,7 +50,7 @@ impl OffchainStateReader for OffchainReaderForAdapter {
             .await?;
 
         Ok(TestResponse {
-            http_headers: Some(format!("{:#?}", result.http_headers_without_date())),
+            http_headers: Some(result.http_headers_without_date()),
             response_body: result.response_body_json_pretty(),
             service_version: result.graphql_version().ok(),
         })

diff --git a/crates/sui-transactional-test-runner/Cargo.toml b/crates/sui-transactional-test-runner/Cargo.toml
@@ -16,6 +16,7 @@ bcs.workspace = true
 bimap.workspace = true
 clap.workspace = true
 eyre.workspace = true
+http.workspace = true
 once_cell.workspace = true
 rand.workspace = true
 regex.workspace = true

diff --git a/crates/sui-transactional-test-runner/src/offchain_state.rs b/crates/sui-transactional-test-runner/src/offchain_state.rs
@@ -6,7 +6,7 @@ use std::time::Duration;
 
 pub struct TestResponse {
     pub response_body: String,
-    pub http_headers: Option<String>,
+    pub http_headers: Option<http::HeaderMap>,
     pub service_version: Option<String>,
 }
 

diff --git a/crates/sui-transactional-test-runner/src/test_adapter.rs b/crates/sui-transactional-test-runner/src/test_adapter.rs
@@ -631,7 +631,7 @@ impl<'a> MoveTestAdapter<'a> for SuiTestAdapter {
 
                 let mut output = vec![];
                 if show_headers {
-                    output.push(format!("Headers: {:#?}", resp.http_headers));
+                    output.push(format!("Headers: {:#?}", resp.http_headers.unwrap()));
                 }
                 if show_service_version {
                     output.push(format!(