Add tests

Signed-off-by: dan437 <80175477+dan437@users.noreply.github.com>

Author: dan437

ui/pages/confirmations/components/transactions/nested-transaction-tag/nested-transaction-tag.test.tsx

@@ -88,4 +88,85 @@ describe('NestedTransactionTag', () => {
     });
     expect(container).toBeEmptyDOMElement();
   });
+
+  it('handles transition from batch to non-batch transactions without error', () => {
+    useNestedTransactionLabelsMock.mockReturnValue([
+      FUNCTION_NAME_MOCK,
+      FUNCTION_NAME_MOCK,
+    ]);
+
+    // First render with batch transactions
+    const { getByText, rerender, container } = render({
+      nestedTransactions: [
+        BATCH_TRANSACTION_PARAMS_MOCK,
+        BATCH_TRANSACTION_PARAMS_MOCK,
+      ],
+    });
+    expect(getByText('Includes 2 transactions')).toBeInTheDocument();
+
+    // Re-render with no nested transactions (simulating switching to different confirmation)
+    // This should not throw React hooks error
+    const storeWithNoNested = configureStore(
+      getMockConfirmStateForTransaction(
+        genUnapprovedContractInteractionConfirmation({
+          address: ADDRESS_MOCK,
+          nestedTransactions: undefined,
+        }),
+      ),
+    );
+
+    expect(() => {
+      rerender(
+        renderWithConfirmContextProvider(
+          <NestedTransactionTag />,
+          storeWithNoNested,
+        ).container.firstChild as React.ReactElement,
+      );
+    }).not.toThrow();
+  });
+
+  it('handles transition between confirmations with different nested transaction counts without error', () => {
+    // Start with 2 nested transactions
+    useNestedTransactionLabelsMock.mockReturnValue([
+      FUNCTION_NAME_MOCK,
+      FUNCTION_NAME_MOCK,
+    ]);
+
+    const { getByText, container } = render({
+      nestedTransactions: [
+        BATCH_TRANSACTION_PARAMS_MOCK,
+        BATCH_TRANSACTION_PARAMS_MOCK,
+      ],
+    });
+    expect(getByText('Includes 2 transactions')).toBeInTheDocument();
+
+    // Update mock to return 3 labels for next render
+    useNestedTransactionLabelsMock.mockReturnValue([
+      FUNCTION_NAME_MOCK,
+      FUNCTION_NAME_MOCK,
+      FUNCTION_NAME_MOCK,
+    ]);
+
+    // Render with 3 nested transactions - should not throw due to different hook call count
+    const storeWith3Nested = configureStore(
+      getMockConfirmStateForTransaction(
+        genUnapprovedContractInteractionConfirmation({
+          address: ADDRESS_MOCK,
+          nestedTransactions: [
+            BATCH_TRANSACTION_PARAMS_MOCK,
+            BATCH_TRANSACTION_PARAMS_MOCK,
+            BATCH_TRANSACTION_PARAMS_MOCK,
+          ],
+        }),
+      ),
+    );
+
+    // This should not throw - the wrapper/inner pattern ensures clean remount
+    expect(() => {
+      renderWithConfirmContextProvider(
+        <NestedTransactionTag />,
+        storeWith3Nested,
+      );
+    }).not.toThrow();
+  });
 });