chore: add tests for chat-room and counter (#797)

NathanFlurry · NathanFlurry · commit ed5f01a77d52 · 2025-03-26T20:50:43.000Z
diff --git a/docs/platforms/cloudflare-workers.mdx b/docs/platforms/cloudflare-workers.mdx
@@ -48,7 +48,7 @@ The Cloudflare Durable Objects platform leverages Cloudflare Workers to run Acto
 		In a separate terminal, run the auto-generated test client:
 
 		```sh
-		npx tsx tests/client.ts
+		npx tsx scripts/connect.ts
 		# Outputs:
 		# Event: 5
 		# Action: 5
diff --git a/docs/platforms/nodejs.mdx b/docs/platforms/nodejs.mdx
@@ -39,7 +39,7 @@ import CreateActorCli from '/snippets/create-actor-cli.mdx';
 		In a separate terminal, run the auto-generated test client:
 
 		```sh
-		npx tsx tests/client.ts
+		npx tsx scripts/connect.ts
 		# Outputs:
 		# Event: 5
 		# RPC: 5
diff --git a/docs/platforms/rivet.mdx b/docs/platforms/rivet.mdx
@@ -71,7 +71,7 @@ import CreateActorCli from '/snippets/create-actor-cli.mdx';
 		Then run the client:
 
 		```sh
-		npx tsx tests/client.ts
+		npx tsx scripts/connect.ts
 		# Outputs:
 		# Event: 5
 		# Action: 5
diff --git a/examples/chat-room/package.json b/examples/chat-room/package.json
@@ -4,15 +4,17 @@
   "private": true,
   "type": "module",
   "scripts": {
-    "check-types": "tsc --noEmit"
+    "check-types": "tsc --noEmit",
+    "test": "vitest run"
   },
   "devDependencies": {
     "@types/node": "^22.13.9",
     "@types/prompts": "^2",
     "actor-core": "workspace:*",
     "prompts": "^2.4.2",
     "tsx": "^3.12.7",
-    "typescript": "^5.5.2"
+    "typescript": "^5.5.2",
+    "vitest": "^3.0.9"
   },
   "example": {
     "platforms": [
diff --git a/examples/chat-room/scripts/cli.ts b/examples/chat-room/scripts/cli.ts
diff --git a/examples/chat-room/scripts/connect.ts b/examples/chat-room/scripts/connect.ts
diff --git a/examples/chat-room/tests/chat-room.test.ts b/examples/chat-room/tests/chat-room.test.ts
@@ -0,0 +1,56 @@
+import { test, expect } from "vitest";
+import { setupTest } from "actor-core/test";
+import { app } from "../src/index";
+
+test("chat room should handle messages", async () => {
+	const { client } = await setupTest(app);
+	
+	// Connect to chat room
+	const chatRoom = await client.chatRoom.get();
+	
+	// Initial history should be empty
+	const initialMessages = await chatRoom.getHistory();
+	expect(initialMessages).toEqual([]);
+	
+	// Test event emission
+	let receivedUsername = "";
+	let receivedMessage = "";
+	chatRoom.on("newMessage", (username: string, message: string) => {
+		receivedUsername = username;
+		receivedMessage = message;
+	});
+	
+	// Send a message
+	const testUser = "william";
+	const testMessage = "All the world's a stage.";
+	await chatRoom.sendMessage(testUser, testMessage);
+	
+	// Verify event was emitted with correct data
+	expect(receivedUsername).toBe(testUser);
+	expect(receivedMessage).toBe(testMessage);
+	
+	// Verify message was stored in history
+	const updatedMessages = await chatRoom.getHistory();
+	expect(updatedMessages).toEqual([
+		{ username: testUser, message: testMessage }
+	]);
+	
+	// Send multiple messages and verify
+	const users = ["romeo", "juliet", "othello"];
+	const messages = ["Wherefore art thou?", "Here I am!", "The green-eyed monster."];
+	
+	for (let i = 0; i < users.length; i++) {
+		await chatRoom.sendMessage(users[i], messages[i]);
+		
+		// Verify event emission
+		expect(receivedUsername).toBe(users[i]);
+		expect(receivedMessage).toBe(messages[i]);
+	}
+	
+	// Verify all messages are in history in correct order
+	const finalHistory = await chatRoom.getHistory();
+	expect(finalHistory).toEqual([
+		{ username: testUser, message: testMessage },
+		...users.map((username, i) => ({ username, message: messages[i] }))
+	]);
+});
diff --git a/examples/counter/package.json b/examples/counter/package.json
@@ -4,13 +4,15 @@
   "private": true,
   "type": "module",
   "scripts": {
-    "check-types": "tsc --noEmit"
+    "check-types": "tsc --noEmit",
+    "test": "vitest run"
   },
   "devDependencies": {
     "@types/node": "^22.13.9",
     "actor-core": "workspace:*",
     "tsx": "^3.12.7",
-    "typescript": "^5.7.3"
+    "typescript": "^5.7.3",
+    "vitest": "^3.0.9"
   },
   "example": {
     "platforms": [
diff --git a/examples/counter/scripts/connect.ts b/examples/counter/scripts/connect.ts
diff --git a/examples/counter/src/counter.ts b/examples/counter/src/counter.ts
@@ -8,6 +8,9 @@ const counter = actor({
 			c.broadcast("newCount", c.state.count);
 			return c.state.count;
 		},
+		getCount: (c) => {
+			return c.state.count;
+		},
 	},
 });
 
diff --git a/examples/counter/tests/counter.test.ts b/examples/counter/tests/counter.test.ts
@@ -0,0 +1,36 @@
+import { test, expect } from "vitest";
+import { setupTest } from "actor-core/test";
+import { app } from "../src/index";
+
+test("it should count", async () => {
+	const { client } = await setupTest(app);
+	const counter = await client.counter.get();
+	
+	// Test initial count
+	expect(await counter.getCount()).toBe(0);
+	
+	// Test event emission
+	let eventCount = -1;
+	counter.on("newCount", (count: number) => {
+		eventCount = count;
+	});
+	
+	// Test increment
+	const incrementAmount = 5;
+	const result = await counter.increment(incrementAmount);
+	expect(result).toBe(incrementAmount);
+	
+	// Verify event was emitted with correct count
+	expect(eventCount).toBe(incrementAmount);
+	
+	// Test multiple increments
+	for (let i = 1; i <= 3; i++) {
+		const newCount = await counter.increment(incrementAmount);
+		expect(newCount).toBe(incrementAmount * (i + 1));
+		expect(eventCount).toBe(incrementAmount * (i + 1));
+	}
+	
+	// Verify final count
+	expect(await counter.getCount()).toBe(incrementAmount * 4);
+});
+
diff --git a/yarn.lock b/yarn.lock
@@ -4211,6 +4211,7 @@ __metadata:
     prompts: "npm:^2.4.2"
     tsx: "npm:^3.12.7"
     typescript: "npm:^5.5.2"
+    vitest: "npm:^3.0.9"
   languageName: unknown
   linkType: soft
 
@@ -4455,6 +4456,7 @@ __metadata:
     actor-core: "workspace:*"
     tsx: "npm:^3.12.7"
     typescript: "npm:^5.7.3"
+    vitest: "npm:^3.0.9"
   languageName: unknown
   linkType: soft
 
