Make [...entity] work in JS

tonsky · Oct 14, 2024 · 7ef5a69 · 7ef5a69
1 parent 71a4502
commit 7ef5a69
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/datascript/impl/entity.cljc b/src/datascript/impl/entity.cljc
@@ -156,6 +156,11 @@
 
 (defn entity? [x] (instance? Entity x))
 
+#?(:cljs
+  (unchecked-set (.-prototype Entity) cljs.core/ITER_SYMBOL
+     (fn []
+       (this-as this# (.entries this#)))))
+
 #?(:clj
    (defmethod print-method Entity [e, ^java.io.Writer w]
      (.write w (str e))))

diff --git a/test/js/tests.js b/test/js/tests.js
@@ -318,6 +318,8 @@ function test_entity_iterators() {
   assert_eq_set(["Ivan", ["X", "Y"]], values);
   var entries = [...e.entries()];
   assert_eq_set([["name", "Ivan"], ["aka", ["X", "Y"]]], entries);
+  var entries2 = [...e];
+  assert_eq_set(entries, entries2);
 }
 
 function test_pull() {