WIP

Nathan Sobo created 5 months ago

Change summary

crates/acp/src/acp.rs | 43 ++++++++++++++++++++++++++++++++-----------
1 file changed, 32 insertions(+), 11 deletions(-)

Detailed changes

crates/acp/src/acp.rs

@@ -808,17 +808,29 @@ mod tests {
             fake_server.on_user_message(move |params, server, mut cx| async move {
                 server
                     .update(&mut cx, |server, cx| {
-                        let future =
-                            server
-                                .connection
-                                .request(acp::StreamAssistantMessageChunkParams {
-                                    thread_id: params.thread_id,
-                                    chunk: acp::AssistantMessageChunk::Thought {
-                                        chunk: "Thinking ".into(),
-                                    },
-                                });
-
-                        cx.spawn(async move |_, _| future.await)
+                        server.send_to_zed(
+                            acp::StreamAssistantMessageChunkParams {
+                                thread_id: params.thread_id.clone(),
+                                chunk: acp::AssistantMessageChunk::Thought {
+                                    chunk: "Thinking ".into(),
+                                },
+                            },
+                            cx,
+                        )
+                    })?
+                    .await
+                    .unwrap();
+                server
+                    .update(&mut cx, |server, cx| {
+                        server.send_to_zed(
+                            acp::StreamAssistantMessageChunkParams {
+                                thread_id: params.thread_id,
+                                chunk: acp::AssistantMessageChunk::Thought {
+                                    chunk: "hard!".into(),
+                                },
+                            },
+                            cx,
+                        )
                     })?
                     .await
                     .unwrap();
@@ -1191,5 +1203,14 @@ mod tests {
                     handler(request, server, cx).boxed_local()
                 }));
         }
+
+        fn send_to_zed<T: acp::ClientRequest>(
+            &self,
+            message: T,
+            cx: &Context<Self>,
+        ) -> Task<Result<T::Response, acp::Error>> {
+            let future = self.connection.request(message);
+            cx.foreground_executor().spawn(future)
+        }
     }
 }