24 lines
911 B
Clojure
24 lines
911 B
Clojure
(ns babashka.shutdown-hook-test
|
|
{:no-doc true}
|
|
(:import [java.nio.charset Charset])
|
|
(:require [babashka.test-utils :as tu]
|
|
[clojure.java.io :as io]
|
|
[clojure.test :refer [deftest is]]))
|
|
|
|
(defn- stream-to-string
|
|
([in] (stream-to-string in (.name (Charset/defaultCharset))))
|
|
([in enc]
|
|
(with-open [bout (java.io.StringWriter.)]
|
|
(io/copy in bout :encoding enc)
|
|
(.toString bout))))
|
|
|
|
(deftest shutdown-hook-test
|
|
(let [script "(-> (Runtime/getRuntime) (.addShutdownHook (Thread. #(println \"bye\"))))"
|
|
pb (ProcessBuilder. (if tu/jvm?
|
|
["lein" "bb" "-e" script]
|
|
["./bb" "-e" script]))
|
|
process (.start pb)
|
|
output (.getInputStream process)]
|
|
(when-let [s (not-empty (stream-to-string (.getErrorStream process)))]
|
|
(prn "ERROR:" s))
|
|
(is (= "bye\n" (stream-to-string output)))))
|