From fb626a4f01ba7aa0e4fc848c34df134ce7b97965 Mon Sep 17 00:00:00 2001
From: Andrey Antukh <niwi@niwi.nz>
Date: Mon, 6 Feb 2017 22:07:46 +0100
Subject: [PATCH] Add better test helper to user ns.

---
 src/user.clj | 28 ++++++++++++----------------
 1 file changed, 12 insertions(+), 16 deletions(-)

diff --git a/src/user.clj b/src/user.clj
index 303aca0..0503234 100644
--- a/src/user.clj
+++ b/src/user.clj
@@ -6,20 +6,16 @@
 
 ;; --- Development Stuff
 
-(defn test-vars
-  [& vars]
-  (repl/refresh)
-  (test/test-vars
-   (map (fn [sym]
-          (require (symbol (namespace sym)))
-          (resolve sym))
-        vars)))
+(defn test
+  ([] (test #"^datoteka.tests.*"))
+  ([o]
+   (repl/refresh)
+   (cond
+     (instance? java.util.regex.Pattern o)
+     (test/run-all-tests o)
 
-(defn test-ns
-  [ns]
-  (repl/refresh)
-  (test/test-ns ns))
-
-(defn test-all
-  ([] (test/run-all-tests #"^datoteka.tests.*"))
-  ([re] (test/run-all-tests re)))
+     (symbol? o)
+     (if-let [sns (namespace o)]
+       (do (require (symbol sns))
+           (test/test-vars [(resolve o)]))
+       (test/test-ns o)))))