added liquid assets sscript

package.json

@@ -10,6 +10,7 @@
     "start": "node ./lib/index.js",
     "start:zombienet-post-script": "yarn build:scripts && node ./scripts/build/testNetworkSetup.js",
     "start:zombienet-foreign-assets-script": "yarn build:scripts && node ./scripts/build/testNetworkForeignAssets.js",
+    "start:zombienet-liquidity-assets-script": "yarn build:scripts && node ./scripts/build/testNetworkLiquidAssets.js",
     "lint": "substrate-dev-run-lint",
     "lint:fix": "substrate-dev-run-lint --fix",
     "test": "NODE_ENV=test substrate-exec-jest --detectOpenHandles",

scripts/testNetworkLiquidAssets.ts

@@ -0,0 +1,136 @@
+// Copyright 2023 Parity Technologies (UK) Ltd.
+
+import { ApiPromise, WsProvider } from '@polkadot/api';
+import { Keyring } from '@polkadot/keyring';
+import { KeyringPair } from '@polkadot/keyring/types'
+import { cryptoWaitReady } from '@polkadot/util-crypto';
+import chalk from 'chalk';
+
+import { KUSAMA_ASSET_HUB_WS_URL } from './consts';
+import { awaitBlockProduction, delay, logWithDate } from './util';
+
+
+const ASSET_ID = 1;
+const ASSET_NAME = "Testy";
+const ASSET_TICKER = "TSTY";
+const ASSET_DECIMALS = 12;
+const ASSET_MIN = 1;
+
+const asset = {
+    parents: 0,
+    interior: {
+        X2: [
+            { palletInstance: 50 },
+            { generalIndex: ASSET_ID },
+        ]
+    }
+
+};
+
+const native = {
+    parents: 1,
+    interior: {
+        Here: '',
+    },
+};
+
+const createAssetCall = (api: ApiPromise, admin: KeyringPair) => {
+    return api.tx.assets.create(ASSET_ID, admin.address, ASSET_MIN)
+};
+
+const setMetadataCall = (api: ApiPromise) => {
+    return api.tx.assets.setMetadata(ASSET_ID, ASSET_NAME, ASSET_TICKER, ASSET_DECIMALS)
+};
+
+const mintCall = (api: ApiPromise, to: KeyringPair, amount: number) => {
+    return api.tx.assets.mint(ASSET_ID, to.address, amount)
+}
+
+const createLiquidityPoolCall = (api: ApiPromise) => {
+    // For now, we have to override the types of the Assets until PJS is updated 
+    return api.tx.assetConversion.createPool(<any>native, <any>asset)
+};
+
+const addLiquidityCall = (api: ApiPromise, amountNative: number, amountAsset: number, to: KeyringPair) => {
+    // For now, we have to override the types of the Assets until PJS is updated 
+    return api.tx.assetConversion.addLiquidity(<any>native, <any>asset, amountNative, amountAsset, 0, 0, to.address)
+};
+
+const transferLPTokensCall = (api: ApiPromise, token: number, amount: number, to: KeyringPair) => {
+    return api.tx.poolAssets.transferKeepAlive(token, to.address, amount)
+}
+
+const main = async () => {
+    logWithDate(chalk.yellow('Initializing script to create a liquidity pool on Kusama Asset Hub'));
+    await cryptoWaitReady();
+
+    const keyring = new Keyring({ type: 'sr25519' });
+    const alice = keyring.addFromUri('//Alice');
+    const bob = keyring.addFromUri('//Bob');
+
+    const api = await ApiPromise.create({
+        provider: new WsProvider(KUSAMA_ASSET_HUB_WS_URL),
+        noInitWarn: true,
+    });
+
+    await api.isReady;
+
+    logWithDate(chalk.green('Created a connection to Kusama AssetHub'));
+
+    const txs = [];
+    const create = createAssetCall(api, alice);
+    const setMetadata = setMetadataCall(api);
+    const mint = mintCall(api, alice, 10000000000000);
+    const createPool = createLiquidityPoolCall(api);
+    const addLiquidity = addLiquidityCall(api, 10000000000000, 10000000000000, alice);
+
+    txs.push(create);
+    txs.push(setMetadata);
+    txs.push(mint);
+    txs.push(createPool);
+    txs.push(addLiquidity);
+
+
+    await api.tx.utility.batch(txs).signAndSend(alice);
+
+    await delay(24000);
+
+    const nextLpToken = await api.query.assetConversion.nextPoolAssetId();
+
+    const lpToken = Number(nextLpToken) - 1;
+
+    logWithDate(chalk.yellow('Asset and Liquidity Pool created'));
+
+    logWithDate(chalk.green(`Liquidity Pool Token ID: ${lpToken}`));
+
+    const startingBalances = await api.query.poolAssets.account.entries(lpToken);
+
+    startingBalances.slice(1).forEach(([{ args: [lpToken, address] }, value]) => {
+        logWithDate(chalk.blue(`LP Token: ${lpToken}, Account: ${address}, Starting liquidty token balance: ${value.unwrap().balance.toHuman()}`));
+    });
+
+    await delay(24000);
+
+    logWithDate(chalk.magenta('Sending 1,000,000,000,000 lpTokens from Alice to Bob on Kusama Asset Hub'));
+
+    await transferLPTokensCall(api, 0, 1000000000000, bob).signAndSend(alice);
+
+    await delay(24000);
+
+    const newBalances = await api.query.poolAssets.account.entries(lpToken);
+    newBalances.slice(1).forEach(([{ args: [lpToken, address] }, value]) => {
+        logWithDate(chalk.blue(`LP Token: ${lpToken}, Account: ${address}, New liquidty token balance: ${value.unwrap().balance.toHuman()}`));
+    });
+
+    await api.disconnect().then(() => {
+        logWithDate(chalk.yellow('Polkadot-js successfully disconnected from asset-hub'));
+    });
+
+};
+
+// eslint-disable-next-line @typescript-eslint/no-floating-promises
+awaitBlockProduction(KUSAMA_ASSET_HUB_WS_URL).then(async () => {
+    await main()
+        .catch(console.error)
+        .finally(() => process.exit());
+});