1 files changed, 45 insertions, 3 deletions
diff --git a/packages/session-orchestrator/src/queue.test.ts b/packages/session-orchestrator/src/queue.test.ts
index 225d1af..adf5d9a 100644
--- a/packages/session-orchestrator/src/queue.test.ts
+++ b/packages/session-orchestrator/src/queue.test.ts
@@ -72,6 +72,14 @@ function createInMemoryStore(): ConversationStore & {
 		async setCwd(conversationId, cwd) {
 			cwdData.set(conversationId, cwd);
 		},
+		async clearCwd(conversationId) {
+			cwdData.delete(conversationId);
+		},
+		async getComputerId() {
+			return null;
+		},
+		async setComputerId() {},
+		async clearComputerId() {},
 		async getReasoningEffort(conversationId) {
 			return effortData.get(conversationId) ?? null;
 		},
@@ -110,13 +118,44 @@ function createInMemoryStore(): ConversationStore & {
 			return null;
 		},
 		async ensureWorkspace(id) {
-			return { id, title: id, defaultCwd: null, createdAt: 0, lastActivityAt: 0 };
+			return {
+				id,
+				title: id,
+				defaultCwd: null,
+				defaultComputerId: null,
+				createdAt: 0,
+				lastActivityAt: 0,
+			};
 		},
 		async setWorkspaceTitle(id, title) {
-			return { id, title, defaultCwd: null, createdAt: 0, lastActivityAt: 0 };
+			return {
+				id,
+				title,
+				defaultCwd: null,
+				defaultComputerId: null,
+				createdAt: 0,
+				lastActivityAt: 0,
+			};
 		},
 		async setWorkspaceDefaultCwd(id, defaultCwd) {
-			return { id, title: id, defaultCwd, createdAt: 0, lastActivityAt: 0 };
+			return {
+				id,
+				title: id,
+				defaultCwd,
+				defaultComputerId: null,
+				createdAt: 0,
+				lastActivityAt: 0,
+			};
+		},
+		async setWorkspaceDefaultComputerId(id, defaultComputerId) {
+			return {
+				id,
+				title: id,
+				defaultCwd: null,
+				defaultComputerId,
+				createdAt: 0,
+				lastActivityAt: 0,
+			};
 		},
 		async deleteWorkspace() {
 			return { closedCount: 0 };
@@ -131,6 +170,9 @@ function createInMemoryStore(): ConversationStore & {
 		async getEffectiveCwd(conversationId) {
 			return cwdData.get(conversationId) ?? null;
 		},
+		async getEffectiveComputer() {
+			return null;
+		},
 	};
 }