1 files changed, 101 insertions, 0 deletions
diff --git a/ipc/ipdl/test/cxx/TestAsyncReturns.cpp b/ipc/ipdl/test/cxx/TestAsyncReturns.cpp
new file mode 100644
index 0000000000..76597143c0
--- /dev/null
+++ b/ipc/ipdl/test/cxx/TestAsyncReturns.cpp
@@ -0,0 +1,101 @@
+#include "TestAsyncReturns.h"
+
+#include "IPDLUnitTests.h"  // fail etc.
+
+#include "mozilla/AbstractThread.h"
+#include "mozilla/Unused.h"
+
+namespace mozilla {
+namespace _ipdltest {
+
+static uint32_t sMagic1 = 0x105b59fb;
+static uint32_t sMagic2 = 0x09b6f5e3;
+
+//-----------------------------------------------------------------------------
+// parent
+
+TestAsyncReturnsParent::TestAsyncReturnsParent() {
+  MOZ_COUNT_CTOR(TestAsyncReturnsParent);
+}
+
+TestAsyncReturnsParent::~TestAsyncReturnsParent() {
+  MOZ_COUNT_DTOR(TestAsyncReturnsParent);
+}
+
+void TestAsyncReturnsParent::Main() {
+  SendNoReturn()->Then(
+      MessageLoop::current()->SerialEventTarget(), __func__,
+      [](bool unused) { fail("resolve handler should not be called"); },
+      [](ResponseRejectReason&& aReason) {
+        // MozPromise asserts in debug build if the
+        // handler is not called
+        if (aReason != ResponseRejectReason::ChannelClosed) {
+          fail("reject with wrong reason");
+        }
+        passed("reject handler called on channel close");
+      });
+  SendPing()->Then(
+      MessageLoop::current()->SerialEventTarget(), __func__,
+      [this](bool one) {
+        if (one) {
+          passed("take one argument");
+        } else {
+          fail("get one argument but has wrong value");
+        }
+
+        // Also try with the callback-based API.
+        SendPing(
+            [this](bool one) {
+              if (one) {
+                passed("take one argument");
+              } else {
+                fail("get one argument but has wrong value");
+              }
+              Close();
+            },
+            [](ResponseRejectReason&& aReason) { fail("sending Ping"); });
+      },
+      [](ResponseRejectReason&& aReason) { fail("sending Ping"); });
+}
+
+mozilla::ipc::IPCResult TestAsyncReturnsParent::RecvPong(
+    PongResolver&& aResolve) {
+  aResolve(std::tuple<const uint32_t&, const uint32_t&>(sMagic1, sMagic2));
+  return IPC_OK();
+}
+
+//-----------------------------------------------------------------------------
+// child
+
+TestAsyncReturnsChild::TestAsyncReturnsChild() {
+  MOZ_COUNT_CTOR(TestAsyncReturnsChild);
+}
+
+TestAsyncReturnsChild::~TestAsyncReturnsChild() {
+  MOZ_COUNT_DTOR(TestAsyncReturnsChild);
+}
+
+mozilla::ipc::IPCResult TestAsyncReturnsChild::RecvNoReturn(
+    NoReturnResolver&& aResolve) {
+  // Not resolving the promise intentionally
+  return IPC_OK();
+}
+
+mozilla::ipc::IPCResult TestAsyncReturnsChild::RecvPing(
+    PingResolver&& aResolve) {
+  SendPong()->Then(
+      MessageLoop::current()->SerialEventTarget(), __func__,
+      [aResolve](const std::tuple<uint32_t, uint32_t>& aParam) {
+        if (std::get<0>(aParam) == sMagic1 && std::get<1>(aParam) == sMagic2) {
+          passed("take two arguments");
+        } else {
+          fail("get two argument but has wrong value");
+        }
+        aResolve(true);
+      },
+      [](ResponseRejectReason&& aReason) { fail("sending Pong"); });
+  return IPC_OK();
+}
+
+}  // namespace _ipdltest
+}  // namespace mozilla