From cdb7cc02e4041f28e3bd9f0f78cf862f4b63e96f Mon Sep 17 00:00:00 2001 From: Konrad Mrożek <kmrozek@comscore.com> Date: Fri, 12 Jan 2024 13:16:08 +0000 Subject: [PATCH] Disable colors in malli pretty output --- clojure/src/test_runner.clj | 180 ++++++++++++++++++++++++++++++++++++++++++++++++++---------- 1 files changed, 150 insertions(+), 30 deletions(-) diff --git a/clojure/src/test_runner.clj b/clojure/src/test_runner.clj index 648e58a..eb7c47e 100644 --- a/clojure/src/test_runner.clj +++ b/clojure/src/test_runner.clj @@ -1,37 +1,157 @@ -(binding [*compile-files* true] - (require 'clojure.test)) - (ns test-runner - (:require [clojure.test :as t])) + (:require + [clojure.string :as string] + [clojure.test :as t])) -(when-not (.exists (java.io.File. "classes")) - (.. (java.io.File. "classes") mkdir)) +(def class-cache-dir ".cache/classes") +(.mkdirs (java.io.File. class-cache-dir)) -(defmulti emacs-report :type) +(defmacro with-class-cache [& body] + `(binding [*compile-path* class-cache-dir + *compile-files* true] + ~@body)) -(defmethod emacs-report :pass +(with-class-cache + (require '[clojure.test :as t] + '[lambdaisland.deep-diff2 :as ddiff])) + +(def no-colors-printer (ddiff/printer {:print-color false})) +(defmulti vim-report :type) + +(defmethod vim-report :begin-test-ns [m] + (println "\nTesting" (ns-name (:ns m)))) + +(defmethod vim-report :begin-test-var [m] + (println "\nExecuting" (:name (meta (:var m))))) + +(defmethod vim-report :fail [m] - (t/with-test-out - (println "PASS:"))) + (t/inc-report-counter :fail) + (when-let [source-file (some-> t/*testing-vars* + first + meta + :file)] + (println (str "FAIL:" source-file ":" (:line m) ":" (t/testing-vars-str m) ":" (t/testing-contexts-str) ":" (:message m "FAIL"))) + (println (str "FAIL-CONTINUE:EXPECTED:" (pr-str (:expected m)))) + (println (str "FAIL-CONTINUE:ACTUAL:" (pr-str (:actual m)))))) -(defn -main [& [test-file]] - (binding [*compile-files* true] +(defmethod t/assert-expr '= [msg form] + (let [args (rest form) + pred (first form)] + `(let [values# (list ~@args) + result# (apply ~pred values#)] + (if result# + (t/do-report {:type :pass, :message ~msg, + :expected '~form, :actual (cons '~pred values#)}) + (do + (t/do-report {:type :fail, :message ~msg, + :expected '~form, :actual (list '~'not (cons '~pred values#))}) + (-> (apply ddiff/diff values#) + (ddiff/pretty-print no-colors-printer)))) + result#))) + +(defn- find-line-number [source-file m] + (if (instance? Throwable (:actual m)) + (let [fname (-> source-file (java.io.File.) (.getName))] + (->> m + :actual + Throwable->map + :trace + (some (fn [[_ _ e-file e-line]] + (when (= e-file fname) + e-line))))) + (:line m))) + +(defmethod vim-report :error + [m] + (t/inc-report-counter :error) + (when-let [source-file (some-> t/*testing-vars* + first + meta + :file)] + (let [line (find-line-number source-file m)] + (println (str "ERROR:" source-file ":" line ":" (t/testing-vars-str m) ":" (t/testing-contexts-str) ":" (:message m "FAIL"))) + (println (str "ERROR-CONTINUE:EXPECTED:" (pr-str (:expected m)))) + (println (str "ERROR-CONTINUE:ACTUAL:" + (if (instance? Throwable (:actual m)) + (ex-message (:actual m)) + (pr-str (:actual m))))) + (when (instance? Throwable (:actual m)) + (.printStackTrace (:actual m)))))) + +(defmethod vim-report :default + [_]) + +(defn- clj-file? [f] + (re-matches #"^.*\.cljs?$" (.getName f))) + +(defn find-closest-test [test-file test-line] + (->> (all-ns) + (mapcat ns-publics) + (map second) + (filter (comp :test meta)) + (filter (comp #{test-file} :file meta)) + (map #(vector % (- test-line (-> % meta :line)))) + (filterv (comp pos? second)) + (sort-by second) + first + first)) + +(defn -main [& {:strs [-test-file -test-line] :or {-test-file "test"}}] + (with-class-cache (compile 'test-runner) - (let [test-namespaces (->> (or test-file "test") - (java.io.File.) - (file-seq) - (filter (memfn isFile)) - (map (memfn getPath)) - (map load-file) - (map (comp :ns meta)) - (into #{}))] - (System/exit (if (pos? (reduce (fn [total-fails n] - (with-redefs [t/report emacs-report] - (let [results (t/run-tests n)] - (+ total-fails - (:fail results 0) - (:error results 0))))) - 0 - test-namespaces)) - 1 - 0))))) + (println "Detecting test files in" -test-file) + (let [test-files (->> -test-file + (java.io.File.) + (file-seq) + (filter (memfn isFile)) + (filter clj-file?) + (map (memfn getAbsolutePath)) + (set))] + (println "Loading test files...") + (run! load-file test-files) + (when (find-ns 'malli.core) + (println "Malli detected. Instrument functions...") + (require 'malli.dev) + (require 'malli.dev.pretty) + (require 'malli.dev.virhe) + (alter-var-root (get (ns-publics 'malli.dev.virhe) '-color) (constantly (fn [_ body _] [:span body]))) + ((find-var 'malli.dev/start!) {:report ((find-var 'malli.dev.pretty/thrower))})) + (let [test-namespaces (->> (all-ns) + (mapcat ns-publics) + (map (comp meta second)) + (filter :test) + (filter (comp test-files :file)) + (map :ns) + (set))] + + (set-validator! #'t/*testing-contexts* + (let [validator (get-validator #'t/*testing-contexts*)] + (fn [v] + (println "Testing:" (string/join " " v)) + (println) + (if validator + (validator v) + true)))) + + (with-redefs [t/report vim-report] + (System/exit + (if (pos? (if (and -test-file -test-line) + + (if-let [test-var (find-closest-test (.. (java.io.File. -test-file) getAbsolutePath) + (parse-long -test-line))] + (->> (t/run-test-var test-var) + ((juxt :fail :error)) + (apply +)) + (do + (println "No test found") + 0)) + (reduce (fn [total-fails n] + (let [results (t/run-tests n)] + (+ total-fails + (:fail results 0) + (:error results 0)))) + 0 + test-namespaces))) + 1 + 0))))))) -- Gitblit v1.9.3