1 files changed, 72 insertions, 0 deletions
diff --git a/js/src/jsapi-tests/testSlowScript.cpp b/js/src/jsapi-tests/testSlowScript.cpp
new file mode 100644
index 0000000000..154120eed1
--- /dev/null
+++ b/js/src/jsapi-tests/testSlowScript.cpp
@@ -0,0 +1,72 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "js/PropertyAndElement.h"  // JS_DefineFunction
+#include "jsapi-tests/tests.h"
+
+static bool InterruptCallback(JSContext* cx) { return false; }
+
+static unsigned sRemain;
+
+static bool RequestInterruptCallback(JSContext* cx, unsigned argc,
+                                     JS::Value* vp) {
+  JS::CallArgs args = JS::CallArgsFromVp(argc, vp);
+  if (!sRemain--) {
+    JS_RequestInterruptCallback(cx);
+  }
+  args.rval().setUndefined();
+  return true;
+}
+
+BEGIN_TEST(testSlowScript) {
+  JS_AddInterruptCallback(cx, InterruptCallback);
+  JS_DefineFunction(cx, global, "requestInterruptCallback",
+                    RequestInterruptCallback, 0, 0);
+
+  CHECK(
+      test("while (true)"
+           "  for (i in [0,0,0,0])"
+           "    requestInterruptCallback();"));
+
+  CHECK(
+      test("while (true)"
+           "  for (i in [0,0,0,0])"
+           "    for (j in [0,0,0,0])"
+           "      requestInterruptCallback();"));
+
+  CHECK(
+      test("while (true)"
+           "  for (i in [0,0,0,0])"
+           "    for (j in [0,0,0,0])"
+           "      for (k in [0,0,0,0])"
+           "        requestInterruptCallback();"));
+
+  CHECK(
+      test("function* f() { while (true) yield requestInterruptCallback() }"
+           "for (i of f()) ;"));
+
+  CHECK(
+      test("function* f() { while (true) yield 1 }"
+           "for (i of f())"
+           "  requestInterruptCallback();"));
+
+  return true;
+}
+
+bool test(const char* bytes) {
+  JS::RootedValue v(cx);
+
+  sRemain = 0;
+  CHECK(!evaluate(bytes, __FILE__, __LINE__, &v));
+  CHECK(!JS_IsExceptionPending(cx));
+  JS_ClearPendingException(cx);
+
+  sRemain = 1000;
+  CHECK(!evaluate(bytes, __FILE__, __LINE__, &v));
+  CHECK(!JS_IsExceptionPending(cx));
+  JS_ClearPendingException(cx);
+
+  return true;
+}
+END_TEST(testSlowScript)