implement configure balance tracker

cloudwalk · Mar 14, 2024 · 4b9fa21 · 4b9fa21
1 parent fac06a5
commit 4b9fa21
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/e2e-contracts/integration/test/integration.ts b/e2e-contracts/integration/test/integration.ts
@@ -86,7 +86,12 @@ async function deployBalanceTracker() {
 }
 
 async function configureBalanceTracker() {
-  //TODO: Set hooks
+  const REVERT_POLICY = BigInt(1);
+  const toHookStruct = (hook : IERC20Hookable.HookStructOutput) => ({account: hook.account, policy: hook.policy});
+  const hooksOutput = await brlCoin.getAfterTokenTransferHooks();
+  const hooks: IERC20Hookable.HookStruct[] = hooksOutput.map(toHookStruct);
+  const newHook = {account: await balanceTracker.getAddress(), policy: REVERT_POLICY};
+  hooks.push(newHook);
 }
 
 async function deployYieldStreamer() {
@@ -145,6 +150,7 @@ describe("Integration Test", function () {
 
     it("Configure BalanceTracker", async function () {
       await configureBalanceTracker();
+      console.log(await brlCoin.getAfterTokenTransferHooks());
     });
 
     it("Deploy YieldStreamer", async function () {