1 files changed, 54 insertions, 0 deletions
diff --git a/tools/lint/eslint/eslint-plugin-mozilla/tests/reject-addtask-only.js b/tools/lint/eslint/eslint-plugin-mozilla/tests/reject-addtask-only.js
new file mode 100644
index 0000000000..f29726d9fc
--- /dev/null
+++ b/tools/lint/eslint/eslint-plugin-mozilla/tests/reject-addtask-only.js
@@ -0,0 +1,54 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// ------------------------------------------------------------------------------
+// Requirements
+// ------------------------------------------------------------------------------
+
+var rule = require("../lib/rules/reject-addtask-only");
+var RuleTester = require("eslint").RuleTester;
+
+const ruleTester = new RuleTester({ parserOptions: { ecmaVersion: "latest" } });
+
+// ------------------------------------------------------------------------------
+// Tests
+// ------------------------------------------------------------------------------
+
+function invalidError(output) {
+  return [
+    {
+      messageId: "addTaskNotAllowed",
+      type: "CallExpression",
+      suggestions: [{ messageId: "addTaskNotAllowedSuggestion", output }],
+    },
+  ];
+}
+
+ruleTester.run("reject-addtask-only", rule, {
+  valid: [
+    "add_task(foo())",
+    "add_task(foo()).skip()",
+    "add_task(function() {})",
+    "add_task(function() {}).skip()",
+  ],
+  invalid: [
+    {
+      code: "add_task(foo()).only()",
+      errors: invalidError("add_task(foo())"),
+    },
+    {
+      code: "add_task(foo()).only(bar())",
+      errors: invalidError("add_task(foo())"),
+    },
+    {
+      code: "add_task(function() {}).only()",
+      errors: invalidError("add_task(function() {})"),
+    },
+    {
+      code: "add_task(function() {}).only(bar())",
+      errors: invalidError("add_task(function() {})"),
+    },
+  ],
+});