added handling for custom testid in the sections (#3193)

Author: nitzanyiz

src/components/sectionsWheelPicker/SectionsWheelPicker.driver.tsx

@@ -1,16 +1,17 @@
 import _ from 'lodash';
+import type {ReactTestInstance} from 'react-test-renderer';
 import {useComponentDriver, ComponentProps} from '../../testkit/new/Component.driver';
 import {WheelPickerDriver} from '../WheelPicker/WheelPicker.driver';
-import {SectionsWheelPickerProps} from './index';
+import type {SectionsWheelPickerProps} from './index';
 
 export const SectionsWheelPickerDriver = (props: ComponentProps) => {
   const driver = useComponentDriver(props);
-  const sections = driver.getElement().children as SectionsWheelPickerProps['sections'];
+  const sections = _.map(driver.getElement().children as ReactTestInstance[], (child) => child.props) as SectionsWheelPickerProps['sections'];
   const sectionsDrivers = _.map(sections, (section, index) => {
     const sectionTestID = `${props.testID}.${index}`;
     return WheelPickerDriver({
       renderTree: props.renderTree,
-      testID: section?.testID || sectionTestID
+      testID: section.testID || sectionTestID
     });
   });
   return {...driver, sections, sectionsDrivers};