|
1 | 1 | import { AnchorProvider, BN } from '@project-serum/anchor'; |
2 | | -import { TokenInfo, TokenListProvider } from '@solana/spl-token-registry'; |
3 | 2 | import { Cluster, Keypair, PublicKey } from '@solana/web3.js'; |
4 | 3 | import { DEFAULT_SLIPPAGE, DEVNET_COIN, DEVNET_POOL, MAINNET_POOL } from '../constants'; |
5 | 4 | import AmmImpl from '../index'; |
6 | 5 | import { calculateSwapQuote, getOnchainTime } from '../utils'; |
7 | | -import { DEVNET, MAINNET, airDropSol, getOrCreateATA, mockWallet } from './utils'; |
| 6 | +import { DEVNET, MAINNET, airDropSol, getOrCreateATA, getValidatedTokens, mockWallet } from './utils'; |
8 | 7 | import { NATIVE_MINT } from '@solana/spl-token'; |
| 8 | +import { TokenInfo } from '../types'; |
9 | 9 |
|
10 | 10 | describe('Interact with Devnet pool', () => { |
11 | 11 | const provider = new AnchorProvider(DEVNET.connection, mockWallet, { |
@@ -676,10 +676,9 @@ describe('Staging pool', () => { |
676 | 676 | const jitoSolDepegPool = new PublicKey('HcHN59j1xArjLuqfCMJ96yJ2CKatxHMFABEZWvcfPrYZ'); |
677 | 677 |
|
678 | 678 | beforeAll(async () => { |
679 | | - const tokenMap = await new TokenListProvider().resolve().then((tokens) => { |
680 | | - return tokens.filterByClusterSlug('mainnet-beta').getList(); |
681 | | - }); |
682 | | - const SOL = tokenMap.find((token) => token.address === 'So11111111111111111111111111111111111111112'); |
| 679 | + const tokenList = await getValidatedTokens(); |
| 680 | + |
| 681 | + const SOL = tokenList.find((token) => token.address === 'So11111111111111111111111111111111111111112'); |
683 | 682 | const jitoSOL: TokenInfo = { |
684 | 683 | chainId: SOL!.chainId, |
685 | 684 | address: 'J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn', |
@@ -724,13 +723,12 @@ describe('Interact with Mainnet pool', () => { |
724 | 723 | let depegPool: AmmImpl; |
725 | 724 |
|
726 | 725 | beforeAll(async () => { |
727 | | - const tokenListContainer = await new TokenListProvider().resolve(); |
728 | | - const tokenMap = tokenListContainer.filterByClusterSlug(MAINNET.cluster).getList(); |
| 726 | + const tokenList = await getValidatedTokens(); |
729 | 727 |
|
730 | | - const USDT = tokenMap.find((token) => token.address === 'Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB'); |
731 | | - const USDC = tokenMap.find((token) => token.address === 'EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v'); |
732 | | - const SOL = tokenMap.find((token) => token.address === 'So11111111111111111111111111111111111111112'); |
733 | | - const STSOL = tokenMap.find((token) => token.address === '7dHbWXmci3dT8UFYWYZweBLXgycu7Y3iL6trKn1Y7ARj'); |
| 728 | + const USDT = tokenList.find((token) => token.address === 'Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB'); |
| 729 | + const USDC = tokenList.find((token) => token.address === 'EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v'); |
| 730 | + const SOL = tokenList.find((token) => token.address === 'So11111111111111111111111111111111111111112'); |
| 731 | + const STSOL = tokenList.find((token) => token.address === '7dHbWXmci3dT8UFYWYZweBLXgycu7Y3iL6trKn1Y7ARj'); |
734 | 732 |
|
735 | 733 | const pools = [ |
736 | 734 | { pool: MAINNET_POOL.USDC_SOL, tokenInfoA: USDC!, tokenInfoB: SOL! }, |
|
0 commit comments