all tools

cursor-server/tests/support/fake_cursor.rs

@@ -0,0 +1,9 @@
+use bytes::Bytes;
+use cursor_server::{cursor::connect, Result};
+use prost::Message;
+
+pub fn decode_single<M: Message + Default>(frame: &Bytes) -> Result<M> {
+    let frames = connect::decode_frames(frame)?;
+    assert_eq!(frames.len(), 1);
+    Ok(M::decode(frames[0].1.clone())?)
+}

cursor-server/tests/support/fake_provider.rs

@@ -0,0 +1,50 @@
+#![allow(dead_code)]
+
+use std::{
+    collections::VecDeque,
+    sync::{Arc, Mutex},
+};
+
+use cursor_server::{
+    prompting::ModelRequest,
+    provider::{Provider, ProviderStream, ResponseEvent},
+    Error,
+};
+use futures_util::stream;
+use tokio_util::sync::CancellationToken;
+
+type FakeResponse = Vec<Result<ResponseEvent, Error>>;
+
+#[derive(Clone, Default)]
+pub struct FakeProvider {
+    responses: Arc<Mutex<VecDeque<FakeResponse>>>,
+    requests: Arc<Mutex<Vec<ModelRequest>>>,
+}
+
+impl FakeProvider {
+    pub fn push(&self, events: Vec<ResponseEvent>) {
+        self.responses
+            .lock()
+            .unwrap()
+            .push_back(events.into_iter().map(Ok).collect());
+    }
+    pub fn push_error(&self, error: Error) {
+        self.responses.lock().unwrap().push_back(vec![Err(error)]);
+    }
+    pub fn requests(&self) -> Vec<ModelRequest> {
+        self.requests.lock().unwrap().clone()
+    }
+}
+
+impl Provider for FakeProvider {
+    fn stream(&self, request: ModelRequest, _cancellation: CancellationToken) -> ProviderStream {
+        self.requests.lock().unwrap().push(request);
+        let events = self
+            .responses
+            .lock()
+            .unwrap()
+            .pop_front()
+            .expect("fake response configured");
+        Box::pin(stream::iter(events))
+    }
+}

cursor-server/tests/support/fixtures.rs

@@ -0,0 +1,17 @@
+#![allow(dead_code)]
+
+use cursor_server::{
+    model::{CanonicalMessage, Origin, Role},
+    store::Store,
+};
+
+pub async fn temp_store() -> (tempfile::TempDir, Store) {
+    let directory = tempfile::tempdir().unwrap();
+    let url = format!("sqlite://{}", directory.path().join("test.db").display());
+    let store = Store::connect(&url).await.unwrap();
+    (directory, store)
+}
+
+pub fn user(id: &str, text: &str) -> CanonicalMessage {
+    CanonicalMessage::text(id, Role::User, Origin::User, text)
+}