mocks-cofhe/contracts/TestBed.sol

// SPDX-License-Identifier: UNLICENSED
pragma solidity ^0.8.13;

import "@luxfhe/cofhe-contracts/FHE.sol";

contract TestBed {
    euint32 public eNumber;
    uint256 public numberHash;

    function exists() public pure returns (bool) {
        return true;
    }

    function setNumber(InEuint32 memory inNumber) public {
        eNumber = FHE.asEuint32(inNumber);
        numberHash = euint32.unwrap(eNumber);
        FHE.allowThis(eNumber);
        FHE.allowSender(eNumber);
    }

    function setNumberTrivial(uint256 inNumber) public {
        eNumber = FHE.asEuint32(inNumber);
        numberHash = euint32.unwrap(eNumber);
        FHE.allowThis(eNumber);
        FHE.allowSender(eNumber);
    }

    function increment() public {
        eNumber = FHE.add(eNumber, FHE.asEuint32(1));
        FHE.allowThis(eNumber);
        FHE.allowSender(eNumber);
    }

    function add(InEuint32 memory inNumber) public {
        eNumber = FHE.add(eNumber, FHE.asEuint32(inNumber));
        FHE.allowThis(eNumber);
        FHE.allowSender(eNumber);
    }

    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);
    }

    function mul(InEuint32 memory inNumber) public {
        eNumber = FHE.mul(eNumber, FHE.asEuint32(inNumber));
        FHE.allowThis(eNumber);
        FHE.allowSender(eNumber);
    }

    function decrypt() public {
        FHE.decrypt(eNumber);
    }

    function getDecryptResult(euint32 input1) public view returns (uint32) {
        return FHE.getDecryptResult(input1);
    }

    function getDecryptResultSafe(
        euint32 input1
    ) public view returns (uint32 value, bool decrypted) {
        return FHE.getDecryptResultSafe(input1);
    }
}