test: add unit tests

contracts/token/ERC20/EncryptedWETH.sol

@@ -6,8 +6,6 @@ import { EncryptedERC20 } from "./EncryptedERC20.sol";
 import "fhevm/lib/TFHE.sol";
 import "fhevm/gateway/GatewayCaller.sol";
 
-import "hardhat/console.sol";
-
 /**
  * @title             EncryptedWETH
  * @notice            This contract allows users to wrap/unwrap trustlessly
@@ -63,6 +61,13 @@ abstract contract EncryptedWETH is EncryptedERC20, GatewayCaller {
         wrap();
     }
 
+    /**
+     * @notice         Receive function calls wrap().
+     */
+    receive() external payable {
+        wrap();
+    }
+
     /**
      * @notice         Unwrap EncryptedERC20 tokens to ether.
      * @param amount   Amount to unwrap.
@@ -115,8 +120,6 @@ abstract contract EncryptedWETH is EncryptedERC20, GatewayCaller {
         UnwrapRequest memory unwrapRequest = unwrapRequests[requestId];
         delete unwrapRequests[requestId];
 
-        console.log(canUnwrap);
-
         if (canUnwrap) {
             _unsafeBurn(unwrapRequest.account, TFHE.asEuint64(unwrapRequest.amount));
             _totalSupply -= unwrapRequest.amount;
@@ -133,10 +136,8 @@ abstract contract EncryptedWETH is EncryptedERC20, GatewayCaller {
             }
 
             emit Unwrap(unwrapRequest.account, unwrapRequest.amount);
-            console.log("YES");
         } else {
             emit UnwrapFail(unwrapRequest.account, unwrapRequest.amount);
-            console.log("FAIL");
         }
 
         delete isAccountRestricted[unwrapRequest.account];

test/encryptedERC20/EncryptedERC20Wrapped.fixture.ts

@@ -0,0 +1,43 @@
+import { ethers } from "hardhat";
+
+import type { ERC20Mintable, EncryptedERC20Wrapped, TestEncryptedERC20Wrapped } from "../../types";
+import { Signers } from "../signers";
+
+export async function deployERC20AndEncryptedERC20WrappedFixture(
+  signers: Signers,
+  name: string,
+  symbol: string,
+  decimals: number,
+): Promise<[ERC20Mintable, TestEncryptedERC20Wrapped]> {
+  const contractFactoryERC20Mintable = await ethers.getContractFactory("ERC20Mintable");
+  const contractERC20 = await contractFactoryERC20Mintable
+    .connect(signers.alice)
+    .deploy(name, symbol, decimals, signers.alice.address);
+  await contractERC20.waitForDeployment();
+
+  const contractFactoryEncryptedERC20Wrapped = await ethers.getContractFactory("TestEncryptedERC20Wrapped");
+  const contractEncryptedERC20Wrapped = await contractFactoryEncryptedERC20Wrapped
+    .connect(signers.alice)
+    .deploy(contractERC20.getAddress());
+  await contractEncryptedERC20Wrapped.waitForDeployment();
+
+  return [contractERC20, contractEncryptedERC20Wrapped];
+}
+
+export async function mintAndWrap(
+  signers: Signers,
+  user: string,
+  plainToken: ERC20Mintable,
+  token: EncryptedERC20Wrapped,
+  tokenAddress: string,
+  amount: bigint,
+): Promise<void> {
+  let tx = await plainToken.connect(signers[user as keyof Signers]).mint(amount);
+  await tx.wait();
+
+  tx = await plainToken.connect(signers[user as keyof Signers]).approve(tokenAddress, amount);
+  await tx.wait();
+
+  tx = await token.connect(signers[user as keyof Signers]).wrap(amount);
+  await tx.wait();
+}