🎉 Merge tokens-studio

frontend/test/frontend_tests/tokens/helpers/state.cljs

@@ -0,0 +1,74 @@
+(ns frontend-tests.tokens.helpers.state
+  (:require
+   [app.common.types.tokens-lib :as ctob]
+   [app.main.ui.workspace.tokens.style-dictionary :as sd]
+   [beicon.v2.core :as rx]
+   [potok.v2.core :as ptk]))
+
+(defn end
+  "Apply `attributes` that match `token` for `shape-ids`.
+
+  Optionally remove attributes from `attributes-to-remove`,
+  this is useful for applying a single attribute from an attributes set
+  while removing other applied tokens from this set."
+  []
+  (ptk/reify ::end
+    ptk/WatchEvent
+    (watch [_ _ _]
+      (rx/empty))))
+
+(defn end+
+  []
+  (ptk/reify ::end+
+    ptk/WatchEvent
+    (watch [_ state _]
+      (->> (rx/from (-> (get-in state [:workspace-data :tokens-lib])
+                        (ctob/get-active-themes-set-tokens)
+                        (sd/resolve-tokens+)))
+           (rx/mapcat #(rx/of (end)))))))
+
+(defn stop-on
+  "Helper function to be used with async version of run-store.
+
+  Will stop the execution after event with `event-type` has completed."
+  [event-type]
+  (fn [stream]
+    (->> stream
+         #_(rx/tap #(prn (ptk/type %)))
+         (rx/filter #(ptk/type? event-type %)))))
+
+(def stop-on-send-update-indices
+  "Stops on `send-update-indices` function being called, which should be the last function of an event chain."
+  (stop-on ::end))
+
+;; Support for async events in tests
+;; https://chat.kaleidos.net/penpot-partners/pl/tz1yoes3w3fr9qanxqpuhoz3ch
+(defn run-store
+  "Async version of `frontend-tests.helpers.state/run-store`."
+  ([store done events completed-cb]
+   (run-store store done events completed-cb nil))
+  ([store done events completed-cb stopper]
+   (let [stream (ptk/input-stream store)
+         stopper-s (if (fn? stopper)
+                     (stopper stream)
+                     (rx/filter #(= :the/end %) stream))]
+     (->> stream
+          (rx/take-until stopper-s)
+          (rx/last)
+          (rx/tap (fn [_]
+                    (completed-cb @store)))
+          (rx/subs! (fn [_] (done))
+                    (fn [cause]
+                      (js/console.log "[error]:" cause))
+                    (fn [_]
+                      #_(js/console.log "[complete]"))))
+     (doseq [event (concat events [(end+)])]
+       (ptk/emit! store event))
+     (ptk/emit! store :the/end))))
+
+(defn run-store-async
+  "Helper version of `run-store` that automatically stops on the `send-update-indices` event"
+  ([store done events completed-cb]
+   (run-store store done events completed-cb stop-on-send-update-indices))
+  ([store done events completed-cb stop-on]
+   (run-store store done events completed-cb stop-on)))