@chainlink/cre-sdk 1.17.0-alpha.1 → 1.17.0-alpha.2

package/dist/generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen.js

@@ -53,5 +53,3 @@ class CronTrigger {
         return rawOutput;
     }
 }
-export class CronRestrictor {
-}

package/dist/sdk/cre/index.d.ts

@@ -1,31 +1,27 @@
 /**
  * Public API for the CRE SDK.
  */
-import { ClientCapability as AptosClient, ClientRestrictor as AptosRestrictor } from '../../generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen';
-import { ClientCapability as EVMClient, ClientRestrictor as EVMRestrictor } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
-import { ClientCapability as SolanaClient, ClientRestrictor as SolanaRestrictor } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
-import { ClientCapability as ConfidentialHTTPClient, ClientRestrictor as ConfidentialHTTPRestrictor } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
-import { ClientCapability as HTTPClient, ClientRestrictor as HTTPClientRestrictor } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
-import { HTTPCapability, HTTPRestrictor } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
-import { CronCapability, CronRestrictor } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
+import { ClientCapability as EVMClient } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
+import { ClientCapability as SolanaClient } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
+import { ClientCapability as ConfidentialHTTPClient } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
+import { ClientCapability as HTTPClient } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
+import { HTTPCapability } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
+import { CronCapability } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
 /**
  * Public exports for the CRE SDK.
  */
 export { type Log as EVMLog, TxStatus, } from '../../generated/capabilities/blockchain/evm/v1alpha/client_pb';
-export { ReceiverContractExecutionStatus as SolanaReceiverContractExecutionStatus, TxStatus as SolanaTxStatus, } from '../../generated/capabilities/blockchain/solana/v1alpha/client_pb';
+export { ComparisonOperator as SolanaComparisonOperator, type CPIFilterConfigJson as SolanaCPIFilterConfigJson, type FilterLogTriggerRequestJson as SolanaFilterLogTriggerRequestJson, type Log as SolanaLog, ReceiverContractExecutionStatus as SolanaReceiverContractExecutionStatus, type SubkeyConfigJson as SolanaSubkeyConfigJson, TxStatus as SolanaTxStatus, type ValueComparatorJson as SolanaValueComparatorJson, } from '../../generated/capabilities/blockchain/solana/v1alpha/client_pb';
 export type { Payload as HTTPPayload } from '../../generated/capabilities/networking/http/v1alpha/trigger_pb';
 export type { Payload as CronPayload } from '../../generated/capabilities/scheduler/cron/v1/trigger_pb';
-export { TeeType } from '../../generated/sdk/v1alpha/sdk_pb';
-export { ClientCapability as AptosClient, ClientRestrictor as AptosRestrictor, type WriteCreReportRequest as AptosWriteCreReportRequest, type WriteCreReportRequestJson as AptosWriteCreReportRequestJson, } from '../../generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen';
-export { ClientCapability as EVMClient, ClientRestrictor as EVMRestrictor, type WriteCreReportRequest, type WriteCreReportRequestJson, } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
-export { ClientCapability as SolanaClient, ClientRestrictor as SolanaRestrictor, type WriteCreReportRequest as SolanaWriteCreReportRequest, type WriteCreReportRequestJson as SolanaWriteCreReportRequestJson, } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
-export { ClientCapability as ConfidentialHTTPClient, ClientRestrictor as ConfidentialHTTPRestrictor, } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
-export { ClientCapability as HTTPClient, ClientRestrictor as HTTPClientRestrictor, type SendRequester as HTTPSendRequester, } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
-export { HTTPCapability, HTTPRestrictor, } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
-export { CronCapability, CronRestrictor, } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
-export type { NodeRuntime, Runtime, TeeRuntime } from '../runtime';
-export type { AnyTeeConstraint, NitroBinding, NitroRegion, OneOfTees, Region, TeeBinding, TeeConstraint, } from '../workflow';
-export { handler, handlerInTee, NITRO_REGIONS, REGIONS } from '../workflow';
+export { ClientCapability as EVMClient, type WriteCreReportRequest, type WriteCreReportRequestJson, } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
+export { ClientCapability as SolanaClient, type WriteCreReportRequest as SolanaWriteCreReportRequest, type WriteCreReportRequestJson as SolanaWriteCreReportRequestJson, } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
+export { ClientCapability as ConfidentialHTTPClient } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
+export { ClientCapability as HTTPClient, type SendRequester as HTTPSendRequester, } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
+export { HTTPCapability } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
+export { CronCapability } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
+export type { NodeRuntime, Runtime } from '../runtime';
+export { handler } from '../workflow';
 export declare const cre: {
     capabilities: {
         CronCapability: typeof CronCapability;
@@ -33,18 +29,7 @@ export declare const cre: {
         ConfidentialHTTPClient: typeof ConfidentialHTTPClient;
         HTTPClient: typeof HTTPClient;
         EVMClient: typeof EVMClient;
-        AptosClient: typeof AptosClient;
         SolanaClient: typeof SolanaClient;
     };
-    restrictors: {
-        CronRestrictor: typeof CronRestrictor;
-        HTTPRestrictor: typeof HTTPRestrictor;
-        ConfidentialHTTPRestrictor: typeof ConfidentialHTTPRestrictor;
-        HTTPClientRestrictor: typeof HTTPClientRestrictor;
-        EVMRestrictor: typeof EVMRestrictor;
-        SolanaRestrictor: typeof SolanaRestrictor;
-        AptosRestrictor: typeof AptosRestrictor;
-    };
-    handler: <TRawTriggerOutput extends import("@bufbuild/protobuf").Message<string>, TTriggerOutput, TConfig, TResult, TRuntime = import("../runtime").Runtime<TConfig>>(trigger: import("../utils/triggers/trigger-interface").Trigger<TRawTriggerOutput, TTriggerOutput>, fn: import("../workflow").HandlerFn<TConfig, TTriggerOutput, TResult, TRuntime>, hooks?: import("../workflow").Hooks<TConfig, TTriggerOutput>) => import("../workflow").HandlerEntry<TConfig, TRawTriggerOutput, TTriggerOutput, TResult, TRuntime>;
-    handlerInTee: <TRawTriggerOutput extends import("@bufbuild/protobuf").Message<string>, TTriggerOutput, TConfig, TResult>(trigger: import("../utils/triggers/trigger-interface").Trigger<TRawTriggerOutput, TTriggerOutput>, fn: import("../workflow").HandlerFn<TConfig, TTriggerOutput, TResult, import("../runtime").TeeRuntime<TConfig>>, tees: import("../workflow").TeeConstraint, hooks?: import("../workflow").Hooks<TConfig, TTriggerOutput>) => import("../workflow").HandlerEntry<TConfig, TRawTriggerOutput, TTriggerOutput, TResult, import("../runtime").TeeRuntime<TConfig>>;
+    handler: <TRawTriggerOutput extends import("@bufbuild/protobuf").Message<string>, TTriggerOutput, TConfig, TResult>(trigger: import("..").Trigger<TRawTriggerOutput, TTriggerOutput>, fn: import("../workflow").HandlerFn<TConfig, TTriggerOutput, TResult>) => import("../workflow").HandlerEntry<TConfig, TRawTriggerOutput, TTriggerOutput, TResult>;
 };

package/dist/sdk/cre/index.js

@@ -1,35 +1,31 @@
 /**
  * Public API for the CRE SDK.
  */
-import { ClientCapability as AptosClient, ClientRestrictor as AptosRestrictor, } from '../../generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen';
-import { ClientCapability as EVMClient, ClientRestrictor as EVMRestrictor, } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
-import { ClientCapability as SolanaClient, ClientRestrictor as SolanaRestrictor, } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
-import { ClientCapability as ConfidentialHTTPClient, ClientRestrictor as ConfidentialHTTPRestrictor, } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
-import { ClientCapability as HTTPClient, ClientRestrictor as HTTPClientRestrictor, } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
-import { HTTPCapability, HTTPRestrictor, } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
-import { CronCapability, CronRestrictor, } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
+import { ClientCapability as EVMClient } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
+import { ClientCapability as SolanaClient } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
+import { ClientCapability as ConfidentialHTTPClient } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
+import { ClientCapability as HTTPClient } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
+import { HTTPCapability } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
+import { CronCapability } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
 import { prepareRuntime } from '../utils/prepare-runtime';
-import { handler, handlerInTee } from '../workflow';
+import { handler } from '../workflow';
 /**
  * Public exports for the CRE SDK.
  */
 export { TxStatus, } from '../../generated/capabilities/blockchain/evm/v1alpha/client_pb';
-export { ReceiverContractExecutionStatus as SolanaReceiverContractExecutionStatus, TxStatus as SolanaTxStatus, } from '../../generated/capabilities/blockchain/solana/v1alpha/client_pb';
-export { TeeType } from '../../generated/sdk/v1alpha/sdk_pb';
-// Aptos Capability
-export { ClientCapability as AptosClient, ClientRestrictor as AptosRestrictor, } from '../../generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen';
+export { ComparisonOperator as SolanaComparisonOperator, ReceiverContractExecutionStatus as SolanaReceiverContractExecutionStatus, TxStatus as SolanaTxStatus, } from '../../generated/capabilities/blockchain/solana/v1alpha/client_pb';
 // EVM Capability
-export { ClientCapability as EVMClient, ClientRestrictor as EVMRestrictor, } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
+export { ClientCapability as EVMClient, } from '../../generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen';
 // Solana Capability
-export { ClientCapability as SolanaClient, ClientRestrictor as SolanaRestrictor, } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
+export { ClientCapability as SolanaClient, } from '../../generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen';
 // Confidential HTTP Capability
-export { ClientCapability as ConfidentialHTTPClient, ClientRestrictor as ConfidentialHTTPRestrictor, } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
+export { ClientCapability as ConfidentialHTTPClient } from '../../generated-sdk/capabilities/networking/confidentialhttp/v1alpha/client_sdk_gen';
 // HTTP Capability
-export { ClientCapability as HTTPClient, ClientRestrictor as HTTPClientRestrictor, } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
-export { HTTPCapability, HTTPRestrictor, } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
+export { ClientCapability as HTTPClient, } from '../../generated-sdk/capabilities/networking/http/v1alpha/client_sdk_gen';
+export { HTTPCapability } from '../../generated-sdk/capabilities/networking/http/v1alpha/http_sdk_gen';
 // CRON Capability
-export { CronCapability, CronRestrictor, } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
-export { handler, handlerInTee, NITRO_REGIONS, REGIONS } from '../workflow';
+export { CronCapability } from '../../generated-sdk/capabilities/scheduler/cron/v1/cron_sdk_gen';
+export { handler } from '../workflow';
 prepareRuntime();
 export const cre = {
     capabilities: {
@@ -38,18 +34,7 @@ export const cre = {
         ConfidentialHTTPClient,
         HTTPClient,
         EVMClient,
-        AptosClient,
         SolanaClient,
     },
-    restrictors: {
-        CronRestrictor,
-        HTTPRestrictor,
-        ConfidentialHTTPRestrictor,
-        HTTPClientRestrictor,
-        EVMRestrictor,
-        SolanaRestrictor,
-        AptosRestrictor,
-    },
     handler,
-    handlerInTee,
 };

package/dist/sdk/impl/runtime-impl.d.ts

@@ -1,6 +1,6 @@
 import { type Message } from '@bufbuild/protobuf';
 import { type AwaitCapabilitiesRequest, type AwaitCapabilitiesResponse, type AwaitSecretsRequest, type AwaitSecretsResponse, type CapabilityRequest, type GetSecretsRequest, Mode, type Secret, type SecretRequest, type SecretRequestJson } from '../../generated/sdk/v1alpha/sdk_pb';
-import type { BaseRuntime, CallCapabilityParams, NodeRuntime, ReportRequest, ReportRequestJson, Runtime, TeeRuntime } from '..';
+import type { BaseRuntime, CallCapabilityParams, NodeRuntime, ReportRequest, ReportRequestJson, Runtime } from '..';
 import type { Report } from '../report';
 import { type ConsensusAggregation, type PrimitiveTypes, type UnwrapOptions } from '../utils';
 /**
@@ -44,10 +44,7 @@ export declare class BaseRuntimeImpl<C> implements BaseRuntime<C> {
     now(): Date;
     sleep(ms: number): void;
     log(message: string): void;
-    emitMetric(name: string, value: number, type: MetricType, labels?: Record<string, string>): boolean;
 }
-/** Ergonomic union for {@link BaseRuntimeImpl.emitMetric}. */
-export type MetricType = 'counter' | 'gauge';
 /**
  * It is used when a BFT guarantee cannot be provided automatically (e.g. calling a standard API).
  * You tell each node to perform a task on its own.
@@ -97,25 +94,6 @@ export declare class RuntimeImpl<C> extends BaseRuntimeImpl<C> implements Runtim
         result: () => Report;
     };
 }
-export declare class TeeRuntimeImpl<C> implements TeeRuntime<C> {
-    config: C;
-    nextCallId: number;
-    private readonly runtime;
-    constructor(config: C, nextCallId: number, helpers: RuntimeHelpers, maxResponseSize: bigint);
-    getSecret(request: SecretRequest | SecretRequestJson): {
-        result: () => Secret;
-    };
-    now(): Date;
-    log(message: string): void;
-    emitMetric(name: string, value: number, type: MetricType, labels?: Record<string, string>): boolean;
-    callCapability<I extends Message, O extends Message>({ capabilityId, method, payload, inputSchema, outputSchema, }: CallCapabilityParams<I, O>): {
-        result: () => O;
-    };
-    reportFromDon(input: ReportRequest | ReportRequestJson): {
-        result: () => Report;
-    };
-    usingTheDons(): Runtime<C>;
-}
 /**
  * Interface to the WASM host environment.
  * Provides low-level access to capabilities, secrets, and utilities.
@@ -137,9 +115,4 @@ export interface RuntimeHelpers {
     sleep(ms: number): void;
     /** Logs a message to the host environment. */
     log(message: string): void;
-    /**
-     * Emits a user metric to the host. Payload is a protobuf-encoded
-     * `workflows.v2.WorkflowUserMetric`. Returns false if the host rejected it.
-     */
-    emitMetric(payload: Uint8Array): boolean;
 }

package/dist/sdk/impl/runtime-impl.js

@@ -1,8 +1,7 @@
-import { create, toBinary } from '@bufbuild/protobuf';
+import { create } from '@bufbuild/protobuf';
 import { anyPack, anyUnpack } from '@bufbuild/protobuf/wkt';
 import { deserializeErrorFromString } from '../../capabilities/errors';
 import { AwaitCapabilitiesRequestSchema, AwaitSecretsRequestSchema, CapabilityRequestSchema, GetSecretsRequestSchema, Mode, SecretRequestSchema, SimpleConsensusInputsSchema, } from '../../generated/sdk/v1alpha/sdk_pb';
-import { UserMetricType, WorkflowUserMetricSchema, } from '../../generated/workflows/v2/workflow_user_metric_pb';
 import { ConsensusCapability } from '../../generated-sdk/capabilities/internal/consensus/v1alpha/consensus_sdk_gen';
 import { Value, } from '../utils';
 import { CapabilityRuntimeError, DonModeError, NodeModeError, SecretsError } from '../errors';
@@ -140,20 +139,7 @@ export class BaseRuntimeImpl {
     log(message) {
         this.helpers.log(message);
     }
-    emitMetric(name, value, type, labels) {
-        const metric = create(WorkflowUserMetricSchema, {
-            name,
-            value,
-            type: METRIC_TYPE_TO_PROTO[type],
-            labels: labels ?? {},
-        });
-        return this.helpers.emitMetric(toBinary(WorkflowUserMetricSchema, metric));
-    }
 }
-const METRIC_TYPE_TO_PROTO = {
-    counter: UserMetricType.COUNTER,
-    gauge: UserMetricType.GAUGE,
-};
 /**
  * It is used when a BFT guarantee cannot be provided automatically (e.g. calling a standard API).
  * You tell each node to perform a task on its own.
@@ -335,43 +321,6 @@ export class RuntimeImpl extends BaseRuntimeImpl {
         };
     }
 }
-export class TeeRuntimeImpl {
-    config;
-    nextCallId;
-    runtime;
-    constructor(config, nextCallId, helpers, maxResponseSize) {
-        this.config = config;
-        this.nextCallId = nextCallId;
-        this.runtime = new RuntimeImpl(config, nextCallId, helpers, maxResponseSize);
-    }
-    getSecret(request) {
-        return this.runtime.getSecret(request);
-    }
-    now() {
-        return this.runtime.now();
-    }
-    log(message) {
-        this.runtime.log(message);
-    }
-    emitMetric(name, value, type, labels) {
-        return this.runtime.emitMetric(name, value, type, labels);
-    }
-    callCapability({ capabilityId, method, payload, inputSchema, outputSchema, }) {
-        return this.runtime.callCapability({
-            capabilityId,
-            method,
-            payload,
-            inputSchema,
-            outputSchema,
-        });
-    }
-    reportFromDon(input) {
-        return this.runtime.report(input);
-    }
-    usingTheDons() {
-        return this.runtime;
-    }
-}
 function clearIgnoredFields(value, descriptor) {
     if (!descriptor || !value) {
         return;

package/dist/sdk/runtime.d.ts

@@ -1,11 +1,10 @@
 import type { Message } from '@bufbuild/protobuf';
 import type { GenMessage } from '@bufbuild/protobuf/codegenv2';
 import type { ReportRequest, ReportRequestJson } from '../generated/sdk/v1alpha/sdk_pb';
-import type { MetricType } from './impl/runtime-impl';
 import type { Report } from './report';
 import type { ConsensusAggregation, PrimitiveTypes, UnwrapOptions } from './utils';
 import type { SecretsProvider } from '.';
-export type { ReportRequest, ReportRequestJson, MetricType };
+export type { ReportRequest, ReportRequestJson };
 export type CallCapabilityParams<I extends Message, O extends Message> = {
     capabilityId: string;
     method: string;
@@ -24,7 +23,6 @@ export interface BaseRuntime<C> {
     };
     now(): Date;
     log(message: string): void;
-    emitMetric(name: string, value: number, type: MetricType, labels?: Record<string, string>): boolean;
 }
 /**
  * Runtime for Node mode execution.
@@ -32,26 +30,6 @@ export interface BaseRuntime<C> {
 export interface NodeRuntime<C> extends BaseRuntime<C> {
     readonly _isNodeRuntime: true;
 }
-/**
- * Runtime for Tee mode execution.
- */
-export interface TeeRuntime<C> extends BaseRuntime<C>, SecretsProvider {
-    /**
-     * Generates a report from the DON.
-     * Data requestsed throught this method will be routed outside of the TEE.
-     *
-     * @param input - Report request to generate a report from the DON
-     * @returns Report generated from the DON
-     */
-    reportFromDon(input: ReportRequest | ReportRequestJson): {
-        result: () => Report;
-    };
-    /**
-     * Returns the runtime that makes requests to the CRE DONs.
-     * Requests made through this runtime will therefore be routed outside of the TEE
-     */
-    usingTheDons(): Runtime<C>;
-}
 /**
  * Runtime for DON mode execution.
  */

package/dist/sdk/test/contract-mock-core.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Chain-agnostic core of the per-contract mock routing used by
+ * `addContractMock` (EVM) and `addSolanaContractMock` (Solana).
+ *
+ * Builds a handler that routes a request to `handle` when `matches` returns
+ * true, falls through to the previously installed handler otherwise, and
+ * throws `noMatchError` when there is nothing to fall through to. Installing
+ * the returned handler over the previous one is what lets multiple contract
+ * mocks chain on the same capability mock.
+ */
+export declare function chainContractHandler<TReq, TReply>(options: {
+    previous: ((req: TReq) => TReply) | undefined;
+    matches: (req: TReq) => boolean;
+    handle: (req: TReq) => TReply;
+    noMatchError: (req: TReq) => string;
+}): (req: TReq) => TReply;

package/dist/sdk/test/contract-mock-core.js

@@ -0,0 +1,21 @@
+/**
+ * Chain-agnostic core of the per-contract mock routing used by
+ * `addContractMock` (EVM) and `addSolanaContractMock` (Solana).
+ *
+ * Builds a handler that routes a request to `handle` when `matches` returns
+ * true, falls through to the previously installed handler otherwise, and
+ * throws `noMatchError` when there is nothing to fall through to. Installing
+ * the returned handler over the previous one is what lets multiple contract
+ * mocks chain on the same capability mock.
+ */
+export function chainContractHandler(options) {
+    const { previous, matches, handle, noMatchError } = options;
+    return (req) => {
+        if (!matches(req)) {
+            if (previous)
+                return previous(req);
+            throw new Error(noMatchError(req));
+        }
+        return handle(req);
+    };
+}

package/dist/sdk/test/evm-contract-mock.js

@@ -1,4 +1,5 @@
 import { decodeFunctionData, encodeFunctionResult, } from 'viem';
+import { chainContractHandler } from './contract-mock-core';
 function bytesToHexAddress(bytes) {
     return `0x${Array.from(bytes)
         .map((b) => b.toString(16).padStart(2, '0'))
@@ -49,66 +50,67 @@ export function addContractMock(evmMock, options) {
     const mock = {};
     const normalizedAddress = options.address.toLowerCase();
     const previousCallContract = evmMock.callContract;
-    evmMock.callContract = (req) => {
-        const toBytes = req.call?.to;
-        if (!toBytes || bytesToHexAddress(toBytes) !== normalizedAddress) {
-            if (previousCallContract)
-                return previousCallContract(req);
-            throw new Error(`addContractMock: no mock registered for address ${toBytes ? bytesToHexAddress(toBytes) : '(empty)'}`);
-        }
-        const dataBytes = req.call?.data;
-        if (!dataBytes || dataBytes.length < 4) {
-            throw new Error('addContractMock: call data too short (need at least 4 bytes for selector)');
-        }
-        const callDataHex = bytesToHex(dataBytes);
-        let decoded;
-        try {
-            decoded = decodeFunctionData({
+    evmMock.callContract = chainContractHandler({
+        previous: previousCallContract,
+        matches: (req) => {
+            const toBytes = req.call?.to;
+            return !!toBytes && bytesToHexAddress(toBytes) === normalizedAddress;
+        },
+        noMatchError: (req) => `addContractMock: no mock registered for address ${req.call?.to ? bytesToHexAddress(req.call.to) : '(empty)'}`,
+        handle: (req) => {
+            const dataBytes = req.call?.data;
+            if (!dataBytes || dataBytes.length < 4) {
+                throw new Error('addContractMock: call data too short (need at least 4 bytes for selector)');
+            }
+            const callDataHex = bytesToHex(dataBytes);
+            let decoded;
+            try {
+                decoded = decodeFunctionData({
+                    abi: options.abi,
+                    data: callDataHex,
+                });
+            }
+            catch (e) {
+                if (previousCallContract)
+                    return previousCallContract(req);
+                throw new Error(`addContractMock: failed to decode function data for ${options.address}: ${e instanceof Error ? e.message : e}`);
+            }
+            const handler = mock[decoded.functionName];
+            if (typeof handler !== 'function') {
+                throw new Error(`addContractMock: no handler set for ${decoded.functionName} on ${options.address}`);
+            }
+            const result = handler(...(decoded.args ?? []));
+            const encoded = encodeFunctionResult({
                 abi: options.abi,
-                data: callDataHex,
+                functionName: decoded.functionName,
+                result: result,
             });
-        }
-        catch (e) {
-            if (previousCallContract)
-                return previousCallContract(req);
-            throw new Error(`addContractMock: failed to decode function data for ${options.address}: ${e instanceof Error ? e.message : e}`);
-        }
-        const handler = mock[decoded.functionName];
-        if (typeof handler !== 'function') {
-            throw new Error(`addContractMock: no handler set for ${decoded.functionName} on ${options.address}`);
-        }
-        const result = handler(...(decoded.args ?? []));
-        const encoded = encodeFunctionResult({
-            abi: options.abi,
-            functionName: decoded.functionName,
-            result: result,
-        });
-        return {
-            data: hexToUint8Array(encoded),
-        };
-    };
+            return {
+                data: hexToUint8Array(encoded),
+            };
+        },
+    });
     const previousWriteReport = evmMock.writeReport;
-    evmMock.writeReport = (req) => {
-        const receiverBytes = req.receiver;
-        if (!receiverBytes || bytesToHexAddress(receiverBytes) !== normalizedAddress) {
-            if (previousWriteReport)
-                return previousWriteReport(req);
-            throw new Error(`addContractMock: no writeReport mock registered for receiver ${receiverBytes ? bytesToHexAddress(receiverBytes) : '(empty)'}`);
-        }
-        if (typeof mock.writeReport !== 'function') {
-            throw new Error(`addContractMock: no writeReport handler set for ${options.address}`);
-        }
-        if (!req.report) {
-            throw new Error(`addContractMock: writeReport called without report for ${options.address}`);
-        }
-        if (!req.gasConfig) {
-            throw new Error(`addContractMock: writeReport called without gasConfig for ${options.address}`);
-        }
-        return mock.writeReport({
-            receiver: req.receiver,
-            report: req.report,
-            gasConfig: req.gasConfig,
-        });
-    };
+    evmMock.writeReport = chainContractHandler({
+        previous: previousWriteReport,
+        matches: (req) => !!req.receiver && bytesToHexAddress(req.receiver) === normalizedAddress,
+        noMatchError: (req) => `addContractMock: no writeReport mock registered for receiver ${req.receiver ? bytesToHexAddress(req.receiver) : '(empty)'}`,
+        handle: (req) => {
+            if (typeof mock.writeReport !== 'function') {
+                throw new Error(`addContractMock: no writeReport handler set for ${options.address}`);
+            }
+            if (!req.report) {
+                throw new Error(`addContractMock: writeReport called without report for ${options.address}`);
+            }
+            if (!req.gasConfig) {
+                throw new Error(`addContractMock: writeReport called without gasConfig for ${options.address}`);
+            }
+            return mock.writeReport({
+                receiver: req.receiver,
+                report: req.report,
+                gasConfig: req.gasConfig,
+            });
+        },
+    });
     return mock;
 }

package/dist/sdk/test/generated/index.d.ts

@@ -1,5 +1,4 @@
 /** Auto-generated barrel of capability mocks. Do not edit. */
-export { AptosMock } from './capabilities/blockchain/aptos/v1alpha/aptos_mock_gen';
 export { EvmMock } from './capabilities/blockchain/evm/v1alpha/evm_mock_gen';
 export { SolanaMock } from './capabilities/blockchain/solana/v1alpha/solana_mock_gen';
 export { BasicTestActionTriggerMock } from './capabilities/internal/actionandtrigger/v1/basic_test_action_trigger_mock_gen';

package/dist/sdk/test/generated/index.js

@@ -1,5 +1,4 @@
 /** Auto-generated barrel of capability mocks. Do not edit. */
-export { AptosMock } from './capabilities/blockchain/aptos/v1alpha/aptos_mock_gen';
 export { EvmMock } from './capabilities/blockchain/evm/v1alpha/evm_mock_gen';
 export { SolanaMock } from './capabilities/blockchain/solana/v1alpha/solana_mock_gen';
 export { BasicTestActionTriggerMock } from './capabilities/internal/actionandtrigger/v1/basic_test_action_trigger_mock_gen';

package/dist/sdk/test/index.d.ts

@@ -6,3 +6,4 @@ export { type CapabilityHandler, DEFAULT_MAX_RESPONSE_SIZE_BYTES, getTestCapabil
 export { TestWriter } from '../testutils/test-writer';
 export { type AddContractMockOptions, addContractMock, type ContractMock, type WriteReportMockInput, } from './evm-contract-mock';
 export * from './generated';
+export { type AddSolanaContractMockOptions, addSolanaContractMock, type SolanaContractMock, type SolanaWriteReportMockInput, } from './solana-contract-mock';

package/dist/sdk/test/index.js

@@ -6,3 +6,4 @@ export { DEFAULT_MAX_RESPONSE_SIZE_BYTES, getTestCapabilityHandler, newTestRunti
 export { TestWriter } from '../testutils/test-writer';
 export { addContractMock, } from './evm-contract-mock';
 export * from './generated';
+export { addSolanaContractMock, } from './solana-contract-mock';

package/dist/sdk/test/solana-contract-mock.d.ts

@@ -0,0 +1,54 @@
+import type { AccountMeta, ComputeConfig, WriteReportReply, WriteReportReplyJson } from '../../generated/capabilities/blockchain/solana/v1alpha/client_pb';
+import type { ReportResponse } from '../../generated/sdk/v1alpha/sdk_pb';
+import type { SolanaMock } from './generated';
+/**
+ * Strict version of {@link WriteReportRequest} where `report` is guaranteed
+ * to be present. Used by mock handlers so tests don't need to check for
+ * undefined.
+ */
+export interface SolanaWriteReportMockInput {
+    receiver: Uint8Array;
+    report: ReportResponse;
+    remainingAccounts: AccountMeta[];
+    computeConfig?: ComputeConfig;
+}
+/**
+ * A program mock returned by {@link addSolanaContractMock}.
+ *
+ * The Solana CRE capability is write-only, so the only routable handler is
+ * `writeReport`. When set, write-report calls targeting this program's ID are
+ * routed here; calls for other receivers chain to previously registered mocks.
+ */
+export interface SolanaContractMock {
+    writeReport?: (input: SolanaWriteReportMockInput) => WriteReportReply | WriteReportReplyJson;
+}
+export interface AddSolanaContractMockOptions {
+    /** The receiver program ID — base58 string or 32 raw bytes. */
+    programId: string | Uint8Array;
+}
+/**
+ * Registers a typed program mock on a {@link SolanaMock} instance.
+ *
+ * This is the Solana counterpart of {@link addContractMock}: it intercepts
+ * `writeReport` on the provided mock, routing calls by receiver program ID.
+ * Multiple programs can be mocked on the same `SolanaMock` — each call to
+ * `addSolanaContractMock` chains with the previous handler.
+ *
+ * @example
+ * ```ts
+ * const solanaMock = SolanaMock.testInstance(chainSelector);
+ *
+ * const dataStorage = addSolanaContractMock(solanaMock, {
+ *   programId: 'ECL8142j2YQAvs9R9geSsRnkVH2wLEi7soJCRyJ74cfL',
+ * });
+ *
+ * dataStorage.writeReport = ({ report, remainingAccounts }) => {
+ *   return { txSignature: new Uint8Array(64) };
+ * };
+ * ```
+ *
+ * @param solanaMock - The `SolanaMock` instance to attach to.
+ * @param options - The receiver program ID to route on.
+ * @returns A mock object with a settable `writeReport` handler.
+ */
+export declare function addSolanaContractMock(solanaMock: SolanaMock, options: AddSolanaContractMockOptions): SolanaContractMock;