Add E2E tests to PSM pallet (Westend AH)

KNOWN_GOOD_BLOCK_NUMBERS_KUSAMA.env

@@ -1,11 +1,11 @@
-ASSETHUBKUSAMA_BLOCK_NUMBER=15574522
-BASILISK_BLOCK_NUMBER=13861581
-BIFROSTKUSAMA_BLOCK_NUMBER=13405501
-BRIDGEHUBKUSAMA_BLOCK_NUMBER=7994810
-CORETIMEKUSAMA_BLOCK_NUMBER=4930183
-ENCOINTERKUSAMA_BLOCK_NUMBER=13264547
-KARURA_BLOCK_NUMBER=11360932
-KUSAMA_BLOCK_NUMBER=33088991
-MOONRIVER_BLOCK_NUMBER=15883530
-PEOPLEKUSAMA_BLOCK_NUMBER=8639534
-SHIDEN_BLOCK_NUMBER=14431788
+ASSETHUBKUSAMA_BLOCK_NUMBER=15682262
+BASILISK_BLOCK_NUMBER=13885582
+BIFROSTKUSAMA_BLOCK_NUMBER=13444005
+BRIDGEHUBKUSAMA_BLOCK_NUMBER=8014112
+CORETIMEKUSAMA_BLOCK_NUMBER=4949299
+ENCOINTERKUSAMA_BLOCK_NUMBER=13297944
+KARURA_BLOCK_NUMBER=11380250
+KUSAMA_BLOCK_NUMBER=33127751
+MOONRIVER_BLOCK_NUMBER=15919814
+PEOPLEKUSAMA_BLOCK_NUMBER=8678053
+SHIDEN_BLOCK_NUMBER=14463409

KNOWN_GOOD_BLOCK_NUMBERS_POLKADOT.env

@@ -1,11 +1,11 @@
-ACALA_BLOCK_NUMBER=10902025
-ASSETHUBPOLKADOT_BLOCK_NUMBER=14533018
-ASTAR_BLOCK_NUMBER=12990763
-BIFROSTPOLKADOT_BLOCK_NUMBER=11857098
-BRIDGEHUBPOLKADOT_BLOCK_NUMBER=7440339
-COLLECTIVESPOLKADOT_BLOCK_NUMBER=8646902
-CORETIMEPOLKADOT_BLOCK_NUMBER=4031499
-HYDRATION_BLOCK_NUMBER=12083573
-MOONBEAM_BLOCK_NUMBER=15213256
-PEOPLEPOLKADOT_BLOCK_NUMBER=4364097
-POLKADOT_BLOCK_NUMBER=30792329
+ACALA_BLOCK_NUMBER=10921639
+ASSETHUBPOLKADOT_BLOCK_NUMBER=14635494
+ASTAR_BLOCK_NUMBER=13022553
+BIFROSTPOLKADOT_BLOCK_NUMBER=11892624
+BRIDGEHUBPOLKADOT_BLOCK_NUMBER=7459608
+COLLECTIVESPOLKADOT_BLOCK_NUMBER=8666500
+CORETIMEPOLKADOT_BLOCK_NUMBER=4050736
+HYDRATION_BLOCK_NUMBER=12113679
+MOONBEAM_BLOCK_NUMBER=15250085
+PEOPLEPOLKADOT_BLOCK_NUMBER=4382214
+POLKADOT_BLOCK_NUMBER=30831662

KNOWN_GOOD_BLOCK_NUMBERS_WESTEND.env

@@ -0,0 +1 @@
+ASSETHUBWESTEND_BLOCK_NUMBER=14518276

packages/networks/src/chains/assethub.ts

@@ -9,6 +9,7 @@ const custom = {
     dot: { Concrete: { parents: 1, interior: 'Here' } },
     usdt: { Concrete: { parents: 0, interior: { X2: [{ PalletInstance: 50 }, { GeneralIndex: 1984 }] } } },
     usdtIndex: 1984,
+    usdcIndex: 1337,
     eth: {
       parents: 2,
       interior: {
@@ -43,28 +44,108 @@ const custom = {
       },
     },
   },
+  assetHubWestend: {
+    wnd: { Concrete: { parents: 1, interior: 'Here' } },
+    usdtIndex: 1984, // Test-Tether (USDTT), 6 decimals — existing PSM external asset on WAH
+    usdcIndex: 1337, // synthetic USDC injected via Chopsticks override
+    psmStableAssetId: 50000342, // pUSD stable asset ID as deployed on WAH
+  },
 }
 
-const getInitStorages = (config: typeof custom.assetHubPolkadot | typeof custom.assetHubKusama) => ({
-  System: {
-    account: [
-      [[defaultAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
-      [[defaultAccountsSr25519.alice.address], { providers: 1, data: { free: 1000e10 } }],
-      [[testAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
-    ],
-  },
-  Assets: {
-    account: [
-      [[config.usdtIndex, defaultAccounts.alice.address], { balance: 1000e6 }], // USDT
-    ],
-  },
-  ForeignAssets: {
-    account: [
-      [[config.eth, defaultAccounts.alice.address], { balance: 10n ** 18n }], // 1 ETH
-      [[config.eth, '13cKp89Msu7M2PiaCuuGr1BzAsD5V3vaVbDMs3YtjMZHdGwR'], { balance: 10n ** 20n }], // 100 ETH for Sibling 2000
-    ],
-  },
-})
+const getPsmInitStorages = (config: typeof custom.assetHubWestend) => {
+  return {
+    System: {
+      account: [
+        [[defaultAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
+        [[defaultAccountsSr25519.alice.address], { providers: 1, data: { free: 1000e10 } }],
+        [[testAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
+        [[testAccounts.bob.address], { providers: 1, data: { free: 1000e10 } }],
+      ],
+    },
+    Assets: {
+      // USDC (1337) is synthetic — does not exist on WAH. Full entry required.
+      // pUSD (50000342) and USDT (1984) already exist on WAH; only account balances needed.
+      asset: [
+        [
+          [config.usdcIndex],
+          {
+            owner: testAccounts.alice.address,
+            issuer: testAccounts.alice.address,
+            admin: testAccounts.alice.address,
+            freezer: testAccounts.alice.address,
+            supply: 10000e6,
+            deposit: 0,
+            minBalance: 1,
+            isSufficient: true,
+            accounts: 2,
+            sufficients: 2,
+            approvals: 0,
+            status: 'Live',
+          },
+        ],
+      ],
+      metadata: [[[config.usdcIndex], { deposit: 0, name: 'USD Coin', symbol: 'USDC', decimals: 6, isFrozen: false }]],
+      account: [
+        [[config.usdtIndex, testAccounts.alice.address], { balance: 1000e6 }], // USDT for Alice
+        [[config.usdtIndex, testAccounts.bob.address], { balance: 1000e6 }], // USDT for Bob
+        [[config.usdcIndex, testAccounts.alice.address], { balance: 1000e6 }], // USDC for Alice
+        [[config.usdcIndex, testAccounts.bob.address], { balance: 1000e6 }], // USDC for Bob
+        [[config.psmStableAssetId, testAccounts.alice.address], { balance: 1000e6 }], // pUSD for Alice
+      ],
+    },
+    Psm: {
+      // WAH has maxPsmDebtOfTotal at 10%; tests require a higher ceiling.
+      maxPsmDebtOfTotal: 500_000, // Permill: 50% of MaxIssuance
+      // USDC is not registered on WAH; inject alongside the existing USDT entry.
+      externalAssets: [
+        [[config.usdcIndex], { AllEnabled: null }], // USDC -> AllEnabled (synthetic)
+        [[config.usdtIndex], { AllEnabled: null }], // USDT -> AllEnabled (live on WAH)
+      ],
+      mintingFee: [
+        [[config.usdcIndex], 5_000], // Permill: 0.5% for USDC
+        // WAH USDT mintingFee is 0%; override to match test expectations.
+        [[config.usdtIndex], 5_000], // Permill: 0.5% for USDT
+      ],
+      redemptionFee: [
+        [[config.usdcIndex], 5_000], // Permill: 0.5% for USDC
+        // WAH USDT redemptionFee is 0.01%; override to match test expectations.
+        [[config.usdtIndex], 5_000], // Permill: 0.5% for USDT
+      ],
+      assetCeilingWeight: [
+        [[config.usdcIndex], 600_000], // Permill: 60% weight for USDC
+        // WAH USDT ceiling is 100% (single asset); override for two-asset split.
+        [[config.usdtIndex], 400_000], // Permill: 40% weight for USDT
+      ],
+      // WAH has live USDT debt (~90 UNIT); zero it so tests start from a clean state.
+      psmDebt: [[[config.usdtIndex], 0]],
+    },
+  }
+}
+
+const getInitStorages = (config: typeof custom.assetHubPolkadot | typeof custom.assetHubKusama) => {
+  const baseStorages = {
+    System: {
+      account: [
+        [[defaultAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
+        [[defaultAccountsSr25519.alice.address], { providers: 1, data: { free: 1000e10 } }],
+        [[testAccounts.alice.address], { providers: 1, data: { free: 1000e10 } }],
+      ],
+    },
+    Assets: {
+      account: [
+        [[config.usdtIndex, defaultAccounts.alice.address], { balance: 1000e6 }], // USDT
+      ],
+    },
+    ForeignAssets: {
+      account: [
+        [[config.eth, defaultAccounts.alice.address], { balance: 10n ** 18n }], // 1 ETH
+        [[config.eth, '13cKp89Msu7M2PiaCuuGr1BzAsD5V3vaVbDMs3YtjMZHdGwR'], { balance: 10n ** 20n }], // 100 ETH for Sibling 2000
+      ],
+    },
+  }
+
+  return baseStorages
+}
 
 export const assetHubPolkadot = defineChain({
   name: 'assetHubPolkadot',
@@ -97,3 +178,19 @@ export const assetHubKusama = defineChain({
     feeExtractor: standardFeeExtractor,
   },
 })
+
+export const assetHubWestend = defineChain({
+  name: 'assetHubWestend',
+  endpoint: endpoints.assetHubWestend,
+  paraId: 1000,
+  networkGroup: 'westend',
+  custom: custom.assetHubWestend,
+  initStorages: getPsmInitStorages(custom.assetHubWestend),
+  properties: {
+    addressEncoding: 42,
+    proxyBlockProvider: 'NonLocal',
+    schedulerBlockProvider: 'NonLocal',
+    asyncBacking: 'Enabled',
+    feeExtractor: standardFeeExtractor,
+  },
+})

packages/networks/src/pet-chain-endpoints.json

@@ -10,6 +10,10 @@
     "wss://rpc-polkadot.helixstreet.io",
     "wss://rpc-polkadot.luckyfriday.io"
   ],
+  "assetHubWestend": [
+    "wss://westend-asset-hub-rpc.polkadot.io",
+    "wss://asset-hub-westend-rpc.n.dwellir.com"
+  ],
   "assetHubPolkadot": [
     "wss://sys.ibp.network/asset-hub-polkadot",
     "wss://asset-hub-polkadot.dotters.network",

packages/networks/src/types.ts

@@ -46,7 +46,7 @@ type ChainConfigBase = {
   name: string
   endpoint: string | string[]
   isRelayChain?: boolean
-  networkGroup: 'polkadot' | 'kusama'
+  networkGroup: 'polkadot' | 'kusama' | 'westend'
 } & (ChainConfigRelaychain | ChainConfigParachain)
 
 export type ChainConfig<