address comments

DataDog · Jul 8, 2024 · f12fd00 · f12fd00
1 parent 00791c3
commit f12fd00
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/trace-utils/Cargo.toml b/trace-utils/Cargo.toml
@@ -36,4 +36,4 @@ serde_json = "1.0"
 httpmock = { version = "0.7.0"}
 
 [features]
-test-utils = ["httpmock", "testcontainers", "cargo_metadata"]
+test-utils = ["httpmock", "testcontainers", "cargo_metadata"]
diff --git a/trace-utils/src/test_utils/datadog_test_agent.rs b/trace-utils/src/test_utils/datadog_test_agent.rs
@@ -171,7 +171,11 @@ impl DatadogTestAgent {
 
     async fn get_base_uri_string(&self) -> String {
         let container_host = self.container.get_host().await.unwrap().to_string();
-        let container_port = self.container.get_host_port_ipv4(8126).await.unwrap();
+        let container_port = self
+            .container
+            .get_host_port_ipv4(TEST_AGENT_PORT)
+            .await
+            .unwrap();
 
         format!("http://{}:{}", container_host, container_port)
     }

diff --git a/trace-utils/tests/snapshots/compare_v04_trace_snapshot_test.json b/trace-utils/tests/snapshots/compare_v04_trace_snapshot_test.json
@@ -60,4 +60,4 @@
       "type": "web"
     }
   ]
-]
+]