refactor: cleanup bridge code

Author: lrazovic

integration-tests/chopsticks/.papi/descriptors/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.1.0-autogenerated.697579702651170142",
+  "version": "0.1.0-autogenerated.14907376514540113357",
   "name": "@polkadot-api/descriptors",
   "files": [
     "dist"

integration-tests/chopsticks/.papi/metadata/polimec.scale

@@ -3,7 +3,6 @@
   "descriptorPath": ".papi/descriptors",
   "entries": {
     "polimec": {
-      "wsUrl": "ws://127.0.0.1:8000",
       "metadata": ".papi/metadata/polimec.scale"
     },
     "polkadot": {

integration-tests/chopsticks/.papi/polkadot-api.json

@@ -10,6 +10,13 @@ bun install
 bun papi
 ```
 
+> [!NOTE]
+> Sometimes you need to regenerate the Polimec descriptors. To do that, run:
+>
+> ```bash
+> bun papi add polimec --wasm ../../target/release/wbuild/polimec-runtime/polimec_runtime.compact.compressed.wasm
+> ```
+
 To start the chains:
 
 ```bash
@@ -22,3 +29,9 @@ To run the tests:
 bun run test
 ```
 
+
+> [!IMPORTANT]
+> TODO: Add:
+> - [ ] Polimec SA on AH: Add WETH balance to it in the Chopstick ovveride
+> - [ ] Polimec to Asset Hub: WETH transfer
+> - [ ] Polimec to Ethereum: WETH transfer

integration-tests/chopsticks/README.md

@@ -77,12 +77,6 @@ export const polimec_storage = {
           balance: INITIAL_BALANCES.DOT,
         },
       ],
-      [
-        [weth_location, Accounts.BOB],
-        {
-          balance: INITIAL_BALANCES.WETH,
-        },
-      ],
     ],
     Asset: [
       [

integration-tests/chopsticks/bun.lock

@@ -31,4 +31,14 @@ export const polkadot_hub_storage = {
       ],
     ],
   },
+  ForeignAssets: {
+    Account: [
+      [
+        [Asset.WETH, Accounts.POLIMEC],
+        {
+          balance: INITIAL_BALANCES.DOT,
+        },
+      ],
+    ],
+  },
 } as const;

integration-tests/chopsticks/bun.lockb

@@ -7,16 +7,16 @@
     "test": "env LOG_LEVEL=error bun test --timeout 25000"
   },
   "devDependencies": {
-    "@acala-network/chopsticks": "1.0.1",
+    "@acala-network/chopsticks": "1.0.2",
     "@biomejs/biome": "1.9.4",
-    "@polkadot-labs/hdkd": "0.0.10",
+    "@polkadot-labs/hdkd": "0.0.11",
     "@types/bun": "latest"
   },
   "peerDependencies": {
     "typescript": "^5.7.3"
   },
   "dependencies": {
     "@polkadot-api/descriptors": "file:.papi/descriptors",
-    "polkadot-api": "^1.8.2"
+    "polkadot-api": "^1.8.4"
   }
 }

integration-tests/chopsticks/overrides/polimec.ts

@@ -1,4 +1,5 @@
 import { Accounts } from '@/types';
+import { FixedSizeBinary } from 'polkadot-api';
 
 export const INITIAL_BALANCES = {
   USDT: 52000n * 10n ** 6n,
@@ -17,6 +18,9 @@ export const TRANSFER_AMOUNTS = {
 export const DERIVE_PATHS = {
   [Accounts.ALICE]: '//Alice',
   [Accounts.BOB]: '//Bob',
-};
+} as const;
 
 export const WETH_ADDRESS = '0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2';
+export const DEFAULT_TOPIC = FixedSizeBinary.fromArray(Array(32).fill(1));
+export const FEE_AMOUNT = 40_000_000_000n;
+export const WETH_AMOUNT = 15_000_000_000_000n;

integration-tests/chopsticks/overrides/polkadot-hub.ts

@@ -1,34 +1,36 @@
-import { type Accounts, Asset, AssetLocation, AssetSourceRelation, Chains } from '@/types';
-import { flatObject } from '@/utils.ts';
+import { type Accounts, Asset, AssetSourceRelation, Chains } from '@/types';
 import { bridge } from '@polkadot-api/descriptors';
 import { createClient } from 'polkadot-api';
 import { withPolkadotSdkCompat } from 'polkadot-api/polkadot-sdk-compat';
 import { getWsProvider } from 'polkadot-api/ws-provider/web';
 import { BaseChainManager } from './BaseManager';
 
 export class BridgerHubManagaer extends BaseChainManager {
+  private chain = Chains.BridgeHub;
+
   connect() {
-    const client = createClient(withPolkadotSdkCompat(getWsProvider(this.getChainType())));
-    const api = client.getTypedApi(bridge);
+    const provider = withPolkadotSdkCompat(getWsProvider(this.chain));
+    const client = createClient(provider);
 
     // Verify connection
-    if (!client || !api) {
-      throw new Error(`Failed to connect to ${this.getChainType()}`);
+    if (!client) {
+      throw new Error(`Failed to connect to ${this.chain}`);
     }
 
-    this.clients.set(this.getChainType(), { client, api });
+    const api = client.getTypedApi(bridge);
+    this.clients.set(this.chain, { client, api });
   }
 
   disconnect() {
-    this.clients.get(Chains.BridgeHub)?.client.destroy();
+    this.clients.get(this.chain)?.client.destroy();
   }
 
   getChainType() {
-    return Chains.BridgeHub;
+    return this.chain;
   }
 
   getXcmPallet() {
-    const api = this.getApi(Chains.BridgeHub);
+    const api = this.getApi(this.chain);
     return api.tx.PolkadotXcm;
   }
 
@@ -50,19 +52,9 @@ export class BridgerHubManagaer extends BaseChainManager {
     }
   }
 
+  // Note: On BridgeHub, there should be no balance for any asset.
+  // There is DOT, but we are not tracking it.
   async getAssetBalanceOf(account: Accounts, asset: Asset): Promise<bigint> {
-    const api = this.getApi(Chains.BridgeHub);
-    // const asset_source_relation = this.getAssetSourceRelation(asset);
-    // const asset_location = AssetLocation(asset, asset_source_relation).value;
-    // const account_balances_result = await api.apis.FungiblesApi.query_account_balances(account);
-    // if (account_balances_result.success === true && account_balances_result.value.type === 'V4') {
-    //   const assets = account_balances_result.value.value;
-    //   for (const asset of assets) {
-    //     if (Bun.deepEquals(flatObject(asset.id), flatObject(asset_location))) {
-    //       return asset.fun.value as bigint;
-    //     }
-    //   }
-    // }
     return 0n;
   }
 

integration-tests/chopsticks/package.json

@@ -7,28 +7,31 @@ import { getWsProvider } from 'polkadot-api/ws-provider/web';
 import { BaseChainManager } from './BaseManager';
 
 export class PolimecManager extends BaseChainManager {
+  private chain = Chains.Polimec;
+
   connect() {
-    const client = createClient(withPolkadotSdkCompat(getWsProvider(this.getChainType())));
-    const api = client.getTypedApi(polimec);
+    const provider = withPolkadotSdkCompat(getWsProvider(this.chain));
+    const client = createClient(provider);
 
     // Verify connection
-    if (!client || !api) {
-      throw new Error(`Failed to connect to ${this.getChainType()}`);
+    if (!client) {
+      throw new Error(`Failed to connect to ${this.chain}`);
     }
 
-    this.clients.set(this.getChainType(), { client, api });
+    const api = client.getTypedApi(polimec);
+    this.clients.set(this.chain, { client, api });
   }
 
   disconnect() {
-    this.clients.get(Chains.Polimec)?.client.destroy();
+    this.clients.get(this.chain)?.client.destroy();
   }
 
   getChainType() {
-    return Chains.Polimec;
+    return this.chain;
   }
 
   getXcmPallet() {
-    const api = this.getApi(Chains.Polimec);
+    const api = this.getApi(this.chain);
     return api.tx.PolkadotXcm;
   }
 

integration-tests/chopsticks/src/constants.ts

@@ -7,28 +7,31 @@ import { getWsProvider } from 'polkadot-api/ws-provider/web';
 import { BaseChainManager } from './BaseManager';
 
 export class PolkadotHubManager extends BaseChainManager {
+  private chain = Chains.PolkadotHub;
+
   connect() {
-    const client = createClient(withPolkadotSdkCompat(getWsProvider(this.getChainType())));
-    const api = client.getTypedApi(pah);
+    const provider = withPolkadotSdkCompat(getWsProvider(this.chain));
+    const client = createClient(provider);
 
     // Verify connection
-    if (!client || !api) {
-      throw new Error(`Failed to connect to ${this.getChainType()}`);
+    if (!client) {
+      throw new Error(`Failed to connect to ${this.chain}`);
     }
 
-    this.clients.set(this.getChainType(), { client, api });
+    const api = client.getTypedApi(pah);
+    this.clients.set(this.chain, { client, api });
   }
 
   disconnect() {
-    this.clients.get(Chains.PolkadotHub)?.client.destroy();
+    this.clients.get(this.chain)?.client.destroy();
   }
 
   getChainType() {
-    return Chains.PolkadotHub;
+    return this.chain;
   }
 
   getXcmPallet() {
-    const api = this.getApi(Chains.PolkadotHub);
+    const api = this.getApi(this.chain);
     return api.tx.PolkadotXcm;
   }
 

integration-tests/chopsticks/src/managers/BridgeHubManager.ts

@@ -6,28 +6,31 @@ import { getWsProvider } from 'polkadot-api/ws-provider/web';
 import { BaseChainManager } from './BaseManager';
 
 export class PolkadotManager extends BaseChainManager {
+  private chain = Chains.Polkadot as const;
+
   connect() {
-    const client = createClient(withPolkadotSdkCompat(getWsProvider(this.getChainType())));
-    const api = client.getTypedApi(polkadot);
+    const provider = withPolkadotSdkCompat(getWsProvider(this.chain));
+    const client = createClient(provider);
 
     // Verify connection
-    if (!client || !api) {
-      throw new Error(`Failed to connect to ${this.getChainType()}`);
+    if (!client) {
+      throw new Error(`Failed to connect to ${this.chain}`);
     }
 
-    this.clients.set(this.getChainType(), { client, api });
+    const api = client.getTypedApi(polkadot);
+    this.clients.set(this.chain, { client, api });
   }
 
   disconnect() {
-    this.clients.get(Chains.Polkadot)?.client.destroy();
+    this.clients.get(this.chain)?.client.destroy();
   }
 
   getChainType() {
-    return Chains.Polkadot;
+    return this.chain;
   }
 
   getXcmPallet() {
-    const api = this.getApi(Chains.Polkadot);
+    const api = this.getApi(this.chain);
     return api.tx.XcmPallet;
   }
 

integration-tests/chopsticks/src/managers/PolimecManager.ts

@@ -44,13 +44,11 @@ export class ChainSetup {
     this.relaychain = relaychainSetup.chain;
     this.bridgeHub = bridgeHubSetup.chain;
 
-    await Promise.all([
-      connectVertical(this.relaychain, this.polimec),
-      connectVertical(this.relaychain, this.assetHub),
-      connectVertical(this.relaychain, this.bridgeHub),
-      connectParachains([this.polimec, this.assetHub]),
-      connectParachains([this.bridgeHub, this.assetHub]),
-    ]);
+    const parachains = [this.polimec, this.assetHub, this.bridgeHub];
+    for (const parachain of parachains) {
+      await connectVertical(this.relaychain, parachain);
+    }
+    await connectParachains(parachains);
 
     console.log('✅ HRMP channels created');
 

integration-tests/chopsticks/src/managers/PolkadotHubManager.ts

@@ -1,9 +1,9 @@
 import { afterAll, beforeAll, beforeEach, describe, test } from 'bun:test';
 import { TRANSFER_AMOUNTS } from '@/constants';
 import { createChainManager } from '@/managers/Factory';
+import { polimec_storage } from '@/polimec';
 import { ChainSetup } from '@/setup';
 import { BridgeToPolimecTransfer } from '@/transfers/BridgeToPolimec';
-import { HubToPolimecTransfer } from '@/transfers/HubToPolimec';
 import { Accounts, Asset, AssetSourceRelation, Chains } from '@/types';
 
 describe('Bridge Hub -> Polimec Transfer Tests', () => {
@@ -13,7 +13,7 @@ describe('Bridge Hub -> Polimec Transfer Tests', () => {
   const transferTest = new BridgeToPolimecTransfer(sourceManager, hopManager, destManager);
   const chainSetup = new ChainSetup();
 
-  beforeAll(async () => await chainSetup.initialize());
+  beforeAll(async () => await chainSetup.initialize(polimec_storage));
   beforeEach(() => {
     sourceManager.connect();
     hopManager.connect();

integration-tests/chopsticks/src/managers/PolkadotManager.ts

@@ -47,14 +47,4 @@ describe('Polkadot Hub -> Polimec Transfer Tests', () => {
       }),
     { timeout: 25000 },
   );
-
-  // test(
-  //   'Send WETH to Polimec',
-  //   () =>
-  //     transferTest.testTransfer({
-  //       account: Accounts.ALICE,
-  //       assets: [[Asset.WETH, TRANSFER_AMOUNTS.BRIDGED, AssetSourceRelation.Self]],
-  //     }),
-  //   { timeout: 25000 },
-  // );
 });

integration-tests/chopsticks/src/setup.ts

@@ -1,6 +1,13 @@
 import { expect } from 'bun:test';
 import type { BaseChainManager } from '@/managers/BaseManager';
-import type { Accounts, Asset, AssetSourceRelation, BalanceCheck, TransferResult } from '@/types';
+import {
+  type Accounts,
+  type Asset,
+  type AssetSourceRelation,
+  type BalanceCheck,
+  Chains,
+  type TransferResult,
+} from '@/types';
 import { sleep } from 'bun';
 
 export interface TransferOptions {
@@ -26,15 +33,26 @@ export abstract class BaseTransferTest {
   ): void;
 
   async testTransfer(options: TransferOptions) {
-    // const { asset_balances: initialBalances } = await this.getBalances(options);
-    // if (options.assets[0][1] > initialBalances[0].source) {
-    //   throw new Error(`Insufficient balance on Source chain for asset: ${options.assets[0][0]}`);
-    // }
-    const blockNumbers = await this.executeTransfer(options);
-    // await this.waitForBlocks();
-    // await this.verifyExecution();
-    // const { asset_balances: finalBalances } = await this.getBalances(options);
-    // this.verifyFinalBalances(initialBalances, finalBalances, options);
+    // Note: For the bridged tests we use the dry-run Runtime API, so we don't write any data to the chain.
+    const isBridged = this.sourceManager.getChainType() === Chains.BridgeHub;
+
+    let initialBalances: BalanceCheck[] = [];
+    if (!isBridged) {
+      const { asset_balances } = await this.getBalances(options);
+      initialBalances = asset_balances; // Assign within the block
+      if (options.assets[0][1] > asset_balances[0].source) {
+        throw new Error(`Insufficient balance on Source chain for asset: ${options.assets[0][0]}`);
+      }
+    }
+
+    await this.executeTransfer(options);
+
+    if (!isBridged) {
+      await this.waitForBlocks();
+      await this.verifyExecution();
+      const { asset_balances: finalBalances } = await this.getBalances(options);
+      this.verifyFinalBalances(initialBalances, finalBalances, options);
+    }
   }
 
   // TODO: Wait for the next block to be produced.