making trim-v before/after symmetrical

day8 · Dec 9, 2016 · 8958989 · 8958989
1 parent acdcf54
commit 8958989
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/src/re_frame/std_interceptors.cljc b/src/re_frame/std_interceptors.cljc
@@ -63,7 +63,14 @@
     :id      :trim-v
     :before  (fn trimv-before
                [context]
-               (update-in context [:coeffects :event] subvec 1))))
+               (-> context
+                   (update-in [:coeffects :event] subvec 1)
+                   (assoc-in [:coeffects ::untrimmed-event] (get-coeffect context :event))))
+    :after   (fn trimv-after
+               [context]
+               (-> context
+                   (assoc-in [:coeffects :event] (get-coeffect context ::untrimmed-event))
+                   (update :coeffects dissoc ::untrimmed-event)))))
 
 
 ;; -- Interceptor Factories - PART 1 ---------------------------------------------------------------
@@ -274,5 +281,3 @@
                       (assoc-in new-db out-path)
                       (assoc-effect context :db))
                  context)))))
-
-
diff --git a/test/re-frame/interceptor_test.cljs b/test/re-frame/interceptor_test.cljs
@@ -8,10 +8,13 @@
 (enable-console-print!)
 
 (deftest test-trim-v
-  (let [c  (-> (context [:a :b :c] [])
-               ((:before trim-v)))]
+  (let [ctx (context [:a :b :c] [])
+        c  ((:before trim-v) ctx)]
     (is (= (get-coeffect c :event)
-           [:b :c]))))
+           [:b :c]))
+
+    (let [c-after ((:after trim-v) c)]
+      (is (= c-after ctx)))))
 
 
 (deftest test-one-level-path
@@ -114,6 +117,3 @@
                 ((:before change-handler-i))       ;; cause change to :a
                 ((:after change-i))
                 (get-effect :db))))))
-
-
-