1 files changed, 86 insertions, 0 deletions
diff --git a/devtools/client/shared/redux/middleware/xpcshell/test_middleware-task-02.js b/devtools/client/shared/redux/middleware/xpcshell/test_middleware-task-02.js
new file mode 100644
index 0000000000..eaa573a8ae
--- /dev/null
+++ b/devtools/client/shared/redux/middleware/xpcshell/test_middleware-task-02.js
@@ -0,0 +1,86 @@
+/* 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/. */
+
+"use strict";
+
+/**
+ * Tests that task middleware allows dispatching generators that dispatch
+ * additional sync and async actions.
+ */
+
+const {
+  createStore,
+  applyMiddleware,
+} = require("resource://devtools/client/shared/vendor/redux.js");
+const {
+  task,
+} = require("resource://devtools/client/shared/redux/middleware/task.js");
+
+add_task(async function () {
+  const store = applyMiddleware(task)(createStore)(reducer);
+
+  store.dispatch(comboAction());
+  await waitUntilState(store, () => store.getState().length === 4);
+
+  equal(
+    store.getState()[0].type,
+    "fetchAsync-start",
+    "Async dispatched actions in a generator task are fired"
+  );
+  equal(
+    store.getState()[1].type,
+    "fetchAsync-end",
+    "Async dispatched actions in a generator task are fired"
+  );
+  equal(
+    store.getState()[2].type,
+    "fetchSync",
+    "Return values of yielded sync dispatched actions are correct"
+  );
+  equal(
+    store.getState()[3].type,
+    "fetch-done",
+    "Return values of yielded async dispatched actions are correct"
+  );
+  equal(
+    store.getState()[3].data.sync.data,
+    "sync",
+    "Return values of dispatched sync values are correct"
+  );
+  equal(
+    store.getState()[3].data.async,
+    "async",
+    "Return values of dispatched async values are correct"
+  );
+});
+
+function comboAction() {
+  return async function ({ dispatch, getState }) {
+    const data = {};
+    data.async = await dispatch(fetchAsync("async"));
+    data.sync = await dispatch(fetchSync("sync"));
+    dispatch({ type: "fetch-done", data });
+  };
+}
+
+function fetchSync(data) {
+  return { type: "fetchSync", data };
+}
+
+function fetchAsync(data) {
+  return async function ({ dispatch }) {
+    dispatch({ type: "fetchAsync-start" });
+    const val = await new Promise(resolve => resolve(data));
+    dispatch({ type: "fetchAsync-end" });
+    return val;
+  };
+}
+
+function reducer(state = [], action) {
+  info("Action called: " + action.type);
+  if (/fetch/.test(action.type)) {
+    state.push(action);
+  }
+  return [...state];
+}