update token selector test

coinbase · Jun 11, 2024 · 37cafa2 · 37cafa2
1 parent 79f7aef
commit 37cafa2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/token/components/TokenSelectorDropdown.test.tsx b/src/token/components/TokenSelectorDropdown.test.tsx
@@ -30,6 +30,10 @@ describe('TokenSelectorDropdown', () => {
     },
   ];
 
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
   it('renders the TokenSelectorDropdown component', () => {
     render(<TokenSelectorDropdown setToken={setToken} options={options} onToggle={onToggle} />);
 
@@ -57,4 +61,20 @@ describe('TokenSelectorDropdown', () => {
       expect(onToggle).toHaveBeenCalled();
     });
   });
+
+  it('does not call onToggle when onToggle is not provided', async () => {
+    render(<TokenSelectorDropdown setToken={setToken} options={options} />);
+
+    await waitFor(() => {
+      fireEvent.click(screen.getByText(options[0].name));
+
+      expect(setToken).toHaveBeenCalledWith(options[0]);
+      expect(onToggle).not.toHaveBeenCalled();
+    });
+
+    await waitFor(() => {
+      fireEvent.click(document);
+      expect(onToggle).not.toHaveBeenCalled();
+    });
+  });
 });
diff --git a/src/token/types.ts b/src/token/types.ts
@@ -90,7 +90,7 @@ export type TokenSearchReact = {
  * Note: exported as public Type
  */
 export type TokenSelectorReact = {
-  children: ReactElement<{ onToggle: () => void }>;
+  children: ReactElement<{ onToggle?: () => void }>;
   setToken: (token: Token) => void;
   token?: Token;
 };