fix headless? && save capabilities (#249)

* fix headless? && save capabilities * update
clj-commons · Jul 27, 2020 · 1b8ff07 · 1b8ff07
1 parent eede672
commit 1b8ff07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/etaoin/api.clj b/src/etaoin/api.clj
@@ -2958,15 +2958,15 @@
            (throw e))))
 
   (swap! driver dissoc
-         :session :capabilities)
+         :session)
   driver)
 
 (defn stop-driver
   "Stops the driver's process. Removes proces's data from the driver
   instance. Returns a modified driver."
   [driver]
   (proc/kill (:process @driver))
-  (swap! driver dissoc :process :args :env)
+  (swap! driver dissoc :process :args :env :capabilities)
   driver)
 
 (defn boot-driver

diff --git a/src/etaoin/driver.clj b/src/etaoin/driver.clj
@@ -245,7 +245,9 @@
 (defmethod is-headless?
   :default
   [driver]
-  (:headless driver))
+  (if-let [args (get-in driver [:capabilities (options-name driver) :args])]
+    (contains? (set args) "--headless")
+    (:headless driver)))
 
 (defmethod is-headless?
   :phantom