Add arrow function cases

dy · Feb 19, 2024 · 44cf5f2 · 44cf5f2
1 parent 49615f3
commit 44cf5f2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/feature/arrow.js b/feature/arrow.js
@@ -3,7 +3,7 @@ import { compile, operator } from "../src/compile.js"
 import { PREC_ASSIGN, PREC_TOKEN } from "../src/const.js"
 
 // arrow functions (useful for array methods)
-binary('=>', PREC_ASSIGN)
+binary('=>', PREC_ASSIGN, true)
 operator('=>',
   (a, b) => (
     a = a[0] === '()' ? a[1] : a,

diff --git a/test/subscript.js b/test/subscript.js
@@ -413,6 +413,9 @@ test('ext: arrow', async t => {
   is(subscript('() => 1')()(), 1)
   is(subscript('(a) => a+1')()(1), 2)
   is(subscript('(a) => a+b')({ b: 2 })(1), 3)
+  is(subscript('a=>a+=1')()(1), 2)
+  is(parse('a=>a,b=>b=c=d'), [',', ['=>', 'a', 'a'], ['=>', 'b', ['=', 'b', ['=', 'c', 'd']]]])
+  is(parse('a=>b=>c=d'), ['=>', 'a', ['=>', 'b', ['=', 'c', 'd']]])
 })
 
 test('ext: justin', async t => {