SectionWheelPicker driver to take testIDs from children (#2949)

* Changed sections drivers finding

* removed main testID from the section testid in the driver

* removed extra import

* Changed to take testIDs from children

* Fixed testId and removed unused variable

* formattings

* change section type cheking to be mroe readable

* changed implementation

Author: nitzanyiz

src/components/sectionsWheelPicker/SectionsWheelPicker.driver.tsx

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