@actiondock/testing 2.0.8 → 2.0.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@actiondock/testing",
-  "version": "2.0.8",
+  "version": "2.0.10",
   "description": "ActionDock Test Runtime for testing Actions with real Core execution semantics",
   "type": "module",
   "main": "./src/index.ts",
@@ -27,14 +27,20 @@
     "test": "bun test"
   },
   "dependencies": {
-    "@actiondock/core": "^2.0.0",
-    "@actiondock/sdk": "^2.0.0"
+    "@actiondock/core": "^2.0.10",
+    "@actiondock/sdk": "^2.0.10"
   },
   "devDependencies": {
     "@types/bun": "latest",
     "typescript": "^5.7.0"
   },
-  "keywords": ["actiondock", "testing", "agent", "actions", "mock"],
+  "keywords": [
+    "actiondock",
+    "testing",
+    "agent",
+    "actions",
+    "mock"
+  ],
   "author": "team4u",
   "license": "Apache-2.0",
   "repository": {

package/src/runtime.ts

@@ -7,7 +7,6 @@ import {
   RuntimeStateStore,
   setDefaultEventSink,
   setProcessExecutor,
-  setSystemClock,
 } from "@actiondock/core";
 import type {
   ActionDefinition,
@@ -226,8 +225,7 @@ export function createTestRuntime(options: TestRuntimeOptions = {}): TestRuntime
       clock,
     });
 
-  // 全局注入测试时钟与进程执行器
-  setSystemClock(clock);
+  // 全局注入测试进程执行器
   setProcessExecutor(process);
 
   // 初始化配置数据

package/src/storage.ts

@@ -1,8 +1,7 @@
 import {
   type Clock,
-  createDefaultSqliteDriver,
-  getSystemClock,
   type RuntimeStorage,
+  SqliteRuntimeStorage,
   type SqliteDriver,
   type StateEntry,
   type TerminalRunStatus,
@@ -23,579 +22,16 @@ export interface MemoryStorageOptions {
 
 /**
  * 统一内存运行时存储实现。
- * 基于 SQLite 内存模式构建，确保与生产环境核心存储具备完全相同的配置优先级、状态过期契约与运行终态行为。
+ * 基于 SqliteRuntimeStorage 构建，默认使用 :memory: 内存数据库并对接虚拟时钟，
+ * 确保与生产环境具备完全相同的配置优先级、状态过期契约与运行终态行为。
  */
-export class MemoryStorage implements RuntimeStorage {
-  private driver: SqliteDriver;
-  private packageId: string;
-  private clock: Clock;
-  private isClosed = false;
-
+export class MemoryStorage extends SqliteRuntimeStorage {
   constructor(options: MemoryStorageOptions = {}) {
-    this.packageId = options.packageId || "test-pkg";
-    this.clock = options.clock || getSystemClock();
-    this.driver = options.driver || createDefaultSqliteDriver(":memory:");
-    this.init();
-  }
-
-  /**
-   * 初始化内存表结构与索引。
-   */
-  private init(): void {
-    this.driver.exec("PRAGMA journal_mode = MEMORY;");
-    this.driver.exec("PRAGMA synchronous = OFF;");
-
-    this.driver.transaction(() => {
-      this.driver.exec(`
-        CREATE TABLE IF NOT EXISTS config (
-          package_id TEXT NOT NULL,
-          key TEXT NOT NULL,
-          value_json TEXT,
-          updated_at TEXT NOT NULL,
-          PRIMARY KEY (package_id, key)
-        );
-
-        CREATE TABLE IF NOT EXISTS state (
-          package_id TEXT NOT NULL,
-          namespace TEXT NOT NULL,
-          key TEXT NOT NULL,
-          value_json TEXT,
-          updated_at TEXT NOT NULL,
-          expires_at TEXT,
-          PRIMARY KEY (package_id, namespace, key)
-        );
-
-        CREATE TABLE IF NOT EXISTS runs (
-          id TEXT PRIMARY KEY,
-          root_run_id TEXT NOT NULL,
-          parent_run_id TEXT,
-          package_id TEXT NOT NULL,
-          package_instance_id TEXT NOT NULL,
-          action_id TEXT NOT NULL,
-          generation_id TEXT NOT NULL,
-          owner_id TEXT NOT NULL,
-          status TEXT NOT NULL,
-          input_json TEXT,
-          output_json TEXT,
-          error_json TEXT,
-          started_at TEXT NOT NULL,
-          finished_at TEXT,
-          duration_ms INTEGER
-        );
-
-        CREATE INDEX IF NOT EXISTS idx_runs_action ON runs(package_id, action_id);
-        CREATE INDEX IF NOT EXISTS idx_runs_root ON runs(root_run_id);
-        CREATE INDEX IF NOT EXISTS idx_runs_started ON runs(started_at DESC);
-        CREATE INDEX IF NOT EXISTS idx_state_expires ON state(expires_at);
-      `);
+    super({
+      packageId: options.packageId || "test-pkg",
+      dbPath: ":memory:",
+      driver: options.driver,
+      clock: options.clock,
     });
   }
-
-  // --- 配置管理 ---
-
-  getConfig<T = unknown>(key: string): T | undefined {
-    const stmt = this.driver.prepare(
-      "SELECT value_json FROM config WHERE package_id = ? AND key = ?"
-    );
-    const row = stmt.get<{ value_json: string }>(this.packageId, key);
-    if (!row || row.value_json === undefined || row.value_json === null) {
-      return undefined;
-    }
-    try {
-      return JSON.parse(row.value_json) as T;
-    } catch {
-      return row.value_json as unknown as T;
-    }
-  }
-
-  listConfig(): Record<string, unknown> {
-    const stmt = this.driver.prepare(
-      "SELECT key, value_json FROM config WHERE package_id = ?"
-    );
-    const rows = stmt.all<{ key: string; value_json: string }>(this.packageId);
-    const result: Record<string, unknown> = {};
-    for (const row of rows) {
-      try {
-        result[row.key] = JSON.parse(row.value_json);
-      } catch {
-        result[row.key] = row.value_json;
-      }
-    }
-    return result;
-  }
-
-  setConfig(key: string, value: unknown): void {
-    const stmt = this.driver.prepare(`
-      INSERT INTO config (package_id, key, value_json, updated_at)
-      VALUES (?, ?, ?, ?)
-      ON CONFLICT(package_id, key) DO UPDATE SET
-        value_json = excluded.value_json,
-        updated_at = excluded.updated_at
-    `);
-    const valJson = JSON.stringify(value);
-    const now = this.clock.now().toISOString();
-    stmt.run(this.packageId, key, valJson, now);
-  }
-
-  deleteConfig(key: string): boolean {
-    const stmt = this.driver.prepare(
-      "DELETE FROM config WHERE package_id = ? AND key = ?"
-    );
-    const res = stmt.run(this.packageId, key);
-    return res.changes > 0;
-  }
-
-  // --- 状态管理 ---
-
-  async getState<T = unknown>(
-    namespace: string,
-    key: string
-  ): Promise<T | undefined> {
-    const stmt = this.driver.prepare(
-      "SELECT value_json, expires_at FROM state WHERE package_id = ? AND namespace = ? AND key = ?"
-    );
-    const row = stmt.get<{ value_json: string; expires_at?: string }>(
-      this.packageId,
-      namespace,
-      key
-    );
-    if (!row || row.value_json === undefined || row.value_json === null) {
-      return undefined;
-    }
-
-    if (row.expires_at) {
-      const expires = new Date(row.expires_at).getTime();
-      const current = this.clock.now().getTime();
-      if (current >= expires) {
-        this.deleteState(namespace, key).catch(() => {});
-        return undefined;
-      }
-    }
-
-    try {
-      return JSON.parse(row.value_json) as T;
-    } catch {
-      return row.value_json as unknown as T;
-    }
-  }
-
-  async findState<T = unknown>(
-    targetKey: string,
-    namespace?: string
-  ): Promise<StateEntry | undefined> {
-    let ns = namespace;
-    let actualKey = targetKey;
-
-    if (!ns && targetKey.includes(":")) {
-      const parts = targetKey.split(":");
-      ns = parts[0];
-      actualKey = parts.slice(1).join(":");
-    }
-
-    if (ns) {
-      const val = await this.getState<T>(ns, actualKey);
-      if (val === undefined) return undefined;
-
-      const stmt = this.driver.prepare(
-        "SELECT updated_at, expires_at FROM state WHERE package_id = ? AND namespace = ? AND key = ?"
-      );
-      const row = stmt.get<{ updated_at: string; expires_at?: string }>(
-        this.packageId,
-        ns,
-        actualKey
-      );
-
-      return {
-        packageId: this.packageId,
-        namespace: ns,
-        key: actualKey,
-        fullKey: targetKey,
-        value: val,
-        updatedAt: row?.updated_at || this.clock.now().toISOString(),
-        expiresAt: row?.expires_at,
-      };
-    }
-
-    const val = await this.getState<T>("", actualKey);
-    if (val !== undefined) {
-      const stmt = this.driver.prepare(
-        "SELECT updated_at, expires_at FROM state WHERE package_id = ? AND namespace = ? AND key = ?"
-      );
-      const row = stmt.get<{ updated_at: string; expires_at?: string }>(
-        this.packageId,
-        "",
-        actualKey
-      );
-      return {
-        packageId: this.packageId,
-        namespace: "",
-        key: actualKey,
-        fullKey: actualKey,
-        value: val,
-        updatedAt: row?.updated_at || this.clock.now().toISOString(),
-        expiresAt: row?.expires_at,
-      };
-    }
-
-    const stmt = this.driver.prepare(
-      "SELECT namespace, key, value_json, updated_at, expires_at FROM state WHERE package_id = ? AND key = ?"
-    );
-    const rows = stmt.all<{
-      namespace: string;
-      key: string;
-      value_json: string;
-      updated_at: string;
-      expires_at?: string;
-    }>(this.packageId, actualKey);
-
-    const now = this.clock.now().getTime();
-    for (const row of rows) {
-      if (row.expires_at && now >= new Date(row.expires_at).getTime()) {
-        continue;
-      }
-      let parsedVal: unknown;
-      try {
-        parsedVal = JSON.parse(row.value_json);
-      } catch {
-        parsedVal = row.value_json;
-      }
-      return {
-        packageId: this.packageId,
-        namespace: row.namespace,
-        key: row.key,
-        fullKey: row.namespace ? `${row.namespace}:${row.key}` : row.key,
-        value: parsedVal,
-        updatedAt: row.updated_at,
-        expiresAt: row.expires_at,
-      };
-    }
-
-    return undefined;
-  }
-
-  async setState<T = unknown>(
-    namespace: string,
-    key: string,
-    value: T,
-    ttl?: number
-  ): Promise<void> {
-    const stmt = this.driver.prepare(`
-      INSERT INTO state (package_id, namespace, key, value_json, updated_at, expires_at)
-      VALUES (?, ?, ?, ?, ?, ?)
-      ON CONFLICT(package_id, namespace, key) DO UPDATE SET
-        value_json = excluded.value_json,
-        updated_at = excluded.updated_at,
-        expires_at = excluded.expires_at
-    `);
-    const valJson = JSON.stringify(value);
-    const now = this.clock.now();
-    const updatedAt = now.toISOString();
-
-    let expiresAt: string | null = null;
-    if (typeof ttl === "number" && ttl > 0) {
-      expiresAt = new Date(now.getTime() + ttl * 1000).toISOString();
-    }
-
-    stmt.run(this.packageId, namespace, key, valJson, updatedAt, expiresAt);
-  }
-
-  async deleteState(namespace: string, key: string): Promise<boolean> {
-    const stmt = this.driver.prepare(
-      "DELETE FROM state WHERE package_id = ? AND namespace = ? AND key = ?"
-    );
-    const res = stmt.run(this.packageId, namespace, key);
-    return res.changes > 0;
-  }
-
-  async deleteStateSmart(
-    targetKey: string,
-    namespace?: string
-  ): Promise<boolean> {
-    let ns = namespace;
-    let actualKey = targetKey;
-
-    if (!ns && targetKey.includes(":")) {
-      const parts = targetKey.split(":");
-      ns = parts[0];
-      actualKey = parts.slice(1).join(":");
-    }
-
-    if (ns !== undefined) {
-      return this.deleteState(ns, actualKey);
-    }
-
-    const deletedRoot = await this.deleteState("", actualKey);
-    if (deletedRoot) return true;
-
-    const stmt = this.driver.prepare(
-      "DELETE FROM state WHERE package_id = ? AND key = ?"
-    );
-    const res = stmt.run(this.packageId, actualKey);
-    return res.changes > 0;
-  }
-
-  async clearState(
-    options: { namespace?: string; all?: boolean; prefix?: string } = {}
-  ): Promise<number> {
-    let sql = "DELETE FROM state WHERE package_id = ?";
-    const params: unknown[] = [this.packageId];
-
-    if (options.namespace !== undefined) {
-      sql += " AND namespace = ?";
-      params.push(options.namespace);
-    }
-
-    if (options.prefix) {
-      sql += " AND key LIKE ? ESCAPE '\\'";
-      const escapedPrefix = options.prefix.replace(/([%_\\])/g, "\\$1");
-      params.push(`${escapedPrefix}%`);
-    }
-
-    const stmt = this.driver.prepare(sql);
-    const res = stmt.run(...params);
-    return res.changes;
-  }
-
-  async listStateKeys(
-    namespace?: string | null,
-    prefix?: string
-  ): Promise<string[]> {
-    let sql = "SELECT namespace, key, expires_at FROM state WHERE package_id = ?";
-    const params: unknown[] = [this.packageId];
-
-    if (namespace !== null && namespace !== undefined) {
-      sql += " AND namespace = ?";
-      params.push(namespace);
-    }
-
-    if (prefix) {
-      sql += " AND key LIKE ? ESCAPE '\\'";
-      const escapedPrefix = prefix.replace(/([%_\\])/g, "\\$1");
-      params.push(`${escapedPrefix}%`);
-    }
-
-    const stmt = this.driver.prepare(sql);
-    const rows = stmt.all<{
-      namespace: string;
-      key: string;
-      expires_at?: string;
-    }>(...params);
-
-    const now = this.clock.now().getTime();
-    const result: string[] = [];
-
-    for (const row of rows) {
-      if (row.expires_at && now >= new Date(row.expires_at).getTime()) {
-        continue;
-      }
-      if (namespace !== null && namespace !== undefined) {
-        result.push(row.key);
-      } else {
-        result.push(row.namespace ? `${row.namespace}:${row.key}` : row.key);
-      }
-    }
-
-    return result;
-  }
-
-  async listStateEntries(
-    options: { namespace?: string; prefix?: string } = {}
-  ): Promise<StateEntry[]> {
-    let sql = "SELECT namespace, key, value_json, updated_at, expires_at FROM state WHERE package_id = ?";
-    const params: unknown[] = [this.packageId];
-
-    if (options.namespace !== undefined) {
-      sql += " AND namespace = ?";
-      params.push(options.namespace);
-    }
-
-    if (options.prefix) {
-      sql += " AND key LIKE ? ESCAPE '\\'";
-      const escapedPrefix = options.prefix.replace(/([%_\\])/g, "\\$1");
-      params.push(`${escapedPrefix}%`);
-    }
-
-    const stmt = this.driver.prepare(sql);
-    const rows = stmt.all<{
-      namespace: string;
-      key: string;
-      value_json: string;
-      updated_at: string;
-      expires_at?: string;
-    }>(...params);
-
-    const now = this.clock.now().getTime();
-    const results: StateEntry[] = [];
-
-    for (const row of rows) {
-      if (row.expires_at && now >= new Date(row.expires_at).getTime()) {
-        continue;
-      }
-
-      let parsedVal: unknown;
-      try {
-        parsedVal = JSON.parse(row.value_json);
-      } catch {
-        parsedVal = row.value_json;
-      }
-
-      results.push({
-        packageId: this.packageId,
-        namespace: row.namespace,
-        key: row.key,
-        fullKey: row.namespace ? `${row.namespace}:${row.key}` : row.key,
-        value: parsedVal,
-        updatedAt: row.updated_at,
-        expiresAt: row.expires_at,
-      });
-    }
-
-    return results;
-  }
-
-  // --- 运行记录管理 ---
-
-  createRun(record: RunRecord): void {
-    const stmt = this.driver.prepare(`
-      INSERT INTO runs (
-        id, root_run_id, parent_run_id, package_id, package_instance_id,
-        action_id, generation_id, owner_id, status, input_json, output_json,
-        error_json, started_at, finished_at, duration_ms
-      )
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-    `);
-
-    const rootRunId = record.rootRunId || record.id;
-    const parentRunId = record.parentRunId || null;
-    const packageInstanceId = record.packageInstanceId || record.packageId || this.packageId;
-    const generationId = record.generationId || "1";
-    const ownerId = record.ownerId || "local";
-    const inputJson = record.input !== undefined ? JSON.stringify(record.input) : null;
-    const outputJson = record.output !== undefined ? JSON.stringify(record.output) : null;
-    const errorJson = record.error ? JSON.stringify(record.error) : null;
-    const finishedAt = record.finishedAt || null;
-    const durationMs = typeof record.durationMs === "number" ? record.durationMs : null;
-
-    stmt.run(
-      record.id,
-      rootRunId,
-      parentRunId,
-      record.packageId || this.packageId,
-      packageInstanceId,
-      record.actionId,
-      generationId,
-      ownerId,
-      record.status,
-      inputJson,
-      outputJson,
-      errorJson,
-      record.startedAt,
-      finishedAt,
-      durationMs
-    );
-  }
-
-  updateRun(
-    id: string,
-    status: TerminalRunStatus,
-    output?: unknown,
-    error?: RuntimeError,
-    finishedAt?: string
-  ): void {
-    if (this.isClosed) return;
-    try {
-      const stmt = this.driver.prepare(`
-        UPDATE runs
-        SET status = ?, output_json = ?, error_json = ?, finished_at = ?
-        WHERE id = ?
-      `);
-      stmt.run(
-        status,
-        output !== undefined ? JSON.stringify(output) : null,
-        error ? JSON.stringify(error) : null,
-        finishedAt || this.clock.now().toISOString(),
-        id
-      );
-    } catch {
-      // 存储连接已释放时忽略
-    }
-  }
-
-  getRun(id: string): RunRecord | null {
-    const stmt = this.driver.prepare(
-      "SELECT * FROM runs WHERE id = ? AND package_id = ?"
-    );
-    const row = stmt.get<Record<string, unknown>>(id, this.packageId);
-    if (!row) return null;
-    return this.mapRunRecord(row);
-  }
-
-  listRuns(options: { actionId?: string; limit?: number } = {}): RunRecord[] {
-    const limit = options.limit || 50;
-    let rows: Record<string, unknown>[];
-    if (options.actionId) {
-      const stmt = this.driver.prepare(`
-        SELECT * FROM runs
-        WHERE package_id = ? AND action_id = ?
-        ORDER BY started_at DESC
-        LIMIT ?
-      `);
-      rows = stmt.all<Record<string, unknown>>(this.packageId, options.actionId, limit);
-    } else {
-      const stmt = this.driver.prepare(`
-        SELECT * FROM runs
-        WHERE package_id = ?
-        ORDER BY started_at DESC
-        LIMIT ?
-      `);
-      rows = stmt.all<Record<string, unknown>>(this.packageId, limit);
-    }
-    return rows.map((r) => this.mapRunRecord(r));
-  }
-
-  clearRuns(options: { actionId?: string; status?: string } = {}): number {
-    let sql = "DELETE FROM runs WHERE package_id = ?";
-    const params: unknown[] = [this.packageId];
-
-    if (options.actionId) {
-      sql += " AND action_id = ?";
-      params.push(options.actionId);
-    }
-
-    if (options.status) {
-      sql += " AND status = ?";
-      params.push(options.status);
-    }
-
-    const stmt = this.driver.prepare(sql);
-    const res = stmt.run(...params);
-    return res.changes;
-  }
-
-  close(): void {
-    if (!this.isClosed) {
-      this.isClosed = true;
-      this.driver.close();
-    }
-  }
-
-  private mapRunRecord(row: Record<string, unknown>): RunRecord {
-    return {
-      id: String(row.id),
-      rootRunId: String(row.root_run_id || row.id),
-      parentRunId: row.parent_run_id ? String(row.parent_run_id) : undefined,
-      packageId: String(row.package_id),
-      packageInstanceId: String(row.package_instance_id || row.package_id),
-      actionId: String(row.action_id),
-      generationId: String(row.generation_id || "1"),
-      ownerId: String(row.owner_id || "local"),
-      status: row.status as RunRecord["status"],
-      input: row.input_json ? JSON.parse(String(row.input_json)) : undefined,
-      output: row.output_json ? JSON.parse(String(row.output_json)) : undefined,
-      error: row.error_json ? (JSON.parse(String(row.error_json)) as RuntimeError) : undefined,
-      startedAt: String(row.started_at),
-      finishedAt: row.finished_at ? String(row.finished_at) : undefined,
-      durationMs: typeof row.duration_ms === "number" ? row.duration_ms : undefined,
-    };
-  }
 }