Added test

radarhere · Jan 27, 2024 · 4c4b3da · 4c4b3da
1 parent 9392f24
commit 4c4b3da
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Tests/test_imagegrab.py b/Tests/test_imagegrab.py
@@ -119,3 +119,15 @@ def test_grabclipboard_wl_clipboard(self, ext):
             subprocess.call(["wl-copy"], stdin=fp)
         im = ImageGrab.grabclipboard()
         assert_image_equal_tofile(im, image_path)
+
+    @pytest.mark.skipif(
+        (
+            sys.platform != "linux"
+            or not all(shutil.which(cmd) for cmd in ("wl-paste", "wl-copy"))
+        ),
+        reason="Linux with wl-clipboard only",
+    )
+    @pytest.mark.parametrize("arg", ("text", "--clear"))
+    def test_grabclipboard_wl_clipboard_errors(self, arg):
+        subprocess.call(["wl-copy", arg])
+        assert ImageGrab.grabclipboard() is None