1 files changed, 80 insertions, 0 deletions
diff --git a/python/mozbuild/mozbuild/test/action/test_node.py b/python/mozbuild/mozbuild/test/action/test_node.py
new file mode 100644
index 0000000000..f1ab5afd17
--- /dev/null
+++ b/python/mozbuild/mozbuild/test/action/test_node.py
@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+
+# Any copyright is dedicated to the Public Domain.
+# http://creativecommons.org/publicdomain/zero/1.0/
+
+import os
+import unittest
+
+import buildconfig
+import mozpack.path as mozpath
+import mozunit
+
+from mozbuild.action.node import SCRIPT_ALLOWLIST, generate
+from mozbuild.nodeutil import find_node_executable
+
+test_data_path = mozpath.abspath(mozpath.dirname(__file__))
+test_data_path = mozpath.join(test_data_path, "data", "node")
+
+
+def data(name):
+    return os.path.join(test_data_path, name)
+
+
+TEST_SCRIPT = data("node-test-script.js")
+NONEXISTENT_TEST_SCRIPT = data("non-existent-test-script.js")
+
+
+class TestNode(unittest.TestCase):
+    """
+    Tests for node.py.
+    """
+
+    def setUp(self):
+        if not buildconfig.substs.get("NODEJS"):
+            buildconfig.substs["NODEJS"] = find_node_executable()[0]
+        SCRIPT_ALLOWLIST.append(TEST_SCRIPT)
+
+    def tearDown(self):
+        try:
+            SCRIPT_ALLOWLIST.remove(TEST_SCRIPT)
+        except Exception:
+            pass
+
+    def test_generate_no_returned_deps(self):
+        deps = generate("dummy_argument", TEST_SCRIPT)
+
+        self.assertSetEqual(deps, set([]))
+
+    def test_generate_returns_passed_deps(self):
+        deps = generate("dummy_argument", TEST_SCRIPT, "a", "b")
+
+        self.assertSetEqual(deps, set(["a", "b"]))
+
+    def test_called_process_error_handled(self):
+        SCRIPT_ALLOWLIST.append(NONEXISTENT_TEST_SCRIPT)
+
+        with self.assertRaises(SystemExit) as cm:
+            generate("dummy_arg", NONEXISTENT_TEST_SCRIPT)
+
+        self.assertEqual(cm.exception.code, 1)
+        SCRIPT_ALLOWLIST.remove(NONEXISTENT_TEST_SCRIPT)
+
+    def test_nodejs_not_set(self):
+        buildconfig.substs["NODEJS"] = None
+
+        with self.assertRaises(SystemExit) as cm:
+            generate("dummy_arg", TEST_SCRIPT)
+
+        self.assertEqual(cm.exception.code, 1)
+
+    def test_generate_missing_allowlist_entry_exit_code(self):
+        SCRIPT_ALLOWLIST.remove(TEST_SCRIPT)
+        with self.assertRaises(SystemExit) as cm:
+            generate("dummy_arg", TEST_SCRIPT)
+
+        self.assertEqual(cm.exception.code, 1)
+
+
+if __name__ == "__main__":
+    mozunit.main()