fixed test for coverage

Accruent · Dec 7, 2020 · 1e42a52 · 1e42a52
1 parent b809a5c
commit 1e42a52
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/test/Desktop/test_desktop.py b/test/Desktop/test_desktop.py
@@ -703,9 +703,21 @@ def test_start_screen_recording(self):
         mock_desk = MagicMock()
         mock_desk._recording = None
         DesktopLibrary.start_screen_recording(mock_desk)
+        self.assertTrue(mock_desk._recording == unittest.mock.ANY)
 
     def test_stop_screen_recording(self):
         mock_desk = MagicMock()
         mock_desk._recording = None
         DesktopLibrary.start_screen_recording(mock_desk)
-        DesktopLibrary.stop_screen_recording(mock_desk)
+        file = DesktopLibrary.stop_screen_recording(mock_desk)
+        self.assertTrue(file == unittest.mock.ANY)
+
+    def test_save_recording(self):
+        mock_desk = MagicMock()
+        mock_desk._recording = b'some data'
+        mock_desk._is_remotepath_set = MagicMock(return_value=False)
+        options = ["username=test"]
+        mock_desk._get_screenrecord_paths = MagicMock(return_value=("path", "link"))
+        file = DesktopLibrary._save_recording(mock_desk, "filename", options)
+        self.assertTrue(file == "path")
+        os.remove("path")