vitest

Author: ponderingdemocritus

packages/test/package.json

@@ -5,25 +5,29 @@
     "type": "module",
     "types": "dist/index.d.ts",
     "dependencies": {
-        "@ai16z/eliza": "workspace:*",
-        "@ai16z/plugin-solana": "workspace:*",
-        "@ai16z/plugin-node": "workspace:*",
-        "@ai16z/plugin-bootstrap": "workspace:*",
+        "@ai16z/adapter-sqlite": "workspace:*",
         "@ai16z/adapter-sqljs": "workspace:*",
         "@ai16z/adapter-supabase": "workspace:*",
-        "@ai16z/adapter-sqlite": "workspace:*",
-        "tsup": "^8.3.5",
+        "@ai16z/eliza": "workspace:*",
+        "@ai16z/plugin-bootstrap": "workspace:*",
+        "@ai16z/plugin-node": "workspace:*",
+        "@ai16z/plugin-solana": "workspace:*",
+        "@coral-xyz/anchor": "^0.30.1",
         "@solana/spl-token": "0.4.9",
         "@solana/web3.js": "1.95.4",
-        "@coral-xyz/anchor": "^0.30.1",
+        "@types/jest": "^29.5.14",
         "bignumber": "1.1.0",
         "bignumber.js": "9.1.2",
+        "jest": "^29.7.0",
+        "node-cache": "5.1.2",
         "pumpdotfun-sdk": "1.3.2",
-        "node-cache": "5.1.2"
+        "ts-jest": "^29.2.5",
+        "tsup": "^8.3.5",
+        "vitest": "^2.1.4"
     },
     "scripts": {
-        "test": "jest --runInBand",
-        "test:watch": "jest --runInBand --watch"
+        "test": "vitest",
+        "test:watch": "vitest --watch"
     },
     "peerDependencies": {
         "whatwg-url": "7.1.0"

packages/test/src/tests/actions.test.ts

@@ -1,18 +1,19 @@
 import dotenv from "dotenv";
+import { describe, test, expect, beforeAll, afterAll } from "vitest";
 import { createRuntime } from "../test_resources/createRuntime.js";
 import { getOrCreateRelationship } from "../test_resources/getOrCreateRelationship.js";
 import { runAiTest } from "../test_resources/runAiTest.js";
 import { messageHandlerTemplate } from "../test_resources/templates.js";
 import { TEST_ACTION, TEST_ACTION_FAIL } from "../test_resources/testAction.js";
-import { type User } from "../test_resources/types.js";
+import { User } from "../test_resources/types.js";
 import {
     MemoryManager,
     Content,
     IAgentRuntime,
     ModelClass,
     State,
-    type Memory,
-    type UUID,
+    Memory,
+    UUID,
     stringToUuid,
     composeContext,
     generateMessageResponse,
@@ -220,46 +221,57 @@ describe("Actions", () => {
     });
 
     // Validate that TEST_ACTION is in the state
-    test("Validate that TEST_ACTION is in the state", async () => {
-        await runAiTest("Validate TEST_ACTION is in the state", async () => {
-            const message: Memory = {
-                agentId: runtime.agentId,
-                userId: user.id as UUID,
-                content: {
-                    text: "Please respond with the message 'ok' and the action TEST_ACTION",
-                },
-                roomId,
-            };
-
-            const response = await handleMessage(runtime, message);
-            return response.action === "TEST_ACTION"; // Return true if the expected action matches
-        });
-    }, 60000);
+    test(
+        "Validate that TEST_ACTION is in the state",
+        async () => {
+            await runAiTest(
+                "Validate TEST_ACTION is in the state",
+                async () => {
+                    const message: Memory = {
+                        agentId: runtime.agentId,
+                        userId: user.id as UUID,
+                        content: {
+                            text: "Please respond with the message 'ok' and the action TEST_ACTION",
+                        },
+                        roomId,
+                    };
+
+                    const response = await handleMessage(runtime, message);
+                    return response.action === "TEST_ACTION"; // Return true if the expected action matches
+                }
+            );
+        },
+        { timeout: 60000 }
+    );
 
     // Test that TEST_ACTION action handler is called properly
-    test("Test action handler is called", async () => {
-        await runAiTest("Test action handler is called", async () => {
-            const testAction = runtime.actions.find(
-                (action) => action.name === "TEST_ACTION"
-            );
-            if (!testAction || !testAction.handler) {
-                console.error(
-                    "Continue action or its handler function is undefined"
+    test(
+        "Test action handler is called",
+        async () => {
+            await runAiTest("Test action handler is called", async () => {
+                const testAction = runtime.actions.find(
+                    (action) => action.name === "TEST_ACTION"
                 );
-                return false; // Return false to indicate the test setup failed
-            }
-
-            const mockMessage: Memory = {
-                userId: user.id as UUID,
-                agentId: runtime.agentId,
-                content: {
-                    text: "Test message for TEST action",
-                },
-                roomId,
-            };
-
-            const response = await testAction.handler(runtime, mockMessage);
-            return response !== undefined; // Return true if the handler returns a defined response
-        });
-    }, 60000); // You can adjust the timeout if needed
+                if (!testAction || !testAction.handler) {
+                    console.error(
+                        "Continue action or its handler function is undefined"
+                    );
+                    return false; // Return false to indicate the test setup failed
+                }
+
+                const mockMessage: Memory = {
+                    userId: user.id as UUID,
+                    agentId: runtime.agentId,
+                    content: {
+                        text: "Test message for TEST action",
+                    },
+                    roomId,
+                };
+
+                const response = await testAction.handler(runtime, mockMessage);
+                return response !== undefined; // Return true if the handler returns a defined response
+            });
+        },
+        { timeout: 60000 }
+    ); // You can adjust the timeout if needed
 });