From 3b008e92f17f7bf1a7c832262929daf65678d118 Mon Sep 17 00:00:00 2001
From: Aleksey2 Meleshko <a2.meleshko@a1qa.com>
Date: Tue, 28 Feb 2023 10:58:13 +0100
Subject: [PATCH] fix JavaScriptHandling initialization scripts removing
 feature after changes in https://github.com/SeleniumHQ/selenium/pull/11598

---
 .../selenium/browser/devtools/JavaScriptHandling.java        | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java b/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java
index 17e7151..d758b86 100644
--- a/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java
+++ b/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java
@@ -12,6 +12,7 @@
 import org.openqa.selenium.devtools.idealized.Events;
 import org.openqa.selenium.devtools.idealized.Javascript;
 import org.openqa.selenium.devtools.idealized.ScriptId;
+import org.openqa.selenium.devtools.idealized.target.model.SessionID;
 import org.openqa.selenium.devtools.v85.page.Page;
 import org.openqa.selenium.devtools.v85.page.model.ScriptIdentifier;
 import org.openqa.selenium.devtools.v85.runtime.Runtime;
@@ -111,7 +112,9 @@ private void removeInitializationScriptCore(InitializationScript script) {
             final Field pinnedScripts = Javascript.class.getDeclaredField("pinnedScripts");
             pinnedScripts.setAccessible(true);
             //noinspection unchecked
-            ((Map<String, ScriptId>)pinnedScripts.get(engine)).remove(script.getScriptSource());
+            ((Map<SessionID, Map<String, ScriptId>>)pinnedScripts.get(engine))
+                    .get(tools.getDevToolsSession().getCdpSession())
+                    .remove(script.getScriptSource());
             pinnedScripts.setAccessible(false);
         } catch (ReflectiveOperationException e) {
             AqualityServices.getLogger().fatal("Error while removing initialization script", e);