@cofhe/mock-contracts 0.1.0

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "@cofhe/mock-contracts",
+  "version": "0.1.0",
+  "description": "Mock smart contracts for testing CoFHE with FHE primitives locally",
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "sideEffects": false,
+  "license": "MIT",
+  "files": [
+    "dist/**",
+    "contracts/**/*.sol",
+    "foundry/**/*.sol",
+    "test/**/*.sol",
+    "foundry.toml",
+    "remappings.txt"
+  ],
+  "keywords": [
+    "blockchain",
+    "ethereum",
+    "smart-contracts",
+    "solidity",
+    "FHE",
+    "encryption",
+    "privacy",
+    "coprocessor",
+    "fhenix",
+    "cofhe"
+  ],
+  "dependencies": {
+    "@fhenixprotocol/cofhe-contracts": "0.0.13",
+    "@openzeppelin/contracts": "^5.0.0",
+    "@openzeppelin/contracts-upgradeable": "^5.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "forge-std": "github:foundry-rs/forge-std",
+    "tsup": "^8.0.2",
+    "hardhat": "^2.24.1",
+    "@cofhe/tsconfig": "0.1.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "forge build && npx tsx scripts/build-artifacts.ts && tsup",
+    "test": "forge test -vvv",
+    "test:coverage": "forge coverage",
+    "anvil": "anvil --code-size-limit 100000",
+    "lint": "echo 'No linting configured for Solidity'",
+    "type-check": "echo 'No type checking for Solidity'",
+    "clean": "rm -rf out && rm -rf cache",
+    "dev": "anvil --code-size-limit 100000"
+  }
+}

package/remappings.txt

@@ -0,0 +1,5 @@
+forge-std/=node_modules/forge-std/src/
+hardhat/=node_modules/hardhat/
+@openzeppelin/contracts/=node_modules/@openzeppelin/contracts/
+@fhenixprotocol/cofhe-contracts/=node_modules/@fhenixprotocol/cofhe-contracts/
+@cofhe/mock-contracts/=node_modules/@cofhe/mock-contracts/contracts/

package/test/TestBed.t.sol

@@ -0,0 +1,60 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+import { Test } from 'forge-std/Test.sol';
+import { TestBed } from '../contracts/TestBed.sol';
+import { CoFheTest } from '../contracts/foundry/CoFheTest.sol';
+import { FHE, InEuint32, euint8, euint256 } from '@fhenixprotocol/cofhe-contracts/FHE.sol';
+
+contract TestBedTest is Test, CoFheTest {
+  TestBed private testbed;
+
+  address private user = makeAddr('user');
+
+  function setUp() public {
+    // optional ... enable verbose logging from fhe mocks
+    // setLog(true);
+
+    testbed = new TestBed();
+  }
+
+  function testSetNumberFuzz(uint32 n) public {
+    InEuint32 memory number = createInEuint32(n, user);
+
+    //must be the user who sends transaction
+    //or else invalid permissions from fhe allow
+    vm.prank(user);
+    testbed.setNumber(number);
+
+    assertHashValue(testbed.eNumber(), n);
+  }
+
+  function testOverflow() public {
+    euint8 a = FHE.asEuint8(240);
+    euint8 b = FHE.asEuint8(240);
+    euint8 c = FHE.add(a, b);
+
+    assertHashValue(euint8.unwrap(c), (240 + 240) % 256);
+  }
+
+  function testDivideByZero() public {
+    euint8 a = FHE.asEuint8(240);
+    euint8 b = FHE.asEuint8(0);
+    euint8 c = FHE.div(a, b);
+
+    assertHashValue(euint8.unwrap(c), type(uint8).max);
+  }
+
+  function test256BitsNoOverflow() public {
+    euint256 a = FHE.asEuint256(type(uint256).max);
+    euint256 b = FHE.asEuint256(type(uint256).max);
+    euint256 c = FHE.add(a, b);
+
+    uint256 expected;
+    unchecked {
+      expected = type(uint256).max + type(uint256).max;
+    }
+
+    assertHashValue(euint256.unwrap(c), expected);
+  }
+}