add more unit test

Uniswap · Jul 19, 2024 · 0ba3858 · 0ba3858
1 parent 264472a
commit 0ba3858
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/test/mocha/e2e/quote.test.ts b/test/mocha/e2e/quote.test.ts
@@ -1176,7 +1176,7 @@ describe('quote', function () {
                         // if fee-on-transfer flag is not enabled, most likely the simulation will fail due to quote not subtracting the tax
                         simulateFromAddress: enableFeeOnTransferFeeFetching ? simulateFromAddress : undefined,
                         portionBips: FLAT_PORTION.bips,
-                        portionRecipient: FLAT_PORTION.recipient
+                        portionRecipient: FLAT_PORTION.recipient,
                       }
 
                       const queryParams = qs.stringify(quoteReq)

diff --git a/test/mocha/integ/graphql/graphql-token-fee-fetcher.test.ts b/test/mocha/integ/graphql/graphql-token-fee-fetcher.test.ts
@@ -64,9 +64,15 @@ describe('integration test for GraphQLTokenFeeFetcher', () => {
     expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]).to.not.be.undefined
     expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]?.buyFeeBps).to.be.undefined
     expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]?.sellFeeBps).to.be.undefined
+    expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]?.feeTakenOnTransfer).to.not.be.undefined
+    expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]?.externalTransferFailed).to.not.be.undefined
+    expect(tokenFeeMap[WETH9[ChainId.MAINNET]!.address]?.sellReverted).to.not.be.undefined
     expect(tokenFeeMap[BITBOY.address]).to.not.be.undefined
     expect(tokenFeeMap[BITBOY.address]?.buyFeeBps?._hex).equals(BITBOY.buyFeeBps?._hex)
     expect(tokenFeeMap[BITBOY.address]?.sellFeeBps?._hex).equals(BITBOY.sellFeeBps?._hex)
+    expect(tokenFeeMap[BITBOY.address]?.feeTakenOnTransfer).equals(false)
+    expect(tokenFeeMap[BITBOY.address]?.externalTransferFailed).equals(true)
+    expect(tokenFeeMap[BITBOY.address]?.sellReverted).equals(false)
   })
 
   it('Fetch BULLET and BITBOY, should return BOTH', async () => {
@@ -77,9 +83,16 @@ describe('integration test for GraphQLTokenFeeFetcher', () => {
     expect(tokenFeeMap[BULLET.address]).to.not.be.undefined
     expect(tokenFeeMap[BULLET.address]?.buyFeeBps?._hex).equals(BULLET.buyFeeBps?._hex)
     expect(tokenFeeMap[BULLET.address]?.sellFeeBps?._hex).equals(BULLET.sellFeeBps?._hex)
+    expect(tokenFeeMap[BULLET.address]?.feeTakenOnTransfer).equals(false)
+    expect(tokenFeeMap[BULLET.address]?.externalTransferFailed).equals(false)
+    expect(tokenFeeMap[BULLET.address]?.sellReverted).equals(false)
+
     expect(tokenFeeMap[BITBOY.address]).to.not.be.undefined
     expect(tokenFeeMap[BITBOY.address]?.buyFeeBps?._hex).equals(BITBOY.buyFeeBps?._hex)
     expect(tokenFeeMap[BITBOY.address]?.sellFeeBps?._hex).equals(BITBOY.sellFeeBps?._hex)
+    expect(tokenFeeMap[BITBOY.address]?.feeTakenOnTransfer).equals(false)
+    expect(tokenFeeMap[BITBOY.address]?.externalTransferFailed).equals(true)
+    expect(tokenFeeMap[BITBOY.address]?.sellReverted).equals(false)
 
     expect(spyGraphQLFetcher.calledOnce).to.be.true
     expect(spyOnChainFetcher.calledOnce).to.be.false
@@ -97,6 +110,9 @@ describe('integration test for GraphQLTokenFeeFetcher', () => {
     expect(tokenFeeMap[BITBOY.address]).to.not.be.undefined
     expect(tokenFeeMap[BITBOY.address]?.buyFeeBps?._hex).equals(BITBOY.buyFeeBps?._hex)
     expect(tokenFeeMap[BITBOY.address]?.sellFeeBps?._hex).equals(BITBOY.sellFeeBps?._hex)
+    expect(tokenFeeMap[BITBOY.address]?.feeTakenOnTransfer).equals(false)
+    expect(tokenFeeMap[BITBOY.address]?.externalTransferFailed).equals(true)
+    expect(tokenFeeMap[BITBOY.address]?.sellReverted).equals(false)
 
     expect(spyGraphQLFetcher.calledOnce).to.be.true
     expect(spyOnChainFetcher.calledOnce).to.be.true