@ledgerhq/coin-tester-solana 1.1.0 → 1.1.1-nightly.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ledgerhq/coin-tester-solana",
-  "version": "1.1.0",
+  "version": "1.1.1-nightly.1",
   "description": "Ledger Solana Coin Tester",
   "main": "src/scenarii.test.ts",
   "keywords": [
@@ -49,14 +49,14 @@
     "bignumber.js": "^9",
     "docker-compose": "^1",
     "msw": "^2",
-    "@ledgerhq/coin-framework": "^5.2.0",
+    "@ledgerhq/coin-framework": "^5.3.0-nightly.1",
     "@ledgerhq/coin-tester": "^0.6.0",
-    "@ledgerhq/coin-solana": "^0.25.0",
-    "@ledgerhq/cryptoassets": "^13.18.0",
-    "@ledgerhq/live-env": "^2.9.0",
-    "@ledgerhq/live-signer-solana": "^0.2.0",
+    "@ledgerhq/coin-solana": "^0.25.1-nightly.1",
+    "@ledgerhq/cryptoassets": "^13.18.1-nightly.0",
+    "@ledgerhq/live-signer-solana": "^0.2.1-nightly.1",
     "@ledgerhq/types-cryptoassets": "^7.23.0",
-    "@ledgerhq/types-live": "^6.72.0"
+    "@ledgerhq/types-live": "^6.73.0-nightly.1",
+    "@ledgerhq/live-env": "^2.9.1-nightly.0"
   },
   "devDependencies": {
     "@types/jest": "^29",

package/src/connection.ts

@@ -1,4 +1,11 @@
-import { Connection, Keypair, PublicKey } from "@solana/web3.js";
+import {
+  Authorized,
+  Connection,
+  Keypair,
+  PublicKey,
+  StakeProgram,
+  VoteAccountInfo,
+} from "@solana/web3.js";
 import { createAssociatedTokenAccountIdempotent, mintTo } from "@solana/spl-token";
 import { TokenCurrency } from "@ledgerhq/types-cryptoassets";
 import { SolanaTokenProgram } from "@ledgerhq/coin-solana/types";
@@ -16,6 +23,8 @@ export const PAYER = Keypair.fromSecretKey(
     207,
   ]),
 );
+export const STAKE_ACCOUNT = Keypair.generate();
+export let VOTE_ACCOUNT: VoteAccountInfo | null = null;
 
 export async function createSplAccount(
   address: string,
@@ -46,3 +55,25 @@ export async function createSplAccount(
   );
   await connection.confirmTransaction({ signature, ...latest });
 }
+
+export async function initVoteAccount() {
+  const voteAccounts = await connection.getVoteAccounts();
+  VOTE_ACCOUNT = voteAccounts.current[0];
+}
+
+export async function initStakeAccount(address: string, amount: number) {
+  const authority = new PublicKey(address);
+  const transaction = StakeProgram.createAccount({
+    fromPubkey: PAYER.publicKey,
+    stakePubkey: STAKE_ACCOUNT.publicKey,
+    authorized: new Authorized(authority, authority),
+    lamports: amount,
+  });
+  const latest = await connection.getLatestBlockhash();
+  transaction.feePayer = PAYER.publicKey;
+  transaction.recentBlockhash = latest.blockhash;
+  transaction.lastValidBlockHeight = latest.lastValidBlockHeight;
+  transaction.sign(PAYER, STAKE_ACCOUNT);
+  const signature = await connection.sendRawTransaction(transaction.serialize());
+  await connection.confirmTransaction({ signature, ...latest });
+}

package/src/fixtures.ts

@@ -13,6 +13,7 @@ export const RECIPIENT = "Hj69wRzkrFuf1Nby4yzPEFHdsmQdMoVYjvDKZSLjZFEp";
 export const SOLANA = getCryptoCurrencyById("solana");
 export const SOLANA_USDC = getTokenById("solana/spl/epjfwdd5aufqssqem2qn1xzybapc8g4weggkzwytdt1v");
 export const SOLANA_CWIF = getTokenById("solana/spl/7atgf8kqo4wjrd5atgx7t1v2zvvykpjbffnevf1icfv1");
+export const WITHDRAWABLE_AMOUNT = 2e9;
 
 export const makeAccount = (
   address: string,

package/src/scenarii/solana.ts

@@ -3,7 +3,15 @@ import { SolanaAccount, Transaction as SolanaTransaction } from "@ledgerhq/coin-
 import { killSpeculos, spawnSpeculos } from "@ledgerhq/coin-tester/signers/speculos";
 import resolver from "@ledgerhq/coin-solana/hw-getAddress";
 import { LegacySignerSolana } from "@ledgerhq/live-signer-solana";
-import { RECIPIENT, SOLANA, SOLANA_CWIF, SOLANA_USDC, initMSW, makeAccount } from "../fixtures";
+import {
+  RECIPIENT,
+  SOLANA,
+  SOLANA_CWIF,
+  SOLANA_USDC,
+  WITHDRAWABLE_AMOUNT,
+  initMSW,
+  makeAccount,
+} from "../fixtures";
 import { CoinConfig } from "@ledgerhq/coin-framework/config";
 import solanaCoinConfig, { SolanaCoinConfig } from "@ledgerhq/coin-solana/config";
 import BigNumber from "bignumber.js";
@@ -12,7 +20,14 @@ import { airdrop, killAgave, spawnAgave } from "../agave";
 import { encodeAccountIdWithTokenAccountAddress } from "@ledgerhq/coin-solana/logic";
 import { TOKEN_2022_PROGRAM_ID, getAssociatedTokenAddressSync } from "@solana/spl-token";
 import { PublicKey } from "@solana/web3.js";
-import { PAYER, createSplAccount } from "../connection";
+import {
+  PAYER,
+  STAKE_ACCOUNT,
+  VOTE_ACCOUNT,
+  createSplAccount,
+  initStakeAccount,
+  initVoteAccount,
+} from "../connection";
 import { Config, getChainAPI } from "@ledgerhq/coin-solana/network/index";
 import { makeBridges } from "@ledgerhq/coin-solana/bridge/bridge";
 
@@ -26,6 +41,14 @@ type SolanaScenarioTransaction = ScenarioTransaction<SolanaTransaction, SolanaAc
 export const defaultNanoApp = { firmware: "2.4.2", version: "1.9.1" } as const;
 
 function makeScenarioTransactions(address: string): SolanaScenarioTransaction[] {
+  if (!VOTE_ACCOUNT) {
+    throw new Error("Vote account not initialized");
+  }
+
+  if (!STAKE_ACCOUNT) {
+    throw new Error("Stake account not initialized");
+  }
+
   const scenarioSendSolTransaction: SolanaScenarioTransaction = {
     name: "Send 1 Sol",
     amount: new BigNumber(1e9),
@@ -166,6 +189,76 @@ function makeScenarioTransactions(address: string): SolanaScenarioTransaction[]
     },
   };
 
+  const scenarioCreateSolStakeAccountTransaction: SolanaScenarioTransaction = {
+    name: "Create Stake Account 1 Sol",
+    amount: new BigNumber(1e9),
+    model: {
+      kind: "stake.createAccount",
+      uiState: { delegate: { voteAccAddress: VOTE_ACCOUNT.votePubkey } },
+    },
+    expect: (previousAccount, currentAccount) => {
+      const [latestOperation] = currentAccount.operations;
+      expect(currentAccount.operations.length - previousAccount.operations.length).toEqual(1);
+      expect(latestOperation.type).toEqual("DELEGATE");
+      expect(latestOperation.value).toStrictEqual(latestOperation.fee);
+      expect(latestOperation.senders).toStrictEqual([]);
+      expect(latestOperation.recipients).toStrictEqual([]);
+      expect(latestOperation.extra).toStrictEqual({
+        stake: { address: VOTE_ACCOUNT?.votePubkey, amount: new BigNumber(1e9 + 2287880) }, // amount + rent exempt reserve + fee
+      });
+      expect(currentAccount.balance).toStrictEqual(
+        previousAccount.balance.minus(latestOperation.value),
+      );
+      expect(currentAccount.spendableBalance).toStrictEqual(
+        previousAccount.spendableBalance.minus(1e9 + 2297880),
+      );
+    },
+  };
+
+  const scenarioActivateStakeAccount: SolanaScenarioTransaction = {
+    name: "Activate Stake Account",
+    model: {
+      kind: "stake.delegate",
+      uiState: {
+        stakeAccAddr: STAKE_ACCOUNT.publicKey.toBase58(),
+        voteAccAddr: VOTE_ACCOUNT.votePubkey,
+      },
+    },
+    expect: (previousAccount, currentAccount) => {
+      const [latestOperation] = currentAccount.operations;
+      expect(currentAccount.operations.length - previousAccount.operations.length).toEqual(1);
+      expect(latestOperation.type).toEqual("DELEGATE");
+      expect(latestOperation.value).toStrictEqual(latestOperation.fee);
+      expect(latestOperation.senders).toStrictEqual([]);
+      expect(latestOperation.recipients).toStrictEqual([]);
+      expect(latestOperation.extra).toStrictEqual({
+        stake: { address: VOTE_ACCOUNT?.votePubkey, amount: latestOperation.value },
+      });
+      expect(currentAccount.balance).toStrictEqual(
+        previousAccount.balance.minus(latestOperation.value),
+      );
+    },
+  };
+
+  const scenarioDeactivateStakeAccount: SolanaScenarioTransaction = {
+    name: "Deactivate Stake Account",
+    model: {
+      kind: "stake.undelegate",
+      uiState: { stakeAccAddr: STAKE_ACCOUNT.publicKey.toBase58() },
+    },
+    expect: (previousAccount, currentAccount) => {
+      const [latestOperation] = currentAccount.operations;
+      expect(currentAccount.operations.length - previousAccount.operations.length).toEqual(1);
+      expect(latestOperation.type).toEqual("UNDELEGATE");
+      expect(latestOperation.value).toStrictEqual(latestOperation.fee);
+      expect(latestOperation.senders).toStrictEqual([]);
+      expect(latestOperation.recipients).toStrictEqual([]);
+      expect(currentAccount.balance).toStrictEqual(
+        previousAccount.balance.minus(latestOperation.value),
+      );
+    },
+  };
+
   const scenarioSendAllSolTransaction: SolanaScenarioTransaction = {
     name: "Send All Sol",
     useAllAmount: true,
@@ -182,13 +275,73 @@ function makeScenarioTransactions(address: string): SolanaScenarioTransaction[]
       expect(currentAccount.spendableBalance).toStrictEqual(new BigNumber(0));
     },
   };
+
+  const scenarioStakeWithdrawTransaction: SolanaScenarioTransaction = {
+    name: "Withdraw From Stake Account",
+    model: {
+      kind: "stake.withdraw",
+      uiState: {
+        stakeAccAddr: STAKE_ACCOUNT.publicKey.toBase58(),
+      },
+    },
+    expect: (previousAccount, currentAccount) => {
+      const [latestOperation] = currentAccount.operations;
+      expect(currentAccount.operations.length - previousAccount.operations.length).toEqual(1);
+      expect(latestOperation.type).toEqual("WITHDRAW_UNBONDED");
+      expect(latestOperation.value).toStrictEqual(latestOperation.fee);
+      expect(latestOperation.senders).toStrictEqual([]);
+      expect(latestOperation.recipients).toStrictEqual([]);
+      expect(latestOperation.extra).toStrictEqual({
+        stake: {
+          address: STAKE_ACCOUNT?.publicKey.toBase58(),
+          amount: new BigNumber(WITHDRAWABLE_AMOUNT),
+        },
+      });
+      expect(currentAccount.balance).toStrictEqual(
+        previousAccount.balance.minus(latestOperation.value),
+      );
+      expect(currentAccount.spendableBalance).toStrictEqual(
+        previousAccount.spendableBalance.plus(WITHDRAWABLE_AMOUNT),
+      );
+    },
+  };
+
+  const scenarioCreateAllSolStakeAccountTransaction: SolanaScenarioTransaction = {
+    name: "Create Stake Account All Sol",
+    useAllAmount: true,
+    model: {
+      kind: "stake.createAccount",
+      uiState: { delegate: { voteAccAddress: VOTE_ACCOUNT.votePubkey } },
+    },
+    expect: (previousAccount, currentAccount) => {
+      const [latestOperation] = currentAccount.operations;
+      expect(currentAccount.operations.length - previousAccount.operations.length).toEqual(1);
+      expect(latestOperation.type).toEqual("DELEGATE");
+      expect(latestOperation.value).toStrictEqual(latestOperation.fee);
+      expect(latestOperation.senders).toStrictEqual([]);
+      expect(latestOperation.recipients).toStrictEqual([]);
+      expect(latestOperation.extra).toMatchObject({
+        stake: { address: VOTE_ACCOUNT?.votePubkey },
+      });
+      expect(currentAccount.balance).toStrictEqual(
+        previousAccount.balance.minus(latestOperation.value),
+      );
+      expect(currentAccount.spendableBalance).toStrictEqual(new BigNumber(0));
+    },
+  };
+
   return [
     scenarioSendSolTransaction,
     scenarioSendUsdcTransaction,
     scenarioSendAllUsdcTransaction,
     scenarioSendCwifTransaction,
     scenarioSendAllCwifTransaction,
+    scenarioCreateSolStakeAccountTransaction,
+    scenarioActivateStakeAccount,
+    scenarioDeactivateStakeAccount,
     scenarioSendAllSolTransaction,
+    scenarioStakeWithdrawTransaction,
+    scenarioCreateAllSolStakeAccountTransaction,
   ];
 }
 
@@ -235,6 +388,8 @@ export const scenarioSolana: Scenario<SolanaTransaction, SolanaAccount> = {
     await airdrop(PAYER.publicKey.toBase58(), 5);
     await createSplAccount(account.freshAddress, SOLANA_USDC, 5, "spl-token");
     await createSplAccount(account.freshAddress, SOLANA_CWIF, 5, "spl-token-2022");
+    await initVoteAccount();
+    await initStakeAccount(account.freshAddress, WITHDRAWABLE_AMOUNT);
 
     initMSW();