add potential tests for treatment of comments at end of file.

clj-commons · Jan 25, 2015 · 5392ba0 · 5392ba0
1 parent 38036ba
commit 5392ba0
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/test/rewrite_clj/zip/insert_test.clj b/test/rewrite_clj/zip/insert_test.clj
@@ -3,7 +3,8 @@
             [rewrite-clj.zip
              [base :as base]
              [move :as m]
-             [insert :refer :all]]))
+             [insert :refer :all]]
+            [fast-zip.core :as z]))
 
 (tabular
   (fact "about whitespace-aware insertion."
@@ -29,3 +30,10 @@
   "[%s]"    0    append-child     "[1 2 3 4 x]"
   "[ %s]"   0    insert-child     "[x 1 2 3 4]"
   "[%s ]"   0    append-child     "[1 2 3 4 x]")
+
+(future-fact
+  "about inserting after comment."
+  (let [loc (-> (base/of-string "[1 2 3] ; this is a comment")
+                (z/rightmost)
+                (insert-right 'x))]
+    (base/root-string loc) => "[1 2 3] ; this is a comment\nx"))
diff --git a/test/rewrite_clj/zip/remove_test.clj b/test/rewrite_clj/zip/remove_test.clj
@@ -3,7 +3,8 @@
             [rewrite-clj.zip
              [base :as base]
              [move :as m]
-             [remove :as r]]))
+             [remove :as r]]
+            [fast-zip.core :as z]))
 
 (tabular
   (fact "about whitespace-aware removal."
@@ -34,3 +35,10 @@
         (-> root m/next m/down r/remove base/root-string)
         => (str "  :k [[d e f]]\n"
                 "  :keyword 0")))
+
+(future-fact
+  "about removing after comment."
+  (let [loc (-> (base/of-string "; comment\nx")
+                (z/rightmost)
+                (r/remove))]
+    (base/root-string loc) => "; comment"))