Skip to content

Commit

Permalink
test: fix test files after removing zeroex api
Browse files Browse the repository at this point in the history
  • Loading branch information
janndriessen committed May 24, 2024
1 parent fe8056b commit f795f87
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 63 deletions.
16 changes: 1 addition & 15 deletions src/tests/eth2xfli/index.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,32 +4,18 @@ import {
TestFactory,
SignerAccount1,
wei,
ZeroExApiSwapQuote,
resetHardhat,
IndexZeroExSwapQuoteProvider,
} from '../utils'
import { swapQuote01, swapQuote02 } from './quotes'

const { eth, eth2xfli } = QuoteTokens
const swapQuoteProvider = IndexZeroExSwapQuoteProvider
const zeroExApi = ZeroExApiSwapQuote
const zeroExMock = jest.spyOn(zeroExApi, 'getSwapQuote')
zeroExMock
.mockImplementationOnce(async () => {
return swapQuote01
})
.mockImplementationOnce(async () => {
return swapQuote02
})

describe('ETH2xFLI (mainnet)', () => {
let factory: TestFactory
beforeAll(async () => {
const blockNumber = 17826737
const provider = LocalhostProvider
const signer = SignerAccount1
await resetHardhat(provider, blockNumber)
factory = new TestFactory(provider, signer, swapQuoteProvider, zeroExApi)
factory = new TestFactory(provider, signer, swapQuoteProvider)
})

test('can mint ETH2xFLI', async () => {
Expand Down
32 changes: 4 additions & 28 deletions src/tests/icreth/index.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -8,27 +8,18 @@ import {
transferFromWhale,
wei,
wrapETH,
ZeroExApiSwapQuote,
} from '../utils'
import {
mintQuoteUsdc1,
mintQuoteUsdc2,
redeemQuoteUsdc1,
redeemQuoteUsdc2,
} from './quotes'

const { eth, icreth, reth, usdc, weth } = QuoteTokens

const signer = SignerAccount2
const swapQuoteProvider = IndexZeroExSwapQuoteProvider
const zeroExApi = ZeroExApiSwapQuote
const zeroExMock = jest.spyOn(zeroExApi, 'getSwapQuote')

describe('icRETH (mainnet) - ETH', () => {
let factory: TestFactory
beforeAll(async () => {
const provider = LocalhostProvider
factory = new TestFactory(provider, signer, swapQuoteProvider, zeroExApi)
factory = new TestFactory(provider, signer, swapQuoteProvider)
})

test('can mint icRETH', async () => {
Expand Down Expand Up @@ -58,7 +49,7 @@ describe.skip('icRETH (mainnet) - rETH', () => {
let factory: TestFactory
beforeAll(async () => {
const provider = LocalhostProvider
factory = new TestFactory(provider, signer, swapQuoteProvider, zeroExApi)
factory = new TestFactory(provider, signer, swapQuoteProvider)
})

test('can mint icRETH', async () => {
Expand Down Expand Up @@ -95,23 +86,8 @@ describe.skip('icRETH (mainnet) - rETH', () => {
describe.skip('icRETH (mainnet) - USDC', () => {
let factory: TestFactory
beforeAll(async () => {
zeroExMock
.mockImplementationOnce(async () => {
return mintQuoteUsdc1
})
.mockImplementationOnce(async () => {
return mintQuoteUsdc2
})
.mockImplementationOnce(async () => {
return redeemQuoteUsdc1
})
.mockImplementationOnce(async () => {
return redeemQuoteUsdc2
})
const blockNumber = 17940000
const provider = LocalhostProvider
await resetHardhat(provider, blockNumber)
factory = new TestFactory(provider, signer, swapQuoteProvider, zeroExApi)
factory = new TestFactory(provider, signer, swapQuoteProvider)
})

test('can mint icRETH', async () => {
Expand Down Expand Up @@ -149,7 +125,7 @@ describe('icRETH (mainnet) - WETH', () => {
let factory: TestFactory
beforeAll(async () => {
const provider = LocalhostProvider
factory = new TestFactory(provider, signer, swapQuoteProvider, zeroExApi)
factory = new TestFactory(provider, signer, swapQuoteProvider)
})

test('can mint icRETH', async () => {
Expand Down
10 changes: 2 additions & 8 deletions src/tests/utils/factories.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ import {
SwapQuoteProvider,
} from 'quote'

import { ZeroExApi } from 'utils'
import { approveErc20, balanceOf } from './'

class TxTestFactory {
Expand Down Expand Up @@ -87,14 +86,9 @@ export class TestFactory {
constructor(
provider: JsonRpcProvider,
signer: Wallet,
swapQuoteProvider: SwapQuoteProvider,
zeroExApi: ZeroExApi
swapQuoteProvider: SwapQuoteProvider
) {
this.quoteProvider = new FlashMintQuoteProvider(
provider,
swapQuoteProvider,
zeroExApi
)
this.quoteProvider = new FlashMintQuoteProvider(provider, swapQuoteProvider)
this.txFactory = new TxTestFactory(provider, signer)
}

Expand Down
14 changes: 2 additions & 12 deletions src/tests/utils/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -32,20 +32,10 @@ export const LocalhostProviderArbitrum = new JsonRpcProvider(

// Pre-configured TestFactories
export function getArbitrumTestFactory(provider: JsonRpcProvider, signer: any) {
return new TestFactory(
provider,
signer,
IndexZeroExSwapQuoteProviderArbitrum,
ZeroExApiArbitrumSwapQuote
)
return new TestFactory(provider, signer, IndexZeroExSwapQuoteProviderArbitrum)
}
export function getMainnetTestFactory(provider: JsonRpcProvider, signer: any) {
return new TestFactory(
provider,
signer,
IndexZeroExSwapQuoteProvider,
ZeroExApiSwapQuote
)
return new TestFactory(provider, signer, IndexZeroExSwapQuoteProvider)
}

export function getSignerAccount(num = 0, provider: JsonRpcProvider) {
Expand Down

0 comments on commit f795f87

Please sign in to comment.