CLJS-2665: Port clojure.spec.test.alpha/enumerate-namespace

Author: mfikes

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

@@ -22,6 +22,23 @@
     (list x)
     x))
 
+(defn- enumerate-namespace* [sym-or-syms]
+  (into #{}
+    (mapcat
+      (fn [sym]
+        (->> (vals (ana-api/ns-interns sym))
+          (map :name)
+          (map
+            (fn [name-sym]
+              (symbol (name sym) (name name-sym)))))))
+    (collectionize sym-or-syms)))
+
+(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."
+  [ns-sym-or-syms]
+  `'~(enumerate-namespace* (eval ns-sym-or-syms)))
+
 (defn- fn-spec-name?
   [s]
   (symbol? s))

src/test/cljs/cljs/spec/test/test_ns1.cljs

@@ -0,0 +1,5 @@
+(ns cljs.spec.test.test-ns1)
+
+(def x 1)
+
+(def y 2)

src/test/cljs/cljs/spec/test/test_ns2.cljs

@@ -0,0 +1,3 @@
+(ns cljs.spec.test.test-ns2)
+
+(def z 3)

src/test/cljs/cljs/spec/test_test.cljs

@@ -2,7 +2,9 @@
   (:require-macros [cljs.spec.test.test-macros])
   (:require [cljs.test :as test :refer-macros [deftest is are run-tests]]
             [cljs.spec.alpha :as s]
-            [cljs.spec.test.alpha :as stest]))
+            [cljs.spec.test.alpha :as stest]
+            [cljs.spec.test.test-ns1]
+            [cljs.spec.test.test-ns2]))
 
 (s/fdef clojure.core/symbol
   :args (s/alt :separate (s/cat :ns string? :n string?)
@@ -85,3 +87,13 @@
 
 (deftest test-2755
   (is (uri? (ffirst (s/exercise uri? 1)))))
+
+(deftest test-cljs-2665
+  (is (= '#{cljs.spec.test.test-ns1/x cljs.spec.test.test-ns1/y cljs.spec.test.test-ns2/z}
+        (stest/enumerate-namespace '[cljs.spec.test.test-ns1 cljs.spec.test.test-ns2])))
+  (is (= '#{cljs.spec.test.test-ns1/x cljs.spec.test.test-ns1/y cljs.spec.test.test-ns2/z}
+        (stest/enumerate-namespace ['cljs.spec.test.test-ns1 'cljs.spec.test.test-ns2])))
+  (is (= '#{cljs.spec.test.test-ns1/x cljs.spec.test.test-ns1/y}
+        (stest/enumerate-namespace 'cljs.spec.test.test-ns1)))
+  (is (= '#{cljs.spec.test.test-ns2/z}
+        (stest/enumerate-namespace 'cljs.spec.test.test-ns2))))