Tests: Add lab 4 global div class names

Author: agoscinski

tests/conftest.py

@@ -115,7 +115,10 @@ def _selenium_driver(nb_path):
                     "Restart Kernel and Run All Cells…"
                 )
             else:
-                raise ValueError("jupyter lab > 4.0.0 is not supported.")
+                restart_kernel_button_class_name = "jp-ToolbarButtonComponent"
+                restart_kernel_button_title_attribute = (
+                    "Restart the kernel and run all cells"
+                )
         elif JUPYTER_TYPE == "notebook":
             if get_jupyter_version() < Version("7.0.0"):
                 restart_kernel_button_class_name = "btn.btn-default"
@@ -176,7 +179,10 @@ def _selenium_driver(nb_path):
                 )
                 restart_button_text = "Restart"
             else:
-                raise ValueError("jupyter lab > 4.0.0 is not supported.")
+                restart_button_class_name = (
+                    "jp-Dialog-button.jp-mod-accept.jp-mod-warn.jp-mod-styled"
+                )
+                restart_button_text = "Restart"
         elif JUPYTER_TYPE == "notebook":
             if get_jupyter_version() < Version("7.0.0"):
                 restart_button_class_name = "btn.btn-default.btn-sm.btn-danger"

tests/test_widgets.py

@@ -160,6 +160,22 @@ def test_setup_globals():
             "lm-Widget.p-Widget.lm-Panel.p-Panel.jupyter-widgets."
             "widget-container.widget-box.widget-vbox.scwidget-cue-box"
         )
+    elif JUPYTER_TYPE == "lab" and JUPYTER_VERSION >= Version("4.0.0"):
+        BUTTON_CLASS_NAME = "lm-Widget.jupyter-widgets.jupyter-button.widget-button"
+        OUTPUT_CLASS_NAME = (
+            "lm-Widget.jp-RenderedText.jp-mod-trusted.jp-OutputArea-output"
+        )
+
+        TEXT_INPUT_CLASS_NAME = "widget-input"
+        CODE_MIRROR_CLASS_NAME = "cm-content"
+
+        MATPLOTLIB_CANVAS_CLASS_NAME = (
+            "jupyter-widgets.jupyter-matplotlib-canvas-container"
+        )
+        CUE_BOX_CLASS_NAME = (
+            "lm-Widget.lm-Panel.jupyter-widgets.widget-container."
+            "widget-box.widget-vbox.scwidget-cue-box"
+        )
     else:
         raise ValueError(
             f"Tests do not support jupyter type {JUPYTER_TYPE!r} for version"