From 80ac818616c09fa051b3701da831b46aa8ee583c Mon Sep 17 00:00:00 2001
From: Jerko Steiner <jerko.steiner@gmail.com>
Date: Tue, 10 Mar 2020 13:16:03 +0100
Subject: [PATCH] Fix Toolbar.test.tsx

---
 src/client/components/Toolbar.test.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/client/components/Toolbar.test.tsx b/src/client/components/Toolbar.test.tsx
index 0edb80a..5c1da9a 100644
--- a/src/client/components/Toolbar.test.tsx
+++ b/src/client/components/Toolbar.test.tsx
@@ -49,7 +49,7 @@ describe('components/Toolbar', () => {
     onToggleChat = jest.fn()
     onSendFile = jest.fn()
     onHangup = jest.fn()
-    onGetDesktopStream = jest.fn()
+    onGetDesktopStream = jest.fn().mockImplementation(() => Promise.resolve())
     onRemoveStream = jest.fn()
     const div = document.createElement('div')
     const stream: StreamWithURL = {
@@ -170,7 +170,7 @@ describe('components/Toolbar', () => {
         type: STREAM_TYPE_DESKTOP,
       }
       await render()
-      const shareDesktop = node.querySelector('.share-desktop')!
+      const shareDesktop = node.querySelector('.stream-desktop')!
       expect(shareDesktop).toBeDefined()
       TestUtils.Simulate.click(shareDesktop)
       await Promise.resolve()