Fix SelectionList.stories.tsx

Expensify · Mar 8, 2024 · 23de4fd · 23de4fd
1 parent 50d743b
commit 23de4fd
Showing 1 changed file with 12 additions and 14 deletions.
diff --git a/src/stories/SelectionList.stories.tsx b/src/stories/SelectionList.stories.tsx
@@ -3,11 +3,11 @@ import React, {useMemo, useState} from 'react';
 import Badge from '@components/Badge';
 import SelectionList from '@components/SelectionList';
 import RadioListItem from '@components/SelectionList/RadioListItem';
+import type {BaseSelectionListProps, ListItem} from '@components/SelectionList/types';
+import withNavigationFallback from '@components/withNavigationFallback';
 // eslint-disable-next-line no-restricted-imports
 import {defaultStyles} from '@styles/index';
 import CONST from '@src/CONST';
-import withNavigationFallback from '@components/withNavigationFallback';
-import type { BaseSelectionListProps, ListItem } from '@components/SelectionList/types';
 
 const SelectionListWithNavigation = withNavigationFallback(SelectionList);
 
@@ -110,18 +110,15 @@ function WithTextInput(props: BaseSelectionListProps<ListItem>) {
     const [selectedIndex, setSelectedIndex] = useState(1);
 
     const sections = props.sections.map((section) => {
-        const data = section.data.reduce<Array<ListItem & {isSelected: boolean}>>(
-            (memo, item, index) => {
-                if (!item.text.toLowerCase().includes(searchText.trim().toLowerCase())) {
-                    return memo;
-                }
-
-                const isSelected = selectedIndex === index + (section?.indexOffset ?? 0);
-                memo.push({...item, isSelected});
+        const data = section.data.reduce<Array<ListItem & {isSelected: boolean}>>((memo, item, index) => {
+            if (!item.text.toLowerCase().includes(searchText.trim().toLowerCase())) {
                 return memo;
-            },
-            [],
-        );
+            }
+
+            const isSelected = selectedIndex === index + (section?.indexOffset ?? 0);
+            memo.push({...item, isSelected});
+            return memo;
+        }, []);
 
         return {...section, data};
     });
@@ -207,6 +204,7 @@ function WithAlternateText(props: BaseSelectionListProps<ListItem>) {
             {...props}
             sections={sections}
             onSelectRow={onSelectRow}
+            ListItem={RadioListItem}
         />
     );
 }
@@ -412,4 +410,4 @@ WithConfirmButton.args = {
 };
 
 export {Default, WithTextInput, WithHeaderMessage, WithAlternateText, MultipleSelection, WithSectionHeader, WithConfirmButton};
-export default story;
+export default story;