Another fix for CI-only configs

private-attribution · Dec 18, 2024 · 7d91b4f · 7d91b4f
1 parent 3d5fbe3
commit 7d91b4f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ipa-core/src/net/client/mod.rs b/ipa-core/src/net/client/mod.rs
@@ -726,7 +726,7 @@ pub(crate) mod tests {
         };
         test_query_command(
             |client| async move {
-                let data = QueryInput {
+                let data = QueryInput::Inline {
                     query_id: expected_query_id,
                     input_stream: expected_input.to_vec().into(),
                 };

diff --git a/ipa-core/src/net/transport.rs b/ipa-core/src/net/transport.rs
@@ -574,7 +574,7 @@ mod tests {
 
         let mut handle_resps = Vec::with_capacity(helper_shares.len());
         for (i, input_stream) in helper_shares.into_iter().enumerate() {
-            let data = QueryInput {
+            let data = QueryInput::Inline {
                 query_id,
                 input_stream,
             };
@@ -586,7 +586,7 @@ mod tests {
         // convention - first client is shard leader, and we submitted the inputs to it.
         try_join_all(clients.iter().skip(1).map(|ring| {
             try_join_all(ring.each_ref().map(|shard_client| {
-                shard_client.query_input(QueryInput {
+                shard_client.query_input(QueryInput::Inline {
                     query_id,
                     input_stream: BodyStream::empty(),
                 })
@@ -638,7 +638,7 @@ mod tests {
                 |(helper, shard_streams)| async move {
                     try_join_all(shard_streams.into_iter().enumerate().map(
                         |(shard, input_stream)| {
-                            clients[shard][helper].query_input(QueryInput {
+                            clients[shard][helper].query_input(QueryInput::Inline {
                                 query_id,
                                 input_stream,
                             })