From 7d91b4fc5d0aaa37d55a92a48e9e983b4f43947b Mon Sep 17 00:00:00 2001
From: Andy Leiserson <aleiserson@mozilla.com>
Date: Wed, 18 Dec 2024 12:46:54 -0800
Subject: [PATCH] Another fix for CI-only configs

---
 ipa-core/src/net/client/mod.rs | 2 +-
 ipa-core/src/net/transport.rs  | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/ipa-core/src/net/client/mod.rs b/ipa-core/src/net/client/mod.rs
index d4789b198..42d7c1377 100644
--- 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
index d8fbfc4b5..7c75d683e 100644
--- 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,
                             })