Fix instrument for a dot special form case

clojure-emacs · Mar 9, 2020 · cc8d530 · cc8d530
1 parent 1673e08
commit cc8d530
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,7 @@
 
 * [#666](https://github.com/clojure-emacs/cider-nrepl/pull/666): Add a check in apropos var-query-map.
 * [#669](https://github.com/clojure-emacs/cider-nrepl/pull/669): Fix NPE and add isDirectory check in slurp
+* [#672](https://github.com/clojure-emacs/cider-nrepl/pull/672): Fix instrument for a dot special form case
 
 ### New Features
 

diff --git a/src/cider/nrepl/middleware/util/instrument.clj b/src/cider/nrepl/middleware/util/instrument.clj
@@ -73,7 +73,12 @@
             '#{new} (cons (first args) (instrument-coll (rest args)))
             '#{quote & var clojure.core/import*} args
             '#{.} (list* (first args)
-                         (second args)
+                         (let [s (second args)]
+                           (if (coll? s)
+                             (->> (rest s)
+                                  instrument-coll
+                                  (concat (cons (first s) '())))
+                             s))
                          (instrument-coll (rest (rest args))))
             '#{def} (let [sym (first args)]
                       (list* (m/merge-meta sym

diff --git a/test/clj/cider/nrepl/middleware/util/instrument_test.clj b/test/clj/cider/nrepl/middleware/util/instrument_test.clj
@@ -138,6 +138,14 @@
          '#{[(def foo "foo doc" (bp (bar) {:coor [3]} (bar))) []]
             [(bar) [3]]})))
 
+(deftest instrument-dot-test
+  (is (= (t/breakpoint-tester '(. clojure.lang.Numbers add 1 x))
+         '#{[(. clojure.lang.Numbers add 1 (bp x {:coor [4]} x)) []]
+            [x [4]]}))
+  (is (= (t/breakpoint-tester '(. clojure.lang.Numbers (add 1 x)))
+         '#{[x [2 2]]
+            [(. clojure.lang.Numbers (add 1 (bp x {:coor [2 2]} x))) []]})))
+
 (deftest instrument-set!-test
   (is (= (t/breakpoint-tester '(set! foo (bar)))
          '#{[(set! foo (bp (bar) {:coor [2]} (bar))) []]