add test coverage

coinbase · Aug 9, 2024 · f64c051 · f64c051
1 parent 077df47
commit f64c051
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/src/transaction/hooks/useGetTransactionStatus.test.tsx b/src/transaction/hooks/useGetTransactionStatus.test.tsx
@@ -45,6 +45,17 @@ describe('useGetTransactionStatus', () => {
     expect(result.current.actionElement).not.toBeNull();
   });
 
+  it('should return correct status and actionElement when transaction id exists', () => {
+    (useTransactionContext as vi.Mock).mockReturnValue({
+      transactionId: 'ab123',
+    });
+
+    const { result } = renderHook(() => useGetTransactionStatus());
+
+    expect(result.current.label).toBe('Transaction in progress...');
+    expect(result.current.actionElement).not.toBeNull();
+  });
+
   it('should return correct status and actionElement when receipt exists', () => {
     (useTransactionContext as vi.Mock).mockReturnValue({
       receipt: 'receipt',

diff --git a/src/transaction/hooks/useGetTransactionToast.test.tsx b/src/transaction/hooks/useGetTransactionToast.test.tsx
@@ -46,6 +46,28 @@ describe('useGetTransactionToast', () => {
     expect(result.current.actionElement).not.toBeNull();
   });
 
+  it('should return correct status and actionElement when transaction id exists', () => {
+    (useTransactionContext as vi.Mock).mockReturnValue({
+      transactionId: 'ab123',
+    });
+
+    const { result } = renderHook(() => useGetTransactionToast());
+
+    expect(result.current.label).toBe('Transaction in progress');
+    expect(result.current.actionElement).not.toBeNull();
+  });
+
+  it('should return correct status and actionElement when transaction hash exists', () => {
+    (useTransactionContext as vi.Mock).mockReturnValue({
+      transactionHash: 'ab123',
+    });
+
+    const { result } = renderHook(() => useGetTransactionToast());
+
+    expect(result.current.label).toBe('Transaction in progress');
+    expect(result.current.actionElement).not.toBeNull();
+  });
+
   it('should return correct toast and actionElement when error occurs', () => {
     (useTransactionContext as vi.Mock).mockReturnValue({
       errorMessage: 'error',