Update test.selenium for Django 5.1 (#2559)

From django/django@f6629ee:

* `django.test.selenium.ChangeWindowSize` added
* `django.test.selenium.SeleniumTestCase.desktop_size` added
* `django.test.selenium.SeleniumTestCase.mobile_size` added
* `django.test.selenium.SeleniumTestCase.small_screen_size` added

From django/django@be56c98:

* `django.test.selenium.SeleniumTestCase.dark` added
* `django.test.selenium.SeleniumTestCase.rtl` added
* `django.test.selenium.SeleniumTestCase.take_screenshot` added
* `django.test.selenium.SeleniumTestCase.screenshots` added
* `django.test.selenium.screenshot_cases` added

From django/django@39828fa:

* `django.test.selenium.SeleniumTestCase.high_contrast` added
* `django.test.selenium.SeleniumTestCase.set_emulated_media` added

From django/django@ceaf1e2:

* `django.test.selenium.SeleniumTestCase.set_emulated_media` adjusted

Author: adamchainz

django-stubs/test/selenium.pyi

@@ -1,7 +1,12 @@
+from collections.abc import Callable, Sequence
 from contextlib import AbstractContextManager
-from typing import Any
+from types import TracebackType
+from typing import Any, TypeVar
 
 from django.test import LiveServerTestCase
+from typing_extensions import Self
+
+_F = TypeVar("_F", bound=Callable[..., Any])
 
 class SeleniumTestCaseBase:
     browsers: Any
@@ -10,7 +15,28 @@ class SeleniumTestCaseBase:
     def import_webdriver(cls, browser: Any) -> type[Any]: ...  # Type[WebDriver]
     def create_webdriver(self) -> Any: ...  # WebDriver
 
+class ChangeWindowSize:
+    def __init__(self, width: int, height: int, selenium: Any) -> None: ...
+    def __enter__(self) -> Self: ...
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None: ...
+
 class SeleniumTestCase(LiveServerTestCase):
     implicit_wait: int
+    screenshots: bool = False
     selenium: Any
+    def desktop_size(self) -> AbstractContextManager[None]: ...
+    def small_screen_size(self) -> AbstractContextManager[None]: ...
+    def mobile_size(self) -> AbstractContextManager[None]: ...
+    def rtl(self) -> AbstractContextManager[None]: ...
+    def dark(self) -> AbstractContextManager[None]: ...
+    def set_emulated_media(self, *, media: str | None = None, features: list[dict[str, str]] | None = None) -> None: ...
+    def high_contrast(self) -> AbstractContextManager[None]: ...
+    def take_screenshot(self, name: str) -> None: ...
     def disable_implicit_wait(self) -> AbstractContextManager[None]: ...
+
+def screenshot_cases(method_names: str | Sequence[str]) -> Callable[[_F], _F]: ...

scripts/stubtest/allowlist_todo_django51.txt

@@ -79,17 +79,6 @@ django.forms.ModelChoiceField.validate_no_null_characters
 django.forms.RadioSelect.__slotnames__
 django.forms.models.ModelChoiceField.validate_no_null_characters
 django.forms.widgets.RadioSelect.__slotnames__
-django.test.selenium.ChangeWindowSize
-django.test.selenium.SeleniumTestCase.dark
-django.test.selenium.SeleniumTestCase.desktop_size
-django.test.selenium.SeleniumTestCase.high_contrast
-django.test.selenium.SeleniumTestCase.mobile_size
-django.test.selenium.SeleniumTestCase.rtl
-django.test.selenium.SeleniumTestCase.screenshots
-django.test.selenium.SeleniumTestCase.set_emulated_media
-django.test.selenium.SeleniumTestCase.small_screen_size
-django.test.selenium.SeleniumTestCase.take_screenshot
-django.test.selenium.screenshot_cases
 django.test.signals.form_renderer_changed
 django.test.utils.garbage_collect
 django.urls.resolvers.whitespace_set