[py]: more unit test coverage for script keys

symonk · symonk · commit 702c12ab8506 · 2022-08-14T10:03:12.000+01:00
diff --git a/py/test/selenium/webdriver/common/script_pinning_tests.py b/py/test/selenium/webdriver/common/script_pinning_tests.py
@@ -14,10 +14,12 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+import re
 
 import pytest
 
 from selenium.common.exceptions import JavascriptException
+from selenium.webdriver.remote.script_key import ScriptKey
 
 
 def test_should_allow_script_pinning(driver, pages):
@@ -67,3 +69,10 @@ def test_calling_unpinned_script_causes_error(driver, pages):
     driver.unpin(cheese)
     with pytest.raises(JavascriptException):
         driver.execute_script(cheese)
+
+
+def test_unpinning_non_existing_script_raises(driver, pages):
+    pages.load("simpleTest.html")
+    with pytest.raises(KeyError, match=re.escape(r"No script with key: ScriptKey(id=1) existed in {}")):
+        driver.unpin(ScriptKey(1))
+
