example-smart-wallet/packages/hardhat/test/utils.ts

import { type SignerWithAddress } from "@nomicfoundation/hardhat-ethers/signers";
import { type Permission } from "@luxfhe/sdk";
import { type HardhatRuntimeEnvironment } from "hardhat/types";

export const getTokensFromFaucet = async (hre: HardhatRuntimeEnvironment, address: string) => {
  if (hre.network.name === "localluxfhe") {
    if ((await hre.ethers.provider.getBalance(address)).toString() === "0") {
      await hre.luxfhe.getFunds(address);
    }
  }
};

export const createLuxFHEContractPermission = async (
  hre: HardhatRuntimeEnvironment,
  signer: SignerWithAddress,
  contractAddress: string,
): Promise<Permission> => {
  const provider = hre.ethers.provider;

  const permit = await hre.luxfhe.generatePermit(contractAddress, provider, signer);
  const permission = hre.luxfhe.extractPermitPermission(permit);

  return permission;
};

export const unsealMockFheOpsSealed = (sealed: string): bigint => {
  const byteArray = new Uint8Array(sealed.split("").map(c => c.charCodeAt(0)));

  // Step 2: Convert the Uint8Array to BigInt
  let result = BigInt(0);
  for (let i = 0; i < byteArray.length; i++) {
    result = (result << BigInt(8)) + BigInt(byteArray[i]); // Shift and add each byte
  }

  return result;
};

export const takeSnapshot = async (hre: HardhatRuntimeEnvironment): Promise<string> => {
  return hre.network.provider.send("evm_snapshot");
};

export const revertSnapshot = async (hre: HardhatRuntimeEnvironment, snapshot: string) => {
  return hre.network.provider.send("evm_revert", [snapshot]);
};

export const withinSnapshot = async <T>(
  hre: HardhatRuntimeEnvironment,
  body: () => Promise<T>,
): Promise<T | undefined> => {
  const tmp = await takeSnapshot(hre);
  try {
    const res = await body();
    return res;
  } finally {
    await revertSnapshot(hre, tmp);
  }
};

export const getLastBlockTimestamp = async (hre: HardhatRuntimeEnvironment) => {
  // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
  return (await hre.ethers.provider.getBlock("latest"))!.timestamp;
};