space-data-module-sdk 0.5.15 → 0.5.18

package/src/testing/index.d.ts

@@ -131,7 +131,55 @@ export interface PluginInvokeProcessLaunchPlan {
   args: string[];
   env?: Record<string, string | undefined>;
   cwd?: string;
-  wasmPath?: string;
+  wasmPath?: string | null;
+  hostProfile?: "runtime-host";
+}
+
+export interface RuntimeHostTestModuleDefinition {
+  moduleId: string;
+  wasmPath?: string | null;
+  metadata?: unknown;
+  [key: string]: unknown;
+}
+
+export interface RuntimeHostInstalledModule {
+  moduleId: string;
+  metadata: unknown;
+  methodIds: string[];
+}
+
+export interface RuntimeHostRowHandle {
+  schemaFileId: string;
+  rowId: number;
+}
+
+export interface RuntimeHostRowView {
+  handle: RuntimeHostRowHandle;
+  payload: unknown;
+}
+
+export interface RuntimeHostRowQueryResult {
+  columns: string[];
+  rows: unknown[][];
+  rowCount: number;
+}
+
+export interface RuntimeHostRegionDescriptor {
+  regionId: number;
+  layoutId: string;
+  recordByteLength: number;
+  alignment: number;
+  recordCount: number;
+}
+
+export interface RuntimeHostRegionRecord {
+  regionId: number;
+  recordIndex: number;
+  layoutId: string;
+  recordByteLength: number;
+  alignment: number;
+  byteLength: number;
+  bytes: Uint8Array;
 }
 
 export interface PluginInvokeProcessClient {
@@ -140,12 +188,42 @@ export interface PluginInvokeProcessClient {
   invoke(request: {
     methodId?: string | null;
     inputs?: HarnessInputFrame[];
+  }): Promise<{
+      statusCode: number;
+      errorCode?: string | null;
+      errorMessage?: string | null;
+      outputs: HarnessInputFrame[];
+  }>;
+  installModule(definition: RuntimeHostTestModuleDefinition): Promise<RuntimeHostInstalledModule>;
+  listModules(): Promise<RuntimeHostInstalledModule[]>;
+  unloadModule(moduleId: string): Promise<boolean>;
+  invokeModule(requestModuleId: string, request: {
+    methodId?: string | null;
+    inputs?: HarnessInputFrame[];
   }): Promise<{
     statusCode: number;
     errorCode?: string | null;
     errorMessage?: string | null;
     outputs: HarnessInputFrame[];
   }>;
+  appendRow(options: {
+    schemaFileId: string;
+    payload?: unknown;
+  }): Promise<RuntimeHostRowHandle>;
+  listRows(schemaFileId?: string | null): Promise<RuntimeHostRowView[]>;
+  resolveRow(handle: RuntimeHostRowHandle): Promise<RuntimeHostRowView | null>;
+  queryRows(sql: string): Promise<RuntimeHostRowQueryResult>;
+  allocateRegion(options: {
+    layoutId: string;
+    recordByteLength: number;
+    alignment?: number;
+    initialRecords?: Array<Uint8Array | ArrayBuffer | ArrayBufferView | null | undefined>;
+  }): Promise<RuntimeHostRegionDescriptor>;
+  describeRegion(regionId: number): Promise<RuntimeHostRegionDescriptor | null>;
+  resolveRecord(query: {
+    regionId: number;
+    recordIndex: number;
+  }): Promise<RuntimeHostRegionRecord | null>;
   destroy(): Promise<void>;
 }
 
@@ -160,6 +238,10 @@ export interface ModuleHarnessRuntimeDescriptor {
   wasmEdgeBinary?: string;
   wasmEdgeRunnerBinary?: string;
   enableThreads?: boolean;
+  hostProfile?: "runtime-host";
+  modules?: RuntimeHostTestModuleDefinition[];
+  defaultModuleId?: string;
+  metadata?: unknown;
 }
 
 export interface ModuleHarness {
@@ -169,12 +251,42 @@ export interface ModuleHarness {
   invoke(request: {
     methodId?: string | null;
     inputs?: HarnessInputFrame[];
+  }): Promise<{
+      statusCode: number;
+      errorCode?: string | null;
+      errorMessage?: string | null;
+      outputs: HarnessInputFrame[];
+  }>;
+  installModule(definition: RuntimeHostTestModuleDefinition): Promise<RuntimeHostInstalledModule>;
+  listModules(): Promise<RuntimeHostInstalledModule[]>;
+  unloadModule(moduleId: string): Promise<boolean>;
+  invokeModule(moduleId: string, request: {
+    methodId?: string | null;
+    inputs?: HarnessInputFrame[];
   }): Promise<{
     statusCode: number;
     errorCode?: string | null;
     errorMessage?: string | null;
     outputs: HarnessInputFrame[];
   }>;
+  appendRow(options: {
+    schemaFileId: string;
+    payload?: unknown;
+  }): Promise<RuntimeHostRowHandle>;
+  listRows(schemaFileId?: string | null): Promise<RuntimeHostRowView[]>;
+  resolveRow(handle: RuntimeHostRowHandle): Promise<RuntimeHostRowView | null>;
+  queryRows(sql: string): Promise<RuntimeHostRowQueryResult>;
+  allocateRegion(options: {
+    layoutId: string;
+    recordByteLength: number;
+    alignment?: number;
+    initialRecords?: Array<Uint8Array | ArrayBuffer | ArrayBufferView | null | undefined>;
+  }): Promise<RuntimeHostRegionDescriptor>;
+  describeRegion(regionId: number): Promise<RuntimeHostRegionDescriptor | null>;
+  resolveRecord(query: {
+    regionId: number;
+    recordIndex: number;
+  }): Promise<RuntimeHostRegionRecord | null>;
   destroy(): Promise<void>;
 }
 
@@ -234,12 +346,13 @@ export function buildWasmEdgeSpawnEnv(
 ): Record<string, string | undefined>;
 
 export function resolveWasmEdgePluginLaunchPlan(options: {
-  wasmPath: string;
+  wasmPath?: string;
   wasmEdgeBinary?: string;
   wasmEdgeRunnerBinary?: string;
   enableThreads?: boolean;
   invokeArgs?: string[];
   env?: Record<string, string | undefined>;
+  hostProfile?: "runtime-host";
 }): PluginInvokeProcessLaunchPlan;
 
 export function createPluginInvokeProcessClient(options: {
@@ -250,6 +363,14 @@ export function createPluginInvokeProcessClient(options: {
   cwd?: string;
 }): Promise<PluginInvokeProcessClient>;
 
+export function createWasmEdgeStreamProcessClient(options: {
+  launchPlan?: PluginInvokeProcessLaunchPlan;
+  command?: string;
+  args?: string[];
+  env?: Record<string, string | undefined>;
+  cwd?: string;
+}): Promise<PluginInvokeProcessClient>;
+
 export function resolveModuleHarnessLaunchPlan(options: {
   runtime?: ModuleHarnessRuntimeDescriptor;
 } | ModuleHarnessRuntimeDescriptor): PluginInvokeProcessLaunchPlan;

package/src/testing/index.js

@@ -10,6 +10,7 @@ export {
 export {
   buildWasmEdgeSpawnEnv,
   createPluginInvokeProcessClient,
+  createWasmEdgeStreamProcessClient,
   resolveWasmEdgePluginLaunchPlan,
 } from "./processInvoke.js";
 export {

package/src/testing/moduleHarness.js

@@ -1,5 +1,6 @@
 import {
   createPluginInvokeProcessClient,
+  createWasmEdgeStreamProcessClient,
   resolveWasmEdgePluginLaunchPlan,
 } from "./processInvoke.js";
 
@@ -10,6 +11,15 @@ function normalizeRuntimeDescriptor(options = {}) {
   return options;
 }
 
+function isRuntimeHostProfile(runtime = {}) {
+  return (
+    String(runtime.hostProfile ?? "").trim().toLowerCase() === "runtime-host" ||
+    Array.isArray(runtime.modules) ||
+    (typeof runtime.defaultModuleId === "string" &&
+      runtime.defaultModuleId.trim().length > 0)
+  );
+}
+
 export function resolveModuleHarnessLaunchPlan(options = {}) {
   const runtime = normalizeRuntimeDescriptor(options);
   if (runtime.launchPlan && typeof runtime.launchPlan === "object") {
@@ -41,20 +51,111 @@ export async function createModuleHarness(options = {}) {
   const runtime = normalizeRuntimeDescriptor(options);
   const kind = String(runtime.kind ?? "process").trim().toLowerCase();
   const launchPlan = resolveModuleHarnessLaunchPlan(options);
-  const processClient = await createPluginInvokeProcessClient({ launchPlan });
+  const runtimeHost = isRuntimeHostProfile(runtime);
+  const processClient =
+    kind === "wasmedge"
+      ? await createWasmEdgeStreamProcessClient({ launchPlan })
+      : await createPluginInvokeProcessClient({ launchPlan });
+  const configuredModules = Array.isArray(runtime.modules) ? [...runtime.modules] : [];
+  if (
+    kind === "wasmedge" &&
+    runtimeHost &&
+    configuredModules.length === 0 &&
+    typeof runtime.wasmPath === "string" &&
+    runtime.wasmPath.trim().length > 0
+  ) {
+    configuredModules.push({
+      moduleId:
+        typeof runtime.defaultModuleId === "string" &&
+        runtime.defaultModuleId.trim().length > 0
+          ? runtime.defaultModuleId.trim()
+          : "default",
+      wasmPath: runtime.wasmPath,
+      metadata: runtime.metadata ?? null,
+    });
+  }
+
+  for (const moduleDefinition of configuredModules) {
+    await processClient.installModule(moduleDefinition);
+  }
+
+  let defaultModuleId =
+    typeof runtime.defaultModuleId === "string" && runtime.defaultModuleId.trim().length > 0
+      ? runtime.defaultModuleId.trim()
+      : configuredModules[0]?.moduleId ?? null;
 
   return {
     runtime: {
       ...runtime,
       kind,
+      hostProfile: runtimeHost ? "runtime-host" : runtime.hostProfile,
     },
     launchPlan,
     invokeRaw(requestBytes) {
       return processClient.invokeRaw(requestBytes);
     },
     invoke(request) {
+      if (runtimeHost) {
+        if (!defaultModuleId) {
+          throw new Error(
+            "Runtime-host invoke() requires an installed defaultModuleId.",
+          );
+        }
+        return processClient.invokeModule(defaultModuleId, request);
+      }
       return processClient.invoke(request);
     },
+    async installModule(definition) {
+      const installed = await processClient.installModule(definition);
+      if (
+        runtimeHost &&
+        (!defaultModuleId || defaultModuleId.trim().length === 0) &&
+        typeof installed?.moduleId === "string" &&
+        installed.moduleId.trim().length > 0
+      ) {
+        defaultModuleId = installed.moduleId.trim();
+      }
+      return installed;
+    },
+    listModules() {
+      return processClient.listModules();
+    },
+    async unloadModule(moduleId) {
+      const unloaded = await processClient.unloadModule(moduleId);
+      if (unloaded && moduleId === defaultModuleId) {
+        if (runtimeHost) {
+          const remainingModules = await processClient.listModules();
+          defaultModuleId = remainingModules[0]?.moduleId ?? null;
+        } else {
+          defaultModuleId = null;
+        }
+      }
+      return unloaded;
+    },
+    invokeModule(moduleId, request) {
+      return processClient.invokeModule(moduleId, request);
+    },
+    appendRow(options) {
+      return processClient.appendRow(options);
+    },
+    listRows(schemaFileId = null) {
+      return processClient.listRows(schemaFileId);
+    },
+    resolveRow(handle) {
+      return processClient.resolveRow(handle);
+    },
+    queryRows(sql) {
+      return processClient.queryRows(sql);
+    },
+    allocateRegion(options) {
+      return processClient.allocateRegion(options);
+    },
+    describeRegion(regionId) {
+      return processClient.describeRegion(regionId);
+    },
+    resolveRecord(query) {
+      return processClient.resolveRecord(query);
+    },
     destroy() {
       return processClient.destroy();
     },