1 files changed, 63 insertions, 0 deletions
diff --git a/testing/mozbase/mozfile/tests/test_which.py b/testing/mozbase/mozfile/tests/test_which.py
new file mode 100644
index 0000000000..b02f13ccdf
--- /dev/null
+++ b/testing/mozbase/mozfile/tests/test_which.py
@@ -0,0 +1,63 @@
+# Any copyright is dedicated to the Public Domain.
+# https://creativecommons.org/publicdomain/zero/1.0/
+
+import os
+import sys
+
+import mozunit
+import six
+from mozfile import which
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+
+def test_which(monkeypatch):
+    cwd = os.path.join(here, "files", "which")
+    monkeypatch.chdir(cwd)
+
+    if sys.platform == "win32":
+        if six.PY3:
+            import winreg
+        else:
+            import _winreg as winreg
+        bindir = os.path.join(cwd, "win")
+        monkeypatch.setenv("PATH", bindir)
+        monkeypatch.setattr(winreg, "QueryValue", (lambda k, sk: None))
+
+        assert which("foo.exe").lower() == os.path.join(bindir, "foo.exe").lower()
+        assert which("foo").lower() == os.path.join(bindir, "foo.exe").lower()
+        assert (
+            which("foo", exts=[".FOO", ".BAR"]).lower()
+            == os.path.join(bindir, "foo").lower()
+        )
+        assert os.environ.get("PATHEXT") != [".FOO", ".BAR"]
+        assert which("foo.txt") is None
+
+        assert which("bar").lower() == os.path.join(bindir, "bar").lower()
+        assert which("baz").lower() == os.path.join(cwd, "baz.exe").lower()
+
+        registered_dir = os.path.join(cwd, "registered")
+        quux = os.path.join(registered_dir, "quux.exe").lower()
+
+        def mock_registry(key, subkey):
+            assert subkey == (
+                r"SOFTWARE\Microsoft\Windows\CurrentVersion\App Paths\quux.exe"
+            )
+            return quux
+
+        monkeypatch.setattr(winreg, "QueryValue", mock_registry)
+        assert which("quux").lower() == quux
+        assert which("quux.exe").lower() == quux
+
+    else:
+        bindir = os.path.join(cwd, "unix")
+        monkeypatch.setenv("PATH", bindir)
+        assert which("foo") == os.path.join(bindir, "foo")
+        assert which("baz") is None
+        assert which("baz", exts=[".EXE"]) is None
+        assert "PATHEXT" not in os.environ
+        assert which("file") is None
+
+
+if __name__ == "__main__":
+    mozunit.main()