Added tests

app/components/drafts_buttton/drafts_button.test.tsx

@@ -0,0 +1,121 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+import React from 'react';
+import {DeviceEventEmitter} from 'react-native';
+
+import {switchToGlobalDrafts} from '@actions/local/draft';
+import DraftsButton from '@components/drafts_buttton/drafts_button';
+import {Events} from '@constants';
+import {DRAFT} from '@constants/screens';
+import {renderWithIntlAndTheme} from '@test/intl-test-helper';
+import {act, fireEvent} from '@testing-library/react-native';
+
+jest.mock('@actions/local/draft', () => ({
+    switchToGlobalDrafts: jest.fn(),
+}));
+
+describe('components/drafts_button/DraftsButton', () => {
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+
+    const baseProps = {
+        draftsCount: 0,
+        scheduledPostCount: 0,
+        scheduledPostHasError: false,
+    };
+
+    test('should show draft count when greater than 0', () => {
+        const props = {
+            ...baseProps,
+            draftsCount: 5,
+        };
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...props}/>,
+        );
+
+        expect(getByTestId('channel_list.drafts.count')).toHaveTextContent('5');
+    });
+
+    test('should show scheduled post count when greater than 0', () => {
+        const props = {
+            ...baseProps,
+            scheduledPostCount: 3,
+        };
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...props}/>,
+        );
+
+        expect(getByTestId('channel_list.scheduled_post.count')).toHaveTextContent('3');
+    });
+
+    test('should show both drafts and scheduled post count when greater than 0', () => {
+        const props = {
+            ...baseProps,
+            scheduledPostCount: 3,
+            draftsCount: 5,
+        };
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...props}/>,
+        );
+
+        expect(getByTestId('channel_list.drafts.count')).toHaveTextContent('5');
+        expect(getByTestId('channel_list.scheduled_post.count')).toHaveTextContent('3');
+    });
+
+    test('should show both drafts and scheduled post count when greater than 0 and scheduled post has error', () => {
+        const props = {
+            ...baseProps,
+            scheduledPostCount: 3,
+            draftsCount: 5,
+            scheduledPostHasError: true,
+        };
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...props}/>,
+        );
+
+        expect(getByTestId('channel_list.drafts.count')).toHaveTextContent('5');
+        expect(getByTestId('channel_list.scheduled_post.count')).toHaveTextContent('3');
+
+        const countElement = getByTestId('channel_list.scheduled_post.count.container');
+        expect(countElement).toBeVisible();
+        expect(countElement).toHaveStyle({backgroundColor: expect.any(String)});
+    });
+
+    test('should apply error styles when scheduled post has error', () => {
+        const props = {
+            ...baseProps,
+            scheduledPostCount: 1,
+            scheduledPostHasError: true,
+        };
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...props}/>,
+        );
+
+        const countElement = getByTestId('channel_list.scheduled_post.count.container');
+        expect(countElement).toBeVisible();
+        expect(countElement).toHaveStyle({backgroundColor: expect.any(String)});
+    });
+
+    test('should handle press and emit events', () => {
+        const emitSpy = jest.spyOn(DeviceEventEmitter, 'emit');
+
+        const {getByTestId} = renderWithIntlAndTheme(
+            <DraftsButton {...baseProps}/>,
+        );
+
+        const button = getByTestId('channel_list.drafts.button');
+        act(() => {
+            fireEvent.press(button);
+        });
+
+        expect(emitSpy).toHaveBeenCalledWith(Events.ACTIVE_SCREEN, DRAFT);
+        expect(switchToGlobalDrafts).toHaveBeenCalled();
+    });
+});

app/components/drafts_buttton/drafts_button.tsx

@@ -133,7 +133,10 @@ const DraftsButton: React.FC<DraftListProps> = ({
                 <View style={customStyles.countBadgeContainer}>
                     {
                         draftsCount > 0 &&
-                        <View style={customStyles.countContainer}>
+                        <View
+                            testID='channel_list.drafts.count.container'
+                            style={customStyles.countContainer}
+                        >
                             <CompassIcon
                                 name='pencil-outline'
                                 size={14}
@@ -150,15 +153,18 @@ const DraftsButton: React.FC<DraftListProps> = ({
                     }
                     {
                         scheduledPostCount > 0 &&
-                        <View style={[customStyles.countContainer, scheduledPostHasError && customStyles.errorCountContainer]}>
+                        <View
+                            testID='channel_list.scheduled_post.count.container'
+                            style={[customStyles.countContainer, scheduledPostHasError && customStyles.errorCountContainer]}
+                        >
                             <CompassIcon
                                 name='clock-send-outline'
                                 size={14}
                                 color={theme.sidebarText}
                                 style={scheduledPostHasError ? customStyles.badgeOpacityWithError : customStyles.opacity}
                             />
                             <Text
-                                testID='channel_list.schedued_post.count'
+                                testID='channel_list.scheduled_post.count'
                                 style={[customStyles.count, scheduledPostHasError && customStyles.badgeOpacityWithError]}
                             >
                                 {scheduledPostCount}

app/screens/home/channel_list/categories_list/index.test.tsx

@@ -36,6 +36,8 @@ describe('components/categories_list', () => {
                 moreThanOneTeam={false}
                 hasChannels={true}
                 draftsCount={0}
+                scheduledPostHasError={false}
+                scheduledPostCount={0}
             />,
             {database},
         );
@@ -50,6 +52,8 @@ describe('components/categories_list', () => {
                 moreThanOneTeam={false}
                 hasChannels={true}
                 draftsCount={0}
+                scheduledPostCount={0}
+                scheduledPostHasError={false}
             />,
             {database},
         );
@@ -67,6 +71,8 @@ describe('components/categories_list', () => {
                 moreThanOneTeam={false}
                 hasChannels={true}
                 draftsCount={1}
+                scheduledPostCount={0}
+                scheduledPostHasError={false}
             />,
             {database},
         );
@@ -85,6 +91,8 @@ describe('components/categories_list', () => {
                 moreThanOneTeam={false}
                 hasChannels={true}
                 draftsCount={0}
+                scheduledPostCount={0}
+                scheduledPostHasError={false}
             />,
             {database},
         );
@@ -108,6 +116,8 @@ describe('components/categories_list', () => {
                 moreThanOneTeam={true}
                 hasChannels={false}
                 draftsCount={0}
+                scheduledPostCount={0}
+                scheduledPostHasError={false}
             />,
             {database},
         );