From b5ac81d123633d60ea1d2befa8ccba501dfe3b7c Mon Sep 17 00:00:00 2001
From: storywithoutend <storywithoutend@gmail.com>
Date: Fri, 8 Nov 2024 00:13:54 +0800
Subject: [PATCH] fix faling tests

---
 e2e/specs/stateless/extendNames.spec.ts       | 10 ++++----
 .../ExtendNames/ExtendNames-flow.test.tsx     | 23 ++++++++++++++-----
 2 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/e2e/specs/stateless/extendNames.spec.ts b/e2e/specs/stateless/extendNames.spec.ts
index 44ae3943c..b13b390ee 100644
--- a/e2e/specs/stateless/extendNames.spec.ts
+++ b/e2e/specs/stateless/extendNames.spec.ts
@@ -11,7 +11,7 @@ import { daysToSeconds } from '@app/utils/time'
 
 import { test } from '../../../playwright'
 
-test('should be able to extend multiple names in grace period on the address page', async ({
+test('should be able to extend multiple names (including names in grace preiod) on the address page', async ({
   page,
   accounts,
   login,
@@ -45,7 +45,6 @@ test('should be able to extend multiple names in grace period on the address pag
 
   await addresPage.selectToggle.click()
 
-  // await page.pause()
   await expect(await page.locator('.name-detail-item').count()).toBeGreaterThan(0)
   const nameItems = await page.locator('.name-detail-item').all()
   const nameItemTestIds = await Promise.all(
@@ -83,11 +82,10 @@ test('should be able to extend multiple names in grace period on the address pag
   await expect(page.getByTestId('plus-minus-control-label')).toHaveText('2 years')
   await page.getByTestId('plus-minus-control-plus').click()
   await expect(page.getByTestId('plus-minus-control-label')).toHaveText('3 years')
-  await expect(page.getByTestId('invoice-item-0-amount')).not.toBeEmpty()
-  await expect(page.getByTestId('invoice-item-1-amount')).not.toBeEmpty()
+  await expect(page.getByTestId('invoice-item-0-amount')).not.toHaveText('0.0000 ETH')
+  await expect(page.getByTestId('invoice-item-1-amount')).not.toHaveText('0.0000 ETH')
   await expect(page.getByTestId('invoice-total')).not.toHaveText('0.0000 ETH')
 
-  // increment and save
   await page.getByTestId('extend-names-confirm').click()
   await expect(transactionModal.transactionModal).toBeVisible({ timeout: 10000 })
   await transactionModal.autoComplete()
@@ -375,7 +373,7 @@ test('should be able to extend a name by a month', async ({
   await test.step('should show the correct price data', async () => {
     await expect(extendNamesModal.getInvoiceExtensionFee).toContainText('0.0003')
     await expect(extendNamesModal.getInvoiceTransactionFee).toContainText('0.0001')
-    await expect(extendNamesModal.getInvoiceTotal).toContainText('0.0004')
+    await expect(extendNamesModal.getInvoiceTotal).toContainText(/0\.000[3|4]/)
     await expect(page.getByText(/1 month .* extension/)).toBeVisible()
   })
 
diff --git a/src/transaction-flow/input/ExtendNames/ExtendNames-flow.test.tsx b/src/transaction-flow/input/ExtendNames/ExtendNames-flow.test.tsx
index 986d1aa23..2ed1ecea3 100644
--- a/src/transaction-flow/input/ExtendNames/ExtendNames-flow.test.tsx
+++ b/src/transaction-flow/input/ExtendNames/ExtendNames-flow.test.tsx
@@ -1,18 +1,28 @@
 import { mockFunction, render, screen } from '@app/test-utils'
 
 import { describe, expect, it, vi } from 'vitest'
+import { useAccount, useBalance } from 'wagmi'
 
 import { useEstimateGasWithStateOverride } from '@app/hooks/chain/useEstimateGasWithStateOverride'
+import { useExpiry } from '@app/hooks/ensjs/public/useExpiry'
 import { usePrice } from '@app/hooks/ensjs/public/usePrice'
+import { useEthPrice } from '@app/hooks/useEthPrice'
 
 import { makeMockIntersectionObserver } from '../../../../test/mock/makeMockIntersectionObserver'
 import ExtendNames from './ExtendNames-flow'
 
 vi.mock('@app/hooks/chain/useEstimateGasWithStateOverride')
 vi.mock('@app/hooks/ensjs/public/usePrice')
+vi.mock('wagmi')
+vi.mock('@app/hooks/ensjs/public/useExpiry')
+vi.mock('@app/hooks/useEthPrice')
 
 const mockUseEstimateGasWithStateOverride = mockFunction(useEstimateGasWithStateOverride)
 const mockUsePrice = mockFunction(usePrice)
+const mockUseAccount = mockFunction(useAccount)
+const mockUseBalance = mockFunction(useBalance)
+const mockUseEthPrice = mockFunction(useEthPrice)
+const mockUseExpiry = mockFunction(useExpiry)
 
 vi.mock('@ensdomains/thorin', async () => {
   const originalModule = await vi.importActual('@ensdomains/thorin')
@@ -45,6 +55,10 @@ describe('Extendnames', () => {
     },
     isLoading: false,
   })
+  mockUseAccount.mockReturnValue({ address: '0x1234', isConnected: true })
+  mockUseBalance.mockReturnValue({ data: { balance: 100n }, isLoading: false })
+  mockUseEthPrice.mockReturnValue({ data: 100n, isLoading: false })
+  mockUseExpiry.mockReturnValue({ data: { expiry: { date: new Date() } }, isLoading: false })
   it('should render', async () => {
     render(
       <ExtendNames
@@ -72,17 +86,14 @@ describe('Extendnames', () => {
     const { parentElement } = optionBar
     expect(parentElement).toHaveStyle('opacity: 0.5')
   })
-  it('should disabled next button if gas limit estimation is still loading', () => {
-    mockUseEstimateGasWithStateOverride.mockReturnValueOnce({
-      data: { gasEstimate: 21000n, gasCost: 100n },
-      gasPrice: 100n,
-      error: null,
+  it('should disabled next button if the price data is loading ', () => {
+    mockUsePrice.mockReturnValueOnce({
       isLoading: true,
     })
     render(
       <ExtendNames
         {...{
-          data: { names: ['nick.eth'], isSelf: false },
+          data: { names: ['nick.eth'], isSelf: true },
           dispatch: () => null,
           onDismiss: () => null,
         }}