create-fhevm-example 1.4.5 → 1.4.6

package/dist/scripts/commands/add-mode.js

@@ -58,7 +58,7 @@ const utils_1 = require("../shared/utils");
 function detectHardhatProject(targetDir) {
     const packageJsonPath = path.join(targetDir, "package.json");
     const hardhatConfigTs = path.join(targetDir, "hardhat.config.ts");
-    const hardhatConfigJs = path.join(targetDir, "hardhat.config");
+    const hardhatConfigJs = path.join(targetDir, "hardhat.config.js");
     if (!fs.existsSync(packageJsonPath)) {
         return false;
     }
@@ -106,7 +106,7 @@ function updateHardhatConfig(targetDir) {
             ? configPathJs
             : null;
     if (!actualPath) {
-        throw new Error("hardhat.config.ts or hardhat.config.js not found");
+        throw new Error(utils_1.ERROR_MESSAGES.CONFIG_NOT_FOUND);
     }
     let content = fs.readFileSync(actualPath, "utf-8");
     if (content.includes("@fhevm/hardhat-plugin")) {
@@ -139,12 +139,12 @@ function updateHardhatConfig(targetDir) {
 function addExampleFiles(exampleName, targetDir) {
     const example = config_1.EXAMPLES[exampleName];
     if (!example) {
-        throw new Error(`Unknown example: ${exampleName}`);
+        throw new Error(utils_1.ERROR_MESSAGES.UNKNOWN_EXAMPLE(exampleName));
     }
     const rootDir = (0, utils_1.getRootDir)();
     const contractName = (0, utils_1.getContractName)(example.contract);
     if (!contractName) {
-        throw new Error("Could not extract contract name");
+        throw new Error(utils_1.ERROR_MESSAGES.CONTRACT_NAME_FAILED);
     }
     const contractSource = path.join(rootDir, example.contract);
     const testSource = path.join(rootDir, example.test);
@@ -154,15 +154,9 @@ function addExampleFiles(exampleName, targetDir) {
     if (!fs.existsSync(contractsDir)) {
         fs.mkdirSync(contractsDir, { recursive: true });
     }
-    if (fs.existsSync(contractDest)) {
-        // File exists - will be handled by caller with prompts
-        fs.copyFileSync(contractSource, contractDest);
-        p.log.success(`Overwritten: ${contractName}.sol`);
-    }
-    else {
-        fs.copyFileSync(contractSource, contractDest);
-        p.log.success(`Added: ${contractName}.sol`);
-    }
+    const isContractOverwrite = fs.existsSync(contractDest);
+    fs.copyFileSync(contractSource, contractDest);
+    p.log.success(`${isContractOverwrite ? "Overwritten" : "Added"}: ${contractName}.sol`);
     // Handle test file
     const testFileName = path.basename(example.test);
     const testDest = path.join(targetDir, "test", testFileName);
@@ -170,14 +164,9 @@ function addExampleFiles(exampleName, targetDir) {
     if (!fs.existsSync(testDir)) {
         fs.mkdirSync(testDir, { recursive: true });
     }
-    if (fs.existsSync(testDest)) {
-        fs.copyFileSync(testSource, testDest);
-        p.log.success(`Overwritten: ${testFileName}`);
-    }
-    else {
-        fs.copyFileSync(testSource, testDest);
-        p.log.success(`Added: ${testFileName}`);
-    }
+    const isTestOverwrite = fs.existsSync(testDest);
+    fs.copyFileSync(testSource, testDest);
+    p.log.success(`${isTestOverwrite ? "Overwritten" : "Added"}: ${testFileName}`);
     // Handle contract dependencies
     if (example.dependencies) {
         p.log.message("");

package/dist/scripts/commands/doctor.js

@@ -162,4 +162,7 @@ async function main() {
         p.outro(picocolors_1.default.green("✅ All checks passed! You are ready to develop. 🚀"));
     }
 }
-main().catch(console.error);
+main().catch((err) => {
+    console.error(err);
+    process.exit(1);
+});

package/dist/scripts/commands/generate-docs.js

@@ -187,5 +187,8 @@ async function main() {
 }
 const isMainModule = process.argv[1]?.includes("generate-docs");
 if (isMainModule) {
-    main().catch(console.error);
+    main().catch((err) => {
+        console.error(err);
+        process.exit(1);
+    });
 }

package/dist/scripts/index.js

@@ -56,6 +56,17 @@ const builders_1 = require("./shared/builders");
 const ui_1 = require("./shared/ui");
 const add_mode_1 = require("./commands/add-mode");
 const generate_docs_1 = require("./commands/generate-docs");
+// =============================================================================
+// INTERACTIVE MODE
+// =============================================================================
+/** Helper to check if user cancelled and show message */
+function checkCancel(value) {
+    if (p.isCancel(value)) {
+        p.cancel("Operation cancelled.");
+        return true;
+    }
+    return false;
+}
 /** Prompts user to select mode and returns project configuration */
 async function getProjectConfig() {
     // Build options - docs only available in local dev mode
@@ -83,22 +94,16 @@ async function getProjectConfig() {
         message: "What would you like to create?",
         options,
     });
-    if (p.isCancel(mode)) {
-        p.cancel("Operation cancelled.");
+    if (checkCancel(mode))
         return null;
-    }
     let name;
     if (mode === "single") {
         const category = await (0, ui_1.promptSelectCategory)();
-        if (p.isCancel(category)) {
-            p.cancel("Operation cancelled.");
+        if (checkCancel(category))
             return null;
-        }
         const example = await (0, ui_1.promptSelectExampleFromCategory)(category);
-        if (p.isCancel(example)) {
-            p.cancel("Operation cancelled.");
+        if (checkCancel(example))
             return null;
-        }
         name = example;
     }
     else if (mode === "docs") {
@@ -108,10 +113,8 @@ async function getProjectConfig() {
     }
     else {
         const category = await (0, ui_1.promptSelectCategoryProject)();
-        if (p.isCancel(category)) {
-            p.cancel("Operation cancelled.");
+        if (checkCancel(category))
             return null;
-        }
         name = category;
     }
     const projectName = await p.text({
@@ -119,27 +122,21 @@ async function getProjectConfig() {
         placeholder: `my-${name}-project`,
         defaultValue: `my-${name}-project`,
     });
-    if (p.isCancel(projectName)) {
-        p.cancel("Operation cancelled.");
+    if (checkCancel(projectName))
         return null;
-    }
     const outputDir = await p.text({
         message: "Output directory:",
         placeholder: `./${projectName}`,
         defaultValue: `./${projectName}`,
     });
-    if (p.isCancel(outputDir)) {
-        p.cancel("Operation cancelled.");
+    if (checkCancel(outputDir))
         return null;
-    }
     const shouldInstall = await p.confirm({
         message: "Install dependencies and run tests?",
         initialValue: false,
     });
-    if (p.isCancel(shouldInstall)) {
-        p.cancel("Operation cancelled.");
+    if (checkCancel(shouldInstall))
         return null;
-    }
     return {
         mode: mode,
         name,
@@ -205,12 +202,22 @@ async function runInteractiveMode() {
 }
 function parseArgs(args) {
     const parsed = {};
+    // Short flag mappings
+    const shortFlags = {
+        "-e": "example",
+        "-c": "category",
+        "-o": "output",
+        "-a": "add",
+        "-i": "install",
+        "-h": "help",
+    };
     for (let i = 0; i < args.length; i++) {
         const arg = args[i];
+        // Handle long flags (--example, --category, etc.)
         if (arg.startsWith("--")) {
             const key = arg.slice(2);
             const nextArg = args[i + 1];
-            if (nextArg && !nextArg.startsWith("--")) {
+            if (nextArg && !nextArg.startsWith("-")) {
                 parsed[key] = nextArg;
                 i++;
             }
@@ -218,8 +225,20 @@ function parseArgs(args) {
                 parsed[key] = true;
             }
         }
-        else if (arg === "-h") {
-            parsed["help"] = true;
+        // Handle short flags (-e, -c, -o, -a, -i, -h)
+        else if (arg.startsWith("-") && shortFlags[arg]) {
+            const key = shortFlags[arg];
+            const nextArg = args[i + 1];
+            // Flags that take values: -e, -c, -o
+            if (["example", "category", "output"].includes(key) &&
+                nextArg &&
+                !nextArg.startsWith("-")) {
+                parsed[key] = nextArg;
+                i++;
+            }
+            else {
+                parsed[key] = true;
+            }
         }
     }
     return parsed;

package/dist/scripts/shared/builders.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"builders.d.ts","sourceRoot":"","sources":["../../../scripts/shared/builders.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA6DH;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,GAChB,IAAI,CAkDN;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,GAChB,IAAI,CAwEN;AAMD;;GAEG;AACH,wBAAsB,sBAAsB,CAC1C,YAAY,EAAE,MAAM,EAAE,EACtB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CAuEf"}
+{"version":3,"file":"builders.d.ts","sourceRoot":"","sources":["../../../scripts/shared/builders.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAiEH;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,GAChB,IAAI,CAkDN;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,GAChB,IAAI,CAwEN;AAMD;;GAEG;AACH,wBAAsB,sBAAsB,CAC1C,YAAY,EAAE,MAAM,EAAE,EACtB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CA6Df"}

package/dist/scripts/shared/builders.js

@@ -69,7 +69,8 @@ function copyDependencies(dependencies, outputDir) {
     }
 }
 /**
- * Initializes git repository (optional, fails silently)
+ * Initializes git repository (optional, logs warning if skipped)
+ * @returns true if git init succeeded, false otherwise
  */
 function initGitRepo(outputDir) {
     try {
@@ -77,9 +78,11 @@ function initGitRepo(outputDir) {
             cwd: outputDir,
             stdio: "ignore",
         });
+        return true;
     }
     catch {
-        // Git init is optional
+        utils_1.log.dim("   ⚠️ Git initialization skipped (git not available)");
+        return false;
     }
 }
 // =============================================================================
@@ -208,18 +211,9 @@ async function createLocalTestProject(exampleNames, outputDir) {
             Object.assign(allNpmDeps, example.npmDependencies);
         }
     }
-    // 3. Copy dependencies
-    for (const depPath of allContractDeps) {
-        const depFullPath = path.join(rootDir, depPath);
-        if (fs.existsSync(depFullPath)) {
-            const relativePath = depPath.replace(/^contracts\//, "");
-            const depDestPath = path.join(outputDir, "contracts", relativePath);
-            const depDestDir = path.dirname(depDestPath);
-            if (!fs.existsSync(depDestDir)) {
-                fs.mkdirSync(depDestDir, { recursive: true });
-            }
-            fs.copyFileSync(depFullPath, depDestPath);
-        }
+    // 3. Copy dependencies (reuse helper function)
+    if (allContractDeps.size > 0) {
+        copyDependencies(Array.from(allContractDeps), outputDir);
     }
     // 4. Finalize project
     (0, generators_1.updateProjectPackageJson)(outputDir, "fhevm-test-project", `Testing ${exampleNames.length} examples`, Object.keys(allNpmDeps).length > 0 ? allNpmDeps : undefined);

package/dist/scripts/shared/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../scripts/shared/config.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,MAAM,WAAW,aAAa;IAC5B,yCAAyC;IACzC,QAAQ,EAAE,MAAM,CAAC;IACjB,uCAAuC;IACvC,IAAI,EAAE,MAAM,CAAC;IACb,gDAAgD;IAChD,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,uCAAuC;IACvC,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACzC,yCAAyC;IACzC,WAAW,EAAE,MAAM,CAAC;IACpB,4BAA4B;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,8CAA8C;IAC9C,UAAU,EAAE,MAAM,CAAC;IACnB,8BAA8B;IAC9B,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,cAAc;IAC7B,mBAAmB;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,yCAAyC;IACzC,SAAS,EAAE,KAAK,CAAC;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAClD;AAMD,eAAO,MAAM,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CA6RlD,CAAC;AAMF,eAAO,MAAM,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAoJrD,CAAC;AAMF;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,EAAE,CAE1C;AAED;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,MAAM,EAAE,CAE3C;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS,CAElE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAEpE;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAE3D"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../scripts/shared/config.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,MAAM,WAAW,aAAa;IAC5B,yCAAyC;IACzC,QAAQ,EAAE,MAAM,CAAC;IACjB,uCAAuC;IACvC,IAAI,EAAE,MAAM,CAAC;IACb,gDAAgD;IAChD,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,uCAAuC;IACvC,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACzC,yCAAyC;IACzC,WAAW,EAAE,MAAM,CAAC;IACpB,4BAA4B;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,8CAA8C;IAC9C,UAAU,EAAE,MAAM,CAAC;IACnB,8BAA8B;IAC9B,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,cAAc;IAC7B,mBAAmB;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,yCAAyC;IACzC,SAAS,EAAE,KAAK,CAAC;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAClD;AAMD,eAAO,MAAM,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CA+SlD,CAAC;AAMF,eAAO,MAAM,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAiKrD,CAAC;AAMF;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,EAAE,CAE1C;AAED;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,MAAM,EAAE,CAE3C;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS,CAElE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAEpE;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAE3D"}

package/dist/scripts/shared/config.js

@@ -152,14 +152,6 @@ exports.EXAMPLES = {
         docsOutput: "docs/concepts/fhe-access-control.md",
         title: "FHE Access Control"
     },
-    "fhe-anti-patterns": {
-        contract: "contracts/concepts/FHEAntiPatterns.sol",
-        test: "test/concepts/FHEAntiPatterns.ts",
-        description: "Comprehensive guide to FHE anti-patterns and their solutions. Covers 9 critical mistakes: using if/else on encrypted values, incorrect permission patterns, require() statements that leak info, unbounded loops, noise accumulation, and deprecated APIs. Each pattern shows both ❌ WRONG and ✅ CORRECT implementations.",
-        category: "Concepts",
-        docsOutput: "docs/concepts/fhe-anti-patterns.md",
-        title: "FHE Anti Patterns"
-    },
     "fhe-handles": {
         contract: "contracts/concepts/FHEHandles.sol",
         test: "test/concepts/FHEHandles.ts",
@@ -176,6 +168,30 @@ exports.EXAMPLES = {
         docsOutput: "docs/concepts/fhe-input-proof.md",
         title: "FHE Input Proof"
     },
+    "control-flow": {
+        contract: "contracts/concepts/antipatterns/ControlFlow.sol",
+        test: "test/concepts/antipatterns/ControlFlow.ts",
+        description: "Control flow anti-patterns in FHE development. Demonstrates common mistakes when using conditional logic and loops with encrypted values.",
+        category: "Concepts - Antipatterns",
+        docsOutput: "docs/concepts/antipatterns/control-flow.md",
+        title: "Control Flow"
+    },
+    "operations-gas-noise": {
+        contract: "contracts/concepts/antipatterns/OperationsGasNoise.sol",
+        test: "test/concepts/antipatterns/OperationsGasNoise.ts",
+        description: "Operations, gas, and noise anti-patterns in FHE development. Demonstrates performance issues, side-channel leaks, and inefficient encrypted computation patterns.",
+        category: "Concepts - Antipatterns",
+        docsOutput: "docs/concepts/antipatterns/operations-gas-noise.md",
+        title: "Operations Gas Noise"
+    },
+    "permissions": {
+        contract: "contracts/concepts/antipatterns/Permissions.sol",
+        test: "test/concepts/antipatterns/Permissions.ts",
+        description: "Permission management anti-patterns in FHE development. Demonstrates common mistakes with allowThis, allow, and permission propagation across transfers and contracts.",
+        category: "Concepts - Antipatterns",
+        docsOutput: "docs/concepts/antipatterns/permissions.md",
+        title: "Permissions"
+    },
     "encrypted-lottery": {
         contract: "contracts/gaming/EncryptedLottery.sol",
         test: "test/gaming/EncryptedLottery.ts",
@@ -368,10 +384,6 @@ exports.CATEGORIES = {
                 sol: "contracts/concepts/FHEAccessControl.sol",
                 test: "test/concepts/FHEAccessControl.ts",
             },
-            {
-                sol: "contracts/concepts/FHEAntiPatterns.sol",
-                test: "test/concepts/FHEAntiPatterns.ts",
-            },
             {
                 sol: "contracts/concepts/FHEHandles.sol",
                 test: "test/concepts/FHEHandles.ts",
@@ -382,6 +394,23 @@ exports.CATEGORIES = {
             }
         ],
     },
+    conceptsantipatterns: {
+        name: "Concepts - Antipatterns Examples",
+        contracts: [
+            {
+                sol: "contracts/concepts/antipatterns/ControlFlow.sol",
+                test: "test/concepts/antipatterns/ControlFlow.ts",
+            },
+            {
+                sol: "contracts/concepts/antipatterns/OperationsGasNoise.sol",
+                test: "test/concepts/antipatterns/OperationsGasNoise.ts",
+            },
+            {
+                sol: "contracts/concepts/antipatterns/Permissions.sol",
+                test: "test/concepts/antipatterns/Permissions.ts",
+            }
+        ],
+    },
     gaming: {
         name: "Gaming Examples",
         contracts: [

package/dist/scripts/shared/utils.js

@@ -108,8 +108,8 @@ exports.FHEVM_DEPENDENCIES = {
         "@fhevm/solidity": "^0.9.1",
     },
     devDependencies: {
-        "@fhevm/hardhat-plugin": "^0.3.0-1",
-        "@zama-fhe/relayer-sdk": "^0.3.0-5",
+        "@fhevm/hardhat-plugin": "^0.3.0-3",
+        "@zama-fhe/relayer-sdk": "^0.3.0-6",
     },
 };
 exports.CATEGORY_ORDER = [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "create-fhevm-example",
-  "version": "1.4.5",
+  "version": "1.4.6",
   "description": "Create FHEVM example projects with a single command - A comprehensive toolkit for building privacy-preserving smart contracts",
   "bin": {
     "create-fhevm-example": "./dist/scripts/index.js"

package/test/concepts/FHEAntiPatterns.ts

@@ -26,7 +26,8 @@ describe("FHEAntiPatterns", function () {
 
   before(async function () {
     if (!hre.fhevm.isMock) {
-      throw new Error(`This hardhat test suite cannot run on Sepolia Testnet`);
+      console.warn(`This hardhat test suite cannot run on Sepolia Testnet`);
+      this.skip();
     }
     const ethSigners: HardhatEthersSigner[] = await ethers.getSigners();
     signers = { owner: ethSigners[0], alice: ethSigners[1] };

package/test/concepts/antipatterns/ControlFlow.ts

@@ -0,0 +1,125 @@
+import {
+  FhevmType,
+  HardhatFhevmRuntimeEnvironment,
+} from "@fhevm/hardhat-plugin";
+import { HardhatEthersSigner } from "@nomicfoundation/hardhat-ethers/signers";
+import { expect } from "chai";
+import { ethers } from "hardhat";
+import * as hre from "hardhat";
+
+import {
+  FHEControlFlowAntiPatterns,
+  FHEControlFlowAntiPatterns__factory,
+} from "../types";
+import type { Signers } from "./types";
+
+async function deployFixture() {
+  const factory = (await ethers.getContractFactory(
+    "FHEControlFlowAntiPatterns"
+  )) as FHEControlFlowAntiPatterns__factory;
+  const contract = (await factory.deploy()) as FHEControlFlowAntiPatterns;
+  const contractAddress = await contract.getAddress();
+  return { contract, contractAddress };
+}
+
+/**
+ * @notice Tests for FHE control flow anti-patterns
+ * Demonstrates wrong and correct patterns for conditional logic and loops
+ */
+describe("FHEControlFlowAntiPatterns", function () {
+  let contract: FHEControlFlowAntiPatterns;
+  let contractAddress: string;
+  let signers: Signers;
+
+  before(async function () {
+    if (!hre.fhevm.isMock) {
+      throw new Error(`This hardhat test suite cannot run on Sepolia Testnet`);
+    }
+    const ethSigners: HardhatEthersSigner[] = await ethers.getSigners();
+    signers = { owner: ethSigners[0], alice: ethSigners[1] };
+  });
+
+  beforeEach(async function () {
+    const deployment = await deployFixture();
+    contractAddress = deployment.contractAddress;
+    contract = deployment.contract;
+
+    // Initialize contract with test values
+    const fhevm: HardhatFhevmRuntimeEnvironment = hre.fhevm;
+
+    // Create encrypted input for balance only (threshold is fixed at 100)
+    const input = await fhevm
+      .createEncryptedInput(contractAddress, signers.alice.address)
+      .add32(50) // balance
+      .encrypt();
+
+    await contract
+      .connect(signers.alice)
+      .initialize(input.handles[0], input.inputProof);
+  });
+
+  describe("Pattern 1: If/Else Branching", function () {
+    it("should execute correctConditional without leaking information", async function () {
+      const fhevm: HardhatFhevmRuntimeEnvironment = hre.fhevm;
+
+      await contract.connect(signers.alice).correctConditional();
+
+      const encrypted = await contract.getBalance();
+      const decrypted = await fhevm.userDecryptEuint(
+        FhevmType.euint32,
+        encrypted,
+        contractAddress,
+        signers.alice
+      );
+
+      // Balance (50) is not above threshold (100), so no penalty applied
+      expect(decrypted).to.equal(50);
+    });
+
+    it("wrongBranching should return placeholder value", async function () {
+      const result = await contract.wrongBranching.staticCall();
+      expect(result).to.equal(0);
+    });
+  });
+
+  describe("Pattern 2: Require/Revert", function () {
+    it("should return encrypted boolean for validation", async function () {
+      const fhevm: HardhatFhevmRuntimeEnvironment = hre.fhevm;
+
+      // Execute validation (stores result)
+      await contract.connect(signers.alice).correctValidation();
+
+      // Get result via getter
+      const encrypted = await contract.getValidationResult();
+      const decrypted = await fhevm.userDecryptEbool(
+        encrypted,
+        contractAddress,
+        signers.alice
+      );
+
+      // Balance (50) < 100, so should be false
+      expect(decrypted).to.equal(false);
+    });
+
+    it("wrongRequire should return explanation string", async function () {
+      const result = await contract.wrongRequire();
+      expect(result).to.include("doesn't work with encrypted values");
+    });
+  });
+
+  describe("Pattern 3: Encrypted Loop Iterations", function () {
+    it("should use fixed iterations with FHE.select", async function () {
+      const fhevm: HardhatFhevmRuntimeEnvironment = hre.fhevm;
+
+      await contract.connect(signers.alice).correctFixedIterations();
+
+      // Result should count up to min(balance, MAX_ITERATIONS)
+      // Balance is 50, MAX_ITERATIONS is 5, so result should be 5
+    });
+
+    it("wrongEncryptedLoop should return explanation string", async function () {
+      const result = await contract.wrongEncryptedLoop();
+      expect(result).to.include("Loop iterations leak");
+    });
+  });
+});