1 files changed, 76 insertions, 0 deletions
diff --git a/addon-sdk/source/test/test-test-utils-generator.js b/addon-sdk/source/test/test-test-utils-generator.js
new file mode 100644
index 000000000..da3477900
--- /dev/null
+++ b/addon-sdk/source/test/test-test-utils-generator.js
@@ -0,0 +1,76 @@
+/* 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";
+
+// NOTE: this test is explictly testing non-normal,
+//       generator functions.
+
+
+const { defer: async } = require('sdk/lang/functional');
+const { before, after } = require('sdk/test/utils');
+const { resolve } = require("sdk/core/promise");
+
+var AFTER_RUN = 0;
+var BEFORE_RUN = 0;
+
+/*
+ * Tests are dependent on ordering, as the before and after functions
+ * are called outside of each test, and sometimes checked in the next test
+ * (like in the `after` tests)
+ */
+exports.testABeforeAsync = function (assert, done) {
+  assert.equal(BEFORE_RUN, 1, 'before function was called');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+  async(done)();
+};
+
+exports.testABeforeNameAsync = function (assert, done) {
+  assert.equal(BEFORE_RUN, 2, 'before function was called with name');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+  async(done)();
+};
+
+exports.testAfterAsync = function (assert, done) {
+  assert.equal(AFTER_RUN, 1, 'after function was called previously');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+  async(done)();
+};
+
+exports.testAfterNameAsync = function (assert, done) {
+  assert.equal(AFTER_RUN, 2, 'after function was called with name');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+  async(done)();
+};
+
+exports.testSyncABefore = function (assert) {
+  assert.equal(BEFORE_RUN, 1, 'before function was called for sync test');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+};
+
+exports.testSyncAfter = function (assert) {
+  assert.equal(AFTER_RUN, 1, 'after function was called for sync test');
+  BEFORE_RUN = 0;
+  AFTER_RUN = 0;
+};
+
+before(exports, function*(name, assert) {
+  yield resolve();
+  BEFORE_RUN = (name === 'testABeforeNameAsync') ? 2 : 1;
+  assert.pass('assert passed into before function');
+});
+
+after(exports, function*(name, assert) {
+  yield resolve();
+  // testAfterName runs after testAfter, which is where this
+  // check occurs in the assertation
+  AFTER_RUN = (name === 'testAfterAsync') ? 2 : 1;
+  assert.pass('assert passed into after function');
+});
+
+require('sdk/test').run(exports);