From 484b35cb4fcb363053512baf3e619410d8ba03cc Mon Sep 17 00:00:00 2001
From: Iaroslav Postovalov <postovalovya@gmail.com>
Date: Sun, 20 Dec 2020 18:34:35 +0700
Subject: [PATCH] Fix failing tests

---
 .../kmath/estree/internal/JSBuilder.kt        | 23 ++++++++++++-------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/kmath-ast/src/jsMain/kotlin/kscience/kmath/estree/internal/JSBuilder.kt b/kmath-ast/src/jsMain/kotlin/kscience/kmath/estree/internal/JSBuilder.kt
index b6b58179b..c77f0c9ef 100644
--- a/kmath-ast/src/jsMain/kotlin/kscience/kmath/estree/internal/JSBuilder.kt
+++ b/kmath-ast/src/jsMain/kotlin/kscience/kmath/estree/internal/JSBuilder.kt
@@ -56,18 +56,25 @@ internal class JSBuilder<T>(val bodyCallback: JSBuilder<T>.() -> BaseExpression)
         }
     }
 
-    fun variable(name: String): BaseExpression {
-        return MemberExpression(
-            computed = true,
-            optional = false,
-            `object` = Identifier("arguments"),
-            property = SimpleLiteral(name),
-        )
-    }
+    fun variable(name: String): BaseExpression = call(getOrFail, Identifier("arguments"), SimpleLiteral(name))
 
     fun call(function: Function<T>, vararg args: BaseExpression): BaseExpression = SimpleCallExpression(
         optional = false,
         callee = constant(function),
         *args,
     )
+
+    private companion object {
+        @Suppress("UNUSED_VARIABLE")
+        val getOrFail: (`object`: dynamic, key: String) -> dynamic = { `object`, key ->
+            val k = key
+            val o = `object`
+
+            if (!(js("k in o") as Boolean)) {
+                throw NoSuchElementException()
+            }
+
+            js("o[k]")
+        }
+    }
 }