babashka/test/babashka/test_utils.clj
2019-09-06 16:40:50 +02:00

45 lines
1.2 KiB
Clojure

(ns babashka.test-utils
(:require
[babashka.main :as main]
[me.raynes.conch :refer [let-programs] :as sh]))
(set! *warn-on-reflection* true)
(defn bb-jvm [input & args]
(let [es (java.io.StringWriter.)
os (java.io.StringWriter.)]
(binding [*err* es
*out* os]
(let [res (if input
(with-in-str input
(apply main/main args))
(apply main/main args))]
(if (zero? res)
(str os)
(throw (ex-info (str es)
{:stdout (str os)
:stderr (str es)})))))))
(defn bb-native [input & args]
(let-programs [bb "./bb"]
(try (if input
(apply bb (conj (vec args)
{:in input}))
(apply bb args))
(catch Exception e
(let [d (ex-data e)
err-msg (or (:stderr (ex-data e)) "")]
(throw (ex-info err-msg d)))))))
(def bb
(case (System/getenv "BABASHKA_TEST_ENV")
"jvm" #'bb-jvm
"native" #'bb-native
#'bb-jvm))
(def jvm? (= bb #'bb-jvm))
(def native? (not jvm?))
(if jvm?
(println "==== Testing JVM version")
(println "==== Testing native version"))