1 files changed, 118 insertions, 0 deletions
diff --git a/dom/bindings/test/TestInterfaceAsyncIterableSingleWithArgs.cpp b/dom/bindings/test/TestInterfaceAsyncIterableSingleWithArgs.cpp
new file mode 100644
index 0000000000..6c02829186
--- /dev/null
+++ b/dom/bindings/test/TestInterfaceAsyncIterableSingleWithArgs.cpp
@@ -0,0 +1,118 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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 "mozilla/dom/TestInterfaceAsyncIterableSingleWithArgs.h"
+#include "ScriptSettings.h"
+#include "js/Value.h"
+#include "mozilla/dom/TestInterfaceJSMaplikeSetlikeIterableBinding.h"
+#include "nsPIDOMWindow.h"
+#include "mozilla/dom/BindingUtils.h"
+#include "mozilla/dom/IterableIterator.h"
+#include "mozilla/dom/Promise-inl.h"
+
+namespace mozilla::dom {
+
+NS_IMPL_CYCLE_COLLECTION_CLASS(TestInterfaceAsyncIterableSingleWithArgs)
+
+NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN_INHERITED(
+    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
+NS_IMPL_CYCLE_COLLECTION_UNLINK_END
+
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN_INHERITED(
+    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
+
+NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN_INHERITED(
+    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
+  NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mReturnLastCalledWith)
+NS_IMPL_CYCLE_COLLECTION_TRACE_END
+
+NS_IMPL_ADDREF_INHERITED(TestInterfaceAsyncIterableSingleWithArgs,
+                         TestInterfaceAsyncIterableSingle)
+NS_IMPL_RELEASE_INHERITED(TestInterfaceAsyncIterableSingleWithArgs,
+                          TestInterfaceAsyncIterableSingle)
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(
+    TestInterfaceAsyncIterableSingleWithArgs)
+NS_INTERFACE_MAP_END_INHERITING(TestInterfaceAsyncIterableSingle)
+
+// static
+already_AddRefed<TestInterfaceAsyncIterableSingleWithArgs>
+TestInterfaceAsyncIterableSingleWithArgs::Constructor(
+    const GlobalObject& aGlobal, ErrorResult& aRv) {
+  nsCOMPtr<nsPIDOMWindowInner> window =
+      do_QueryInterface(aGlobal.GetAsSupports());
+  if (!window) {
+    aRv.Throw(NS_ERROR_FAILURE);
+    return nullptr;
+  }
+
+  RefPtr<TestInterfaceAsyncIterableSingleWithArgs> r =
+      new TestInterfaceAsyncIterableSingleWithArgs(window);
+  return r.forget();
+}
+
+JSObject* TestInterfaceAsyncIterableSingleWithArgs::WrapObject(
+    JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
+  return TestInterfaceAsyncIterableSingleWithArgs_Binding::Wrap(aCx, this,
+                                                                aGivenProto);
+}
+
+void TestInterfaceAsyncIterableSingleWithArgs::InitAsyncIteratorData(
+    IteratorData& aData, Iterator::IteratorType aType,
+    const TestInterfaceAsyncIteratorOptions& aOptions, ErrorResult& aError) {
+  aData.mMultiplier = aOptions.mMultiplier;
+  aData.mBlockingPromises = aOptions.mBlockingPromises;
+  aData.mFailNextAfter = aOptions.mFailNextAfter;
+  aData.mThrowFromNext = aOptions.mThrowFromNext;
+  if (aOptions.mThrowFromReturn.WasPassed()) {
+    aData.mThrowFromReturn = &aOptions.mThrowFromReturn.Value();
+  }
+}
+
+already_AddRefed<Promise>
+TestInterfaceAsyncIterableSingleWithArgs::GetNextIterationResult(
+    Iterator* aIterator, ErrorResult& aRv) {
+  if (aIterator->Data().mThrowFromNext) {
+    AutoJSAPI jsapi;
+    MOZ_RELEASE_ASSERT(jsapi.Init(GetParentObject()));
+    JS_ReportErrorASCII(jsapi.cx(), "Throwing from next().");
+    aRv.MightThrowJSException();
+    aRv.StealExceptionFromJSContext(jsapi.cx());
+    return nullptr;
+  }
+
+  if (aIterator->Data().mIndex == aIterator->Data().mFailNextAfter) {
+    aRv.ThrowTypeError("Failed because we of 'failNextAfter'.");
+    return nullptr;
+  }
+  return TestInterfaceAsyncIterableSingle::GetNextIterationResult(
+      aIterator, aIterator->Data(), aRv);
+}
+
+already_AddRefed<Promise>
+TestInterfaceAsyncIterableSingleWithArgs::IteratorReturn(
+    JSContext* aCx, Iterator* aIterator, JS::Handle<JS::Value> aValue,
+    ErrorResult& aRv) {
+  ++mReturnCallCount;
+
+  if (RefPtr<TestThrowingCallback> throwFromReturn =
+          aIterator->Data().mThrowFromReturn) {
+    throwFromReturn->Call(aRv, nullptr, CallbackFunction::eRethrowExceptions);
+    return nullptr;
+  }
+
+  RefPtr<Promise> promise = Promise::Create(GetParentObject()->AsGlobal(), aRv);
+  if (NS_WARN_IF(aRv.Failed())) {
+    return nullptr;
+  }
+
+  mReturnLastCalledWith = aValue;
+  promise->MaybeResolve(JS::UndefinedHandleValue);
+  return promise.forget();
+}
+
+}  // namespace mozilla::dom