test(test-suite): delegate user decryption, happy path test

test from FHE.sol call to MultichanACL transaction

Author: rudy-6-4

.github/workflows/test-suite-e2e-tests.yml

@@ -182,6 +182,11 @@ jobs:
         run: |
           ./fhevm-cli test public-decrypt-http-mixed
 
+      - name: Delegate User Decryption (partial test)
+        working-directory: test-suite/fhevm
+        run: |
+          ./fhevm-cli test delegate-user-decryption
+
       - name: Show logs on test failure
         working-directory: test-suite/fhevm
         if: always()

test-suite/e2e/contracts/DelegateUserDecryptDelegate.sol

@@ -0,0 +1,24 @@
+// SPDX-License-Identifier: BSD-3-Clause-Clear
+
+pragma solidity ^0.8.24;
+
+import "@fhevm/solidity/lib/FHE.sol";
+import {E2ECoprocessorConfig} from "./E2ECoprocessorConfigLocal.sol";
+
+
+/// @notice Contract for demonstrating user decryption with delegation
+contract DelegateUserDecryptDelegate is E2ECoprocessorConfig {
+    ebool public storedZBool;
+
+    function useBool(ebool xBool) public returns (ebool zBool) {
+        ebool yBool = FHE.asEbool(true);
+        zBool = FHE.and(xBool, yBool);
+        FHE.allowThis(zBool);
+        FHE.allow(zBool, msg.sender);
+        storedZBool = zBool;
+    }
+
+    function getResult() public view returns (ebool) {
+        return storedZBool;
+    }
+}

test-suite/e2e/contracts/DelegateUserDecryptDelegator.sol

@@ -0,0 +1,28 @@
+// SPDX-License-Identifier: BSD-3-Clause-Clear
+
+pragma solidity ^0.8.24;
+
+import "@fhevm/solidity/lib/FHE.sol";
+import {E2ECoprocessorConfig} from "./E2ECoprocessorConfigLocal.sol";
+
+
+/// @notice Contract for demonstrating user decryption with delegation
+contract DelegateUserDecryptDelegator is E2ECoprocessorConfig {
+    /// @dev Encrypted boolean
+    ebool public xBool;
+
+    /// @notice Constructor to initialize encrypted values and set permissions
+    constructor() {
+        xBool = FHE.asEbool(true);
+        FHE.allowThis(xBool);
+        FHE.allow(xBool, msg.sender);
+    }
+
+    function delegate(address contract_delegate_address) public {
+        FHE.delegateUserDecryption(msg.sender, contract_delegate_address, uint64(block.timestamp + 1 days));
+    }
+
+    function revoke(address contract_delegate_address) public {
+        FHE.revokeUserDecryptionDelegation(msg.sender, contract_delegate_address);
+    }
+}

test-suite/e2e/test/delegateUserDecrypt/delegateUserDecryption.ts

@@ -0,0 +1,51 @@
+import { expect } from 'chai';
+import { ethers } from 'hardhat';
+
+import { createInstances } from '../instance';
+import { getSigners, initSigners } from '../signers';
+import { userDecryptSingleHandle } from '../utils';
+
+describe('Delegate user decryption', function () {
+  before(async function () {
+    await initSigners(2);
+    this.signers = await getSigners();
+    this.instances = await createInstances(this.signers);
+    const contractFactory = await ethers.getContractFactory('DelegateUserDecryptDelegator');
+
+    this.contract = await contractFactory.connect(this.signers.alice).deploy();
+    await this.contract.waitForDeployment();
+    this.contractAddress = await this.contract.getAddress();
+
+    const contractFactoryDelegate = await ethers.getContractFactory('DelegateUserDecryptDelegate');
+
+    this.contractDelegate = await contractFactoryDelegate.connect(this.signers.alice).deploy();
+    await this.contractDelegate.waitForDeployment();
+    this.contractDelegateAddress = await this.contractDelegate.getAddress();
+
+    this.instances = await createInstances(this.signers);
+  });
+
+  const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
+  it('test delegate user decrypt ebool', async function () {
+    const block_time = 1000; // ms
+    const { publicKey, privateKey } = this.instances.alice.generateKeypair();
+    const handle1 = await this.contract.xBool();
+    const tx_use_bool = await this.contractDelegate.useBool(handle1);
+    await tx_use_bool.wait(1); // wait for 1 block
+    const tx_handle2 = await this.contractDelegate.getResult();
+    console.log("Handle2:", tx_handle2);
+    const handle2 = tx_handle2;
+
+    const delegate = await this.contract.delegate(this.contractDelegateAddress);
+    const delegate_result = await delegate.wait(1);
+    expect(delegate_result.status).to.equal(1);
+
+    await sleep( 15 * block_time); // wait for 15 seconds to ensure delegation is active
+
+    const revoke = await this.contract.revoke(this.contractDelegateAddress);
+    const revoke_result = await revoke.wait(1);
+    expect(revoke_result.status).to.equal(1);
+    await sleep( 15 * block_time); // wait for 15 seconds to ensure delegation is revoked
+
+  });
+});

test-suite/fhevm/fhevm-cli

@@ -23,22 +23,22 @@ export CONNECTOR_KMS_WORKER_VERSION=${CONNECTOR_KMS_WORKER_VERSION:-"b815aff"}
 export CONNECTOR_TX_SENDER_VERSION=${CONNECTOR_TX_SENDER_VERSION:-"b815aff"}
 
 # Coprocessor services.
-export DB_MIGRATION_VERSION=${DB_MIGRATION_VERSION:-"v0.9.5"}
+export DB_MIGRATION_VERSION=${DB_MIGRATION_VERSION:-"22fa74e"}
 export GW_LISTENER_VERSION=${GW_LISTENER_VERSION:-"v0.9.5"}
-export HOST_LISTENER_VERSION=${HOST_LISTENER_VERSION:-"v0.9.5"}
-export TX_SENDER_VERSION=${TX_SENDER_VERSION:-"v0.9.5"}
+export HOST_LISTENER_VERSION=${HOST_LISTENER_VERSION:-"22fa74e"}
+export TX_SENDER_VERSION=${TX_SENDER_VERSION:-"22fa74e"}
 export TFHE_WORKER_VERSION=${TFHE_WORKER_VERSION:-"v0.9.5"}
 export SNS_WORKER_VERSION=${SNS_WORKER_VERSION:-"v0.9.5"}
 export ZKPROOF_WORKER_VERSION=${ZKPROOF_WORKER_VERSION:-"v0.9.5"}
 
 # Gateway and Host contracts.
-export GATEWAY_VERSION=${GATEWAY_VERSION:-"b815aff"}
-export HOST_VERSION=${HOST_VERSION:-"5d96886"}
+export GATEWAY_VERSION=${GATEWAY_VERSION:-"22fa74e"}
+export HOST_VERSION=${HOST_VERSION:-"22fa74e"}
 
 # Other services.
 export CORE_VERSION=${CORE_VERSION:-"v0.12.4"}
 export RELAYER_VERSION=${RELAYER_VERSION:-"v0.5.0-2"}
-export TEST_SUITE_VERSION=${TEST_SUITE_VERSION:-"v0.9.5"}
+export TEST_SUITE_VERSION=${TEST_SUITE_VERSION:-"22fa74e"}
 
 
 function print_logo() {
@@ -176,6 +176,17 @@ case $COMMAND in
           log_message="${LIGHT_BLUE}${BOLD}[TEST] USER DECRYPTION${RESET}"
           docker_args+=("-g" "test user decrypt")
           ;;
+        delegate-user-decryption)
+          log_message="${LIGHT_BLUE}${BOLD}[TEST] USER DECRYPTION${RESET}"
+          docker_args+=("-g" "test delegate user decrypt")
+          echo -e "${log_message}"
+          docker exec fhevm-test-suite-e2e-debug "${docker_args[@]}"
+          echo Checking transaction-sender logs for DelegateUserDecryption
+          ./fhevm-cli logs coprocessor-transaction-sender | grep "DelegateUserDecryption txn succeeded"
+          echo Checking transaction-sender logs for RevokeUserDecryptionDelegation
+          ./fhevm-cli logs coprocessor-transaction-sender | grep "RevokeUserDecryptionDelegation txn succeeded"
+          exit 0 # the test has been called unlike other case branch
+          ;;
         public-decryption)
           log_message="${LIGHT_BLUE}${BOLD}[TEST] PUBLIC DECRYPTION${RESET}"
           docker_args+=("-g" "test async decrypt (uint.*|ebytes.* trivial|ebytes64 non-trivial|ebytes256 non-trivial with snapshot|addresses|several addresses)")