@cofhe/mock-contracts 0.5.2 → 0.6.0

package/src/typechain-types/index.ts

@@ -5,4 +5,3 @@ export type { MockACL } from './MockACL';
 export type { MockTaskManager } from './MockTaskManager';
 export type { MockThresholdNetwork } from './MockThresholdNetwork';
 export type { MockZkVerifier } from './MockZkVerifier';
-export type { TestBed } from './TestBed';

package/contracts/TestBed.sol

@@ -1,103 +0,0 @@
-// SPDX-License-Identifier: UNLICENSED
-pragma solidity ^0.8.13;
-
-import { TASK_MANAGER_ADDRESS } from '@fhenixprotocol/cofhe-contracts/FHE.sol';
-import { ITaskManager } from '@fhenixprotocol/cofhe-contracts/ICofhe.sol';
-import '@fhenixprotocol/cofhe-contracts/FHE.sol';
-
-/// @title TestBed
-/// @notice Minimal contract used to smoke-test CoFHE/FHE flows (mocks + permissions + decrypt plumbing).
-/// @dev This contract is intentionally simple and is primarily a convenience for Hardhat-based local
-///      development: the Hardhat plugin can deploy it automatically when `deployTestBed` is enabled.
-///
-///      In Foundry flows, nothing deploys or depends on this contract automatically — tests must
-///      instantiate it explicitly (this repository includes such Foundry tests). You may deploy/use it
-///      in `forge test` if you want a known-good reference target.
-///
-///      Important concepts:
-///      - `eNumber` is an on-chain encrypted handle type (`euint32`). In real CoFHE, this represents a ciphertext.
-///      - `numberHash` stores the unwrapped handle (`ctHash`) as a uint256 for easy inspection/assertions.
-///      - After every state update we call `FHE.allowThis(...)` and `FHE.allowSender(...)` so the contract
-///        and the transaction sender can continue operating on / decrypting the updated handle.
-contract TestBed {
-  euint32 public eNumber;
-  bytes32 public numberHash;
-
-  /// @notice Marker used by deploy scripts/tests to confirm the contract is deployed.
-  function exists() public pure returns (bool) {
-    return true;
-  }
-
-  /// @notice Sets `eNumber` from an encrypted input struct.
-  /// @dev Typically used when testing client-side encryption flows.
-  function setNumber(InEuint32 memory inNumber) public {
-    eNumber = FHE.asEuint32(inNumber);
-    numberHash = euint32.unwrap(eNumber);
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Convenience setter that casts a plaintext value into an encrypted handle.
-  /// @dev Useful for quick smoke tests that don't need pre-encryption.
-  function setNumberTrivial(uint32 inNumber) public {
-    eNumber = FHE.asEuint32(inNumber);
-    numberHash = euint32.unwrap(eNumber);
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Increments `eNumber` by 1 using FHE arithmetic.
-  function increment() public {
-    eNumber = FHE.add(eNumber, FHE.asEuint32(1));
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Adds an encrypted input to `eNumber`.
-  function add(InEuint32 memory inNumber) public {
-    eNumber = FHE.add(eNumber, FHE.asEuint32(inNumber));
-
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Subtracts an encrypted input from `eNumber`, clamped to 0 to avoid underflow.
-  function sub(InEuint32 memory inNumber) public {
-    euint32 inAsEuint32 = FHE.asEuint32(inNumber);
-    euint32 eSubOrZero = FHE.select(FHE.lte(inAsEuint32, eNumber), inAsEuint32, FHE.asEuint32(0));
-    eNumber = FHE.sub(eNumber, eSubOrZero);
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Multiplies `eNumber` by an encrypted input.
-  function mul(InEuint32 memory inNumber) public {
-    eNumber = FHE.mul(eNumber, FHE.asEuint32(inNumber));
-    FHE.allowThis(eNumber);
-    FHE.allowSender(eNumber);
-  }
-
-  /// @notice Requests decryption of `eNumber`.
-  /// @dev In real CoFHE this is asynchronous; in mocks it is simulated.
-  function decrypt() public {
-    // The standalone `FHE.decrypt(...)` helper was removed in
-    // fhenixprotocol/cofhe-contracts 0.1.3, so we now go through the
-    // task manager interface directly.
-    ITaskManager(TASK_MANAGER_ADDRESS).createDecryptTask(uint256(numberHash), msg.sender);
-  }
-
-  /// @notice Reads a decryption result (reverts if not ready depending on implementation).
-  function getDecryptResult(euint32 input1) public view returns (uint32) {
-    return FHE.getDecryptResult(input1);
-  }
-
-  /// @notice Reads a decryption result safely, returning a readiness flag.
-  function getDecryptResultSafe(euint32 input1) public view returns (uint32 value, bool decrypted) {
-    return FHE.getDecryptResultSafe(input1);
-  }
-
-  /// @notice Publishes a decrypt result for an encrypted handle.
-  function publishDecryptResult(euint32 input, uint32 result, bytes memory signature) external {
-    FHE.publishDecryptResult(input, result, signature);
-  }
-}

package/src/TestBed.ts

@@ -1,309 +0,0 @@
-// This file is auto-generated by build-artifacts.ts
-import { type MockArtifact } from './types';
-
-export const TestBedArtifact = {
-  contractName: 'TestBed',
-  isFixed: true,
-  fixedAddress: '0x0000000000000000000000000000000000005003',
-  abi: [
-    {
-      type: 'function',
-      name: 'add',
-      inputs: [
-        {
-          name: 'inNumber',
-          type: 'tuple',
-          internalType: 'struct InEuint32',
-          components: [
-            {
-              name: 'ctHash',
-              type: 'uint256',
-              internalType: 'uint256',
-            },
-            {
-              name: 'securityZone',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'utype',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'signature',
-              type: 'bytes',
-              internalType: 'bytes',
-            },
-          ],
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'decrypt',
-      inputs: [],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'eNumber',
-      inputs: [],
-      outputs: [
-        {
-          name: '',
-          type: 'bytes32',
-          internalType: 'euint32',
-        },
-      ],
-      stateMutability: 'view',
-    },
-    {
-      type: 'function',
-      name: 'exists',
-      inputs: [],
-      outputs: [
-        {
-          name: '',
-          type: 'bool',
-          internalType: 'bool',
-        },
-      ],
-      stateMutability: 'pure',
-    },
-    {
-      type: 'function',
-      name: 'getDecryptResult',
-      inputs: [
-        {
-          name: 'input1',
-          type: 'bytes32',
-          internalType: 'euint32',
-        },
-      ],
-      outputs: [
-        {
-          name: '',
-          type: 'uint32',
-          internalType: 'uint32',
-        },
-      ],
-      stateMutability: 'view',
-    },
-    {
-      type: 'function',
-      name: 'getDecryptResultSafe',
-      inputs: [
-        {
-          name: 'input1',
-          type: 'bytes32',
-          internalType: 'euint32',
-        },
-      ],
-      outputs: [
-        {
-          name: 'value',
-          type: 'uint32',
-          internalType: 'uint32',
-        },
-        {
-          name: 'decrypted',
-          type: 'bool',
-          internalType: 'bool',
-        },
-      ],
-      stateMutability: 'view',
-    },
-    {
-      type: 'function',
-      name: 'increment',
-      inputs: [],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'mul',
-      inputs: [
-        {
-          name: 'inNumber',
-          type: 'tuple',
-          internalType: 'struct InEuint32',
-          components: [
-            {
-              name: 'ctHash',
-              type: 'uint256',
-              internalType: 'uint256',
-            },
-            {
-              name: 'securityZone',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'utype',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'signature',
-              type: 'bytes',
-              internalType: 'bytes',
-            },
-          ],
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'numberHash',
-      inputs: [],
-      outputs: [
-        {
-          name: '',
-          type: 'bytes32',
-          internalType: 'bytes32',
-        },
-      ],
-      stateMutability: 'view',
-    },
-    {
-      type: 'function',
-      name: 'publishDecryptResult',
-      inputs: [
-        {
-          name: 'input',
-          type: 'bytes32',
-          internalType: 'euint32',
-        },
-        {
-          name: 'result',
-          type: 'uint32',
-          internalType: 'uint32',
-        },
-        {
-          name: 'signature',
-          type: 'bytes',
-          internalType: 'bytes',
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'setNumber',
-      inputs: [
-        {
-          name: 'inNumber',
-          type: 'tuple',
-          internalType: 'struct InEuint32',
-          components: [
-            {
-              name: 'ctHash',
-              type: 'uint256',
-              internalType: 'uint256',
-            },
-            {
-              name: 'securityZone',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'utype',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'signature',
-              type: 'bytes',
-              internalType: 'bytes',
-            },
-          ],
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'setNumberTrivial',
-      inputs: [
-        {
-          name: 'inNumber',
-          type: 'uint32',
-          internalType: 'uint32',
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'function',
-      name: 'sub',
-      inputs: [
-        {
-          name: 'inNumber',
-          type: 'tuple',
-          internalType: 'struct InEuint32',
-          components: [
-            {
-              name: 'ctHash',
-              type: 'uint256',
-              internalType: 'uint256',
-            },
-            {
-              name: 'securityZone',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'utype',
-              type: 'uint8',
-              internalType: 'uint8',
-            },
-            {
-              name: 'signature',
-              type: 'bytes',
-              internalType: 'bytes',
-            },
-          ],
-        },
-      ],
-      outputs: [],
-      stateMutability: 'nonpayable',
-    },
-    {
-      type: 'error',
-      name: 'InvalidEncryptedInput',
-      inputs: [
-        {
-          name: 'got',
-          type: 'uint8',
-          internalType: 'uint8',
-        },
-        {
-          name: 'expected',
-          type: 'uint8',
-          internalType: 'uint8',
-        },
-      ],
-    },
-    {
-      type: 'error',
-      name: 'SecurityZoneOutOfBounds',
-      inputs: [
-        {
-          name: 'value',
-          type: 'int32',
-          internalType: 'int32',
-        },
-      ],
-    },
-  ],
-} as const satisfies MockArtifact;

package/src/typechain-types/TestBed.ts

@@ -1,172 +0,0 @@
-/* Autogenerated file. Do not edit manually. */
-/* tslint:disable */
-/* eslint-disable */
-import type {
-  BaseContract,
-  BigNumberish,
-  BytesLike,
-  FunctionFragment,
-  Result,
-  Interface,
-  ContractRunner,
-  ContractMethod,
-  Listener,
-} from 'ethers';
-import type {
-  TypedContractEvent,
-  TypedDeferredTopicFilter,
-  TypedEventLog,
-  TypedListener,
-  TypedContractMethod,
-} from './common';
-
-export type InEuint32Struct = {
-  ctHash: BigNumberish;
-  securityZone: BigNumberish;
-  utype: BigNumberish;
-  signature: BytesLike;
-};
-
-export type InEuint32StructOutput = [ctHash: bigint, securityZone: bigint, utype: bigint, signature: string] & {
-  ctHash: bigint;
-  securityZone: bigint;
-  utype: bigint;
-  signature: string;
-};
-
-export interface TestBedInterface extends Interface {
-  getFunction(
-    nameOrSignature:
-      | 'add'
-      | 'decrypt'
-      | 'eNumber'
-      | 'exists'
-      | 'getDecryptResult'
-      | 'getDecryptResultSafe'
-      | 'increment'
-      | 'mul'
-      | 'numberHash'
-      | 'publishDecryptResult'
-      | 'setNumber'
-      | 'setNumberTrivial'
-      | 'sub'
-  ): FunctionFragment;
-
-  encodeFunctionData(functionFragment: 'add', values: [InEuint32Struct]): string;
-  encodeFunctionData(functionFragment: 'decrypt', values?: undefined): string;
-  encodeFunctionData(functionFragment: 'eNumber', values?: undefined): string;
-  encodeFunctionData(functionFragment: 'exists', values?: undefined): string;
-  encodeFunctionData(functionFragment: 'getDecryptResult', values: [BytesLike]): string;
-  encodeFunctionData(functionFragment: 'getDecryptResultSafe', values: [BytesLike]): string;
-  encodeFunctionData(functionFragment: 'increment', values?: undefined): string;
-  encodeFunctionData(functionFragment: 'mul', values: [InEuint32Struct]): string;
-  encodeFunctionData(functionFragment: 'numberHash', values?: undefined): string;
-  encodeFunctionData(functionFragment: 'publishDecryptResult', values: [BytesLike, BigNumberish, BytesLike]): string;
-  encodeFunctionData(functionFragment: 'setNumber', values: [InEuint32Struct]): string;
-  encodeFunctionData(functionFragment: 'setNumberTrivial', values: [BigNumberish]): string;
-  encodeFunctionData(functionFragment: 'sub', values: [InEuint32Struct]): string;
-
-  decodeFunctionResult(functionFragment: 'add', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'decrypt', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'eNumber', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'exists', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'getDecryptResult', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'getDecryptResultSafe', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'increment', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'mul', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'numberHash', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'publishDecryptResult', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'setNumber', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'setNumberTrivial', data: BytesLike): Result;
-  decodeFunctionResult(functionFragment: 'sub', data: BytesLike): Result;
-}
-
-export interface TestBed extends BaseContract {
-  connect(runner?: ContractRunner | null): TestBed;
-  waitForDeployment(): Promise<this>;
-
-  interface: TestBedInterface;
-
-  queryFilter<TCEvent extends TypedContractEvent>(
-    event: TCEvent,
-    fromBlockOrBlockhash?: string | number | undefined,
-    toBlock?: string | number | undefined
-  ): Promise<Array<TypedEventLog<TCEvent>>>;
-  queryFilter<TCEvent extends TypedContractEvent>(
-    filter: TypedDeferredTopicFilter<TCEvent>,
-    fromBlockOrBlockhash?: string | number | undefined,
-    toBlock?: string | number | undefined
-  ): Promise<Array<TypedEventLog<TCEvent>>>;
-
-  on<TCEvent extends TypedContractEvent>(event: TCEvent, listener: TypedListener<TCEvent>): Promise<this>;
-  on<TCEvent extends TypedContractEvent>(
-    filter: TypedDeferredTopicFilter<TCEvent>,
-    listener: TypedListener<TCEvent>
-  ): Promise<this>;
-
-  once<TCEvent extends TypedContractEvent>(event: TCEvent, listener: TypedListener<TCEvent>): Promise<this>;
-  once<TCEvent extends TypedContractEvent>(
-    filter: TypedDeferredTopicFilter<TCEvent>,
-    listener: TypedListener<TCEvent>
-  ): Promise<this>;
-
-  listeners<TCEvent extends TypedContractEvent>(event: TCEvent): Promise<Array<TypedListener<TCEvent>>>;
-  listeners(eventName?: string): Promise<Array<Listener>>;
-  removeAllListeners<TCEvent extends TypedContractEvent>(event?: TCEvent): Promise<this>;
-
-  add: TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-
-  decrypt: TypedContractMethod<[], [void], 'nonpayable'>;
-
-  eNumber: TypedContractMethod<[], [string], 'view'>;
-
-  exists: TypedContractMethod<[], [boolean], 'view'>;
-
-  getDecryptResult: TypedContractMethod<[input1: BytesLike], [bigint], 'view'>;
-
-  getDecryptResultSafe: TypedContractMethod<
-    [input1: BytesLike],
-    [[bigint, boolean] & { value: bigint; decrypted: boolean }],
-    'view'
-  >;
-
-  increment: TypedContractMethod<[], [void], 'nonpayable'>;
-
-  mul: TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-
-  numberHash: TypedContractMethod<[], [string], 'view'>;
-
-  publishDecryptResult: TypedContractMethod<
-    [input: BytesLike, result: BigNumberish, signature: BytesLike],
-    [void],
-    'nonpayable'
-  >;
-
-  setNumber: TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-
-  setNumberTrivial: TypedContractMethod<[inNumber: BigNumberish], [void], 'nonpayable'>;
-
-  sub: TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-
-  getFunction<T extends ContractMethod = ContractMethod>(key: string | FunctionFragment): T;
-
-  getFunction(nameOrSignature: 'add'): TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'decrypt'): TypedContractMethod<[], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'eNumber'): TypedContractMethod<[], [string], 'view'>;
-  getFunction(nameOrSignature: 'exists'): TypedContractMethod<[], [boolean], 'view'>;
-  getFunction(nameOrSignature: 'getDecryptResult'): TypedContractMethod<[input1: BytesLike], [bigint], 'view'>;
-  getFunction(
-    nameOrSignature: 'getDecryptResultSafe'
-  ): TypedContractMethod<[input1: BytesLike], [[bigint, boolean] & { value: bigint; decrypted: boolean }], 'view'>;
-  getFunction(nameOrSignature: 'increment'): TypedContractMethod<[], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'mul'): TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'numberHash'): TypedContractMethod<[], [string], 'view'>;
-  getFunction(
-    nameOrSignature: 'publishDecryptResult'
-  ): TypedContractMethod<[input: BytesLike, result: BigNumberish, signature: BytesLike], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'setNumber'): TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'setNumberTrivial'): TypedContractMethod<[inNumber: BigNumberish], [void], 'nonpayable'>;
-  getFunction(nameOrSignature: 'sub'): TypedContractMethod<[inNumber: InEuint32Struct], [void], 'nonpayable'>;
-
-  filters: {};
-}