implement checkable-syms

Author: swannodette

src/main/cljs/cljs/spec/test.cljc

@@ -21,6 +21,10 @@
     (list x)
     x))
 
+(defn- fn-spec-name?
+  [s]
+  (symbol? s))
+
 (defmacro enumerate-namespace
   "Given a symbol naming an ns, or a collection of such symbols,
 returns the set of all symbols naming vars in those nses."
@@ -175,4 +179,16 @@ Returns a collection of syms naming the vars unstrumented."
   ([f spec opts]
    `(let [opts# ~opts]
       (validate-check-opts opts#)
-      (check-1 nil ~f ~spec opts#))))
+      (check-1 nil ~f ~spec opts#))))
+
+(defmacro checkable-syms
+  "Given an opts map as per check, returns the set of syms that
+can be checked."
+  ([]
+   `(checkable-syms nil))
+  ([opts]
+   `(let [opts# ~opts]
+      (validate-check-opts opts#)
+      (reduce conj #{}
+        '[~@(filter fn-spec-name? (keys @s/registry-ref))
+          ~@(keys (:spec opts))]))))

src/main/cljs/cljs/spec/test.cljs

@@ -226,15 +226,6 @@ with explain-data + ::s/failure."
   [opts]
   (assert (every? ident? (keys (:gen opts))) "check :gen expects ident keys"))
 
-(defn checkable-syms
-  "Given an opts map as per check, returns the set of syms that
-can be checked."
-  ([] (checkable-syms nil))
-  ([opts]
-   (validate-check-opts opts)
-   (reduce into #{} [(filter fn-spec-name? (keys (s/registry)))
-                     (keys (:spec opts))])))
-
 (defn check
   "Run generative tests for spec conformance on vars named by
 sym-or-syms, a symbol or collection of symbols. If sym-or-syms
@@ -394,6 +385,8 @@ key with a count for each different :type of result."
     (s/fspec
       :args (s/cat :x int?)
       :ret  int?))
+
+  (m/checkable-syms)
   )