bec-widgets 0.82.2__py3-none-any.whl → 0.83.0__py3-none-any.whl

tests/unit_tests/test_spinner.py

@@ -0,0 +1,30 @@
+import pytest
+import qdarktheme
+
+from bec_widgets.utils.reference_utils import snap_and_compare
+from bec_widgets.widgets.spinner.spinner import SpinnerWidget
+
+
+@pytest.fixture
+def spinner_widget(qtbot):
+    qdarktheme.setup_theme("light")
+    spinner = SpinnerWidget()
+    qtbot.addWidget(spinner)
+    qtbot.waitExposed(spinner)
+    yield spinner
+
+
+def test_spinner_widget_paint_event(spinner_widget, qtbot):
+    spinner_widget.paintEvent(None)
+
+
+def test_spinner_widget_rendered(spinner_widget, qtbot, tmpdir):
+    spinner_widget.update()
+    qtbot.wait(200)
+    snap_and_compare(spinner_widget, str(tmpdir), suffix="")
+
+    spinner_widget._started = True
+    spinner_widget.update()
+    qtbot.wait(200)
+
+    snap_and_compare(spinner_widget, str(tmpdir), suffix="started")

tests/unit_tests/test_vscode_widget.py

@@ -25,54 +25,49 @@ def test_vscode_widget(qtbot, vscode_widget):
 def test_start_server(qtbot, mocked_client):
 
     with mock.patch("bec_widgets.widgets.vscode.vscode.subprocess.Popen") as mock_popen:
-        mock_process = mock.Mock()
-        mock_process.stdout.fileno.return_value = 1
-        mock_process.poll.return_value = None
-        mock_process.stdout.read.return_value = (
-            f"available at http://{VSCodeEditor.host}:{VSCodeEditor.port}?tkn={VSCodeEditor.token}"
-        )
-        mock_popen.return_value = mock_process
-
-        widget = VSCodeEditor(client=mocked_client)
-
-        mock_popen.assert_called_once_with(
-            shlex.split(
-                f"code serve-web --port {widget.port} --connection-token={widget.token} --accept-server-license-terms"
-            ),
-            text=True,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.DEVNULL,
-            preexec_fn=os.setsid,
-        )
+        with mock.patch("bec_widgets.widgets.vscode.vscode.select.select") as mock_select:
+            mock_process = mock.Mock()
+            mock_process.stdout.fileno.return_value = 1
+            mock_process.poll.return_value = None
+            mock_process.stdout.read.return_value = f"available at http://{VSCodeEditor.host}:{VSCodeEditor.port}?tkn={VSCodeEditor.token}"
+            mock_popen.return_value = mock_process
+            mock_select.return_value = [[mock_process.stdout], [], []]
+
+            widget = VSCodeEditor(client=mocked_client)
+
+            mock_popen.assert_called_once_with(
+                shlex.split(
+                    f"code serve-web --port {widget.port} --connection-token={widget.token} --accept-server-license-terms"
+                ),
+                text=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.DEVNULL,
+                preexec_fn=os.setsid,
+            )
 
 
 @pytest.fixture
 def patched_vscode_process(qtbot, vscode_widget):
     with mock.patch("bec_widgets.widgets.vscode.vscode.os.killpg") as mock_killpg:
         with mock.patch("bec_widgets.widgets.vscode.vscode.os.getpgid") as mock_getpgid:
-            with mock.patch(
-                "bec_widgets.widgets.website.website.WebsiteWidget.closeEvent"
-            ) as mock_close_event:
-                mock_getpgid.return_value = 123
-                vscode_widget.process = mock.Mock()
-                yield vscode_widget, mock_killpg, mock_close_event
+            mock_getpgid.return_value = 123
+            vscode_widget.process = mock.Mock()
+            yield vscode_widget, mock_killpg
 
 
-def test_close_event(qtbot, patched_vscode_process):
-    vscode_patched, mock_killpg, mock_close_event = patched_vscode_process
+def test_vscode_cleanup(qtbot, patched_vscode_process):
+    vscode_patched, mock_killpg = patched_vscode_process
     vscode_patched.process.pid = 123
     vscode_patched.process.poll.return_value = None
-    vscode_patched.closeEvent(None)
+    vscode_patched.cleanup()
     mock_killpg.assert_called_once_with(123, 15)
     vscode_patched.process.wait.assert_called_once()
-    mock_close_event.assert_called_once()
 
 
 def test_close_event_on_terminated_code(qtbot, patched_vscode_process):
-    vscode_patched, mock_killpg, mock_close_event = patched_vscode_process
+    vscode_patched, mock_killpg = patched_vscode_process
     vscode_patched.process.pid = 123
     vscode_patched.process.poll.return_value = 0
-    vscode_patched.closeEvent(None)
+    vscode_patched.cleanup()
     mock_killpg.assert_not_called()
     vscode_patched.process.wait.assert_not_called()
-    mock_close_event.assert_called_once()