Add await for toggleContentShare

src/providers/ContentShareProvider/index.tsx

@@ -128,11 +128,11 @@ export const ContentShareProvider: React.FC<
         audioVideo.stopContentShare();
       } else {
         if (source && typeof source === 'string') {
-          audioVideo.startContentShareFromScreenCapture(source);
-        } else if (source instanceof MediaStream) {
-          audioVideo.startContentShare(source);
+          await audioVideo.startContentShareFromScreenCapture(source);
+        } else if (source && source instanceof MediaStream) {
+          await audioVideo.startContentShare(source);
         } else {
-          audioVideo.startContentShareFromScreenCapture();
+          await audioVideo.startContentShareFromScreenCapture();
         }
         dispatch({ type: ContentActionType.STARTING });
       }

tst/providers/ContentShareProvider/index.test.tsx

@@ -0,0 +1,55 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import React from 'react';
+import { renderHook, act } from '@testing-library/react';
+import {
+  ContentShareProvider,
+  useContentShareControls,
+} from '../../../src/providers/ContentShareProvider';
+
+// Mock audioVideo object
+const mockAudioVideo = {
+    startContentShareFromScreenCapture: jest.fn(),
+    startContentShare: jest.fn(),
+    stopContentShare: jest.fn(),
+    addObserver: jest.fn(),
+    removeObserver: jest.fn(),
+    addContentShareObserver: jest.fn(),
+    removeContentShareObserver: jest.fn()
+ };
+ // Mock the module containing useAudioVideo hook
+jest.mock('../../../src/providers/AudioVideoProvider', () => ({
+    useAudioVideo: () => mockAudioVideo
+}));
+
+describe('toggleContentShare', () => {
+  it('Should call startContentShareFromScreenCapture', async () => {
+    const { result } = renderHook(() => useContentShareControls(), {
+      wrapper: ({ children }) => (
+        <ContentShareProvider>{children}</ContentShareProvider>
+      ),
+    });
+    await act(async () => {
+      await result.current.toggleContentShare();
+    });
+    expect(mockAudioVideo.startContentShareFromScreenCapture).toHaveBeenCalledTimes(1);
+  });
+
+  it('Can catch error from startContentShareFromScreenCapture', async () => {
+    mockAudioVideo.startContentShareFromScreenCapture = jest.fn(() => Promise.reject(new Error('startContentShare error')));
+    const { result } = renderHook(() => useContentShareControls(), {
+      wrapper: ({ children }) => (
+        <ContentShareProvider>{children}</ContentShareProvider>
+      ),
+    });
+    await expect(async () => { 
+      await act(async () => {
+        await result.current.toggleContentShare();
+    });
+    }).rejects.toThrow('startContentShare error');
+    // expect(() => {
+    //     result.current.toggleContentShare();
+    //   }).toThrow('startContentShare error');
+  });
+});