From 5cc967bf54831eee3eb42a1cb8ed89a6cfbe581d Mon Sep 17 00:00:00 2001
From: Yogev Ben David <yogev132@gmail.com>
Date: Wed, 24 Nov 2021 15:26:08 +0200
Subject: [PATCH] [Navigation Mock] Fix bottomTabPressed event (#7368)

---
 e2e/BottomTabs.test.js                          |  7 +++++++
 lib/src/Mock/Components/ComponentScreen.tsx     | 13 ++++++++++++-
 lib/src/Mock/Stores/EventsStore.ts              |  8 ++++++++
 lib/src/Mock/mocks/NativeEventsReceiver.ts      |  7 +++++--
 playground/src/screens/FirstBottomTabScreen.tsx |  9 +++++++++
 playground/src/screens/LayoutsScreen.tsx        | 15 ++++++++++++++-
 6 files changed, 55 insertions(+), 4 deletions(-)

diff --git a/e2e/BottomTabs.test.js b/e2e/BottomTabs.test.js
index e9bc7412c90..46c1d87789b 100644
--- a/e2e/BottomTabs.test.js
+++ b/e2e/BottomTabs.test.js
@@ -106,4 +106,11 @@ describe('BottomTabs', () => {
     await elementById(TestIDs.POP_BTN).tap();
     await expect(elementById(TestIDs.BOTTOM_TABS)).toBeVisible();
   });
+
+  it('invoke bottomTabPressed event', async () => {
+    await elementById(TestIDs.THIRD_TAB_BAR_BTN).tap();
+    await expect(elementByLabel('BottomTabPressed')).toBeVisible();
+    await elementByLabel('OK').tap();
+    await expect(elementByLabel('First Tab')).toBeVisible();
+  });
 });
diff --git a/lib/src/Mock/Components/ComponentScreen.tsx b/lib/src/Mock/Components/ComponentScreen.tsx
index 6c5eb8e8013..62e21bff0fd 100644
--- a/lib/src/Mock/Components/ComponentScreen.tsx
+++ b/lib/src/Mock/Components/ComponentScreen.tsx
@@ -6,6 +6,8 @@ import { VISIBLE_SCREEN_TEST_ID } from '../constants';
 import { LayoutStore } from '../Stores/LayoutStore';
 import { connect } from '../connect';
 import { TopBar } from './TopBar';
+import { events } from '../Stores/EventsStore';
+import _ from 'lodash';
 
 export const ComponentScreen = connect(
   class extends Component<ComponentProps> {
@@ -34,7 +36,13 @@ export const ComponentScreen = connect(
             <Button
               testID={bottomTabOptions?.testID}
               title={bottomTabOptions?.text || ''}
-              onPress={() => LayoutStore.selectTabIndex(this.props.layoutNode.getBottomTabs(), i)}
+              onPress={() => {
+                events.invokeBottomTabPressed({
+                  tabIndex: i,
+                });
+                if (_.defaultTo(bottomTabOptions?.selectTabOnPress, true))
+                  LayoutStore.selectTabIndex(this.props.layoutNode.getBottomTabs(), i);
+              }}
             />
             <Text>{bottomTabOptions?.badge}</Text>
           </View>
@@ -46,6 +54,9 @@ export const ComponentScreen = connect(
 
     render() {
       const Component = Navigation.mock.store.getWrappedComponent(this.props.layoutNode.data.name);
+      if (!Component)
+        throw new Error(`${this.props.layoutNode.data.name} has not been registered.`);
+
       return (
         <View testID={this.isVisible() ? VISIBLE_SCREEN_TEST_ID : undefined}>
           {this.props.layoutNode.getStack() && (
diff --git a/lib/src/Mock/Stores/EventsStore.ts b/lib/src/Mock/Stores/EventsStore.ts
index 8f27f4bcd85..142e01bac63 100644
--- a/lib/src/Mock/Stores/EventsStore.ts
+++ b/lib/src/Mock/Stores/EventsStore.ts
@@ -4,6 +4,7 @@ import {
   ModalDismissedEvent,
 } from '../../interfaces/ComponentEvents';
 import { ComponentDidAppearEvent, NavigationButtonPressedEvent } from '../../index';
+import { BottomTabPressedEvent } from '../../interfaces/Events';
 
 export const events = {
   navigationButtonPressed: [(_event: NavigationButtonPressedEvent) => {}],
@@ -11,6 +12,7 @@ export const events = {
   componentDidAppear: [(_event: ComponentDidAppearEvent) => {}],
   componentDidDisappear: [(_event: ComponentDidDisappearEvent) => {}],
   modalDismissed: [(_event: ModalDismissedEvent) => {}],
+  bottomTabPressed: [(_event: BottomTabPressedEvent) => {}],
   invokeComponentWillAppear: (event: ComponentWillAppearEvent) => {
     events.componentWillAppear &&
       events.componentWillAppear.forEach((listener) => {
@@ -41,4 +43,10 @@ export const events = {
         listener(event);
       });
   },
+  invokeBottomTabPressed: (event: BottomTabPressedEvent) => {
+    events.bottomTabPressed &&
+      events.bottomTabPressed?.forEach((listener) => {
+        listener(event);
+      });
+  },
 };
diff --git a/lib/src/Mock/mocks/NativeEventsReceiver.ts b/lib/src/Mock/mocks/NativeEventsReceiver.ts
index b6c6adbbde3..77c609c6325 100644
--- a/lib/src/Mock/mocks/NativeEventsReceiver.ts
+++ b/lib/src/Mock/mocks/NativeEventsReceiver.ts
@@ -73,10 +73,13 @@ export class NativeEventsReceiver {
   }
 
   public registerBottomTabPressedListener(
-    _callback: (data: BottomTabPressedEvent) => void
+    callback: (data: BottomTabPressedEvent) => void
   ): EmitterSubscription {
+    events.bottomTabPressed.push(callback);
     return {
-      remove: () => {},
+      remove: () => {
+        _.remove(events.bottomTabPressed, (value) => value === callback);
+      },
     } as EmitterSubscription;
   }
 
diff --git a/playground/src/screens/FirstBottomTabScreen.tsx b/playground/src/screens/FirstBottomTabScreen.tsx
index 757f381b75c..6af83a34b5d 100644
--- a/playground/src/screens/FirstBottomTabScreen.tsx
+++ b/playground/src/screens/FirstBottomTabScreen.tsx
@@ -38,6 +38,11 @@ export default class FirstBottomTabScreen extends React.Component<NavigationComp
 
   dotVisible = true;
   badgeVisible = true;
+  bottomTabPressedListener = Navigation.events().registerBottomTabPressedListener((event) => {
+    if (event.tabIndex == 2) {
+      alert('BottomTabPressed');
+    }
+  });
 
   render() {
     return (
@@ -68,6 +73,10 @@ export default class FirstBottomTabScreen extends React.Component<NavigationComp
     );
   }
 
+  componentWillUnmount() {
+    this.bottomTabPressedListener.remove();
+  }
+
   modifyBottomTabs = () => {
     Navigation.mergeOptions(this.props.componentId, {
       bottomTabs: {
diff --git a/playground/src/screens/LayoutsScreen.tsx b/playground/src/screens/LayoutsScreen.tsx
index 8b3ff54912b..1b4430c5cb4 100644
--- a/playground/src/screens/LayoutsScreen.tsx
+++ b/playground/src/screens/LayoutsScreen.tsx
@@ -75,7 +75,7 @@ export default class LayoutsScreen extends NavigationComponent<NavigationCompone
 
   stack = () => Navigation.showModal(stack(Screens.Stack, 'StackId'));
 
-  bottomTabs = () =>
+  bottomTabs = () => {
     Navigation.showModal({
       bottomTabs: {
         children: [
@@ -88,6 +88,18 @@ export default class LayoutsScreen extends NavigationComponent<NavigationCompone
             },
             'SecondTab'
           ),
+          {
+            component: {
+              name: Screens.Pushed,
+              options: {
+                bottomTab: {
+                  selectTabOnPress: false,
+                  text: 'Tab 3',
+                  testID: testIDs.THIRD_TAB_BAR_BTN,
+                },
+              },
+            },
+          },
         ],
         options: {
           bottomTabs: {
@@ -96,6 +108,7 @@ export default class LayoutsScreen extends NavigationComponent<NavigationCompone
         },
       },
     });
+  };
 
   sideMenu = () =>
     Navigation.showModal({