mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-19 08:15:31 +00:00
02a7b4ebdf
Allow-list all Python code in tree for use with the black linter, and re-format all code in-tree accordingly. To produce this patch I did all of the following: 1. Make changes to tools/lint/black.yml to remove include: stanza and update list of source extensions. 2. Run ./mach lint --linter black --fix 3. Make some ad-hoc manual updates to python/mozbuild/mozbuild/test/configure/test_configure.py -- it has some hard-coded line numbers that the reformat breaks. 4. Make some ad-hoc manual updates to `testing/marionette/client/setup.py`, `testing/marionette/harness/setup.py`, and `testing/firefox-ui/harness/setup.py`, which have hard-coded regexes that break after the reformat. 5. Add a set of exclusions to black.yml. These will be deleted in a follow-up bug (1672023). # ignore-this-changeset Differential Revision: https://phabricator.services.mozilla.com/D94045
188 lines
6.6 KiB
Python
188 lines
6.6 KiB
Python
from __future__ import absolute_import
|
|
import unittest
|
|
|
|
import os
|
|
import six
|
|
import sys
|
|
import os.path
|
|
import time
|
|
from tempfile import mkdtemp
|
|
from shutil import rmtree
|
|
import mozunit
|
|
from mozprocess import processhandler
|
|
|
|
from nsinstall import nsinstall
|
|
import nsinstall as nsinstall_module
|
|
|
|
NSINSTALL_PATH = nsinstall_module.__file__
|
|
|
|
# Run the non-ASCII tests on (a) Windows, or (b) any platform with
|
|
# sys.stdin.encoding set to UTF-8
|
|
import codecs
|
|
|
|
RUN_NON_ASCII_TESTS = sys.platform == "win32" or (
|
|
sys.stdin.encoding is not None
|
|
and codecs.lookup(sys.stdin.encoding) == codecs.lookup("utf-8")
|
|
)
|
|
|
|
|
|
class TestNsinstall(unittest.TestCase):
|
|
"""
|
|
Unit tests for nsinstall.py
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.tmpdir = mkdtemp()
|
|
|
|
def tearDown(self):
|
|
# Unicode strings means non-ASCII children can be deleted properly on
|
|
# Windows
|
|
if sys.stdin.encoding is None:
|
|
tmpdir = six.ensure_text(self.tmpdir)
|
|
else:
|
|
tmpdir = six.ensure_text(self.tmpdir, sys.stdin.encoding)
|
|
rmtree(tmpdir)
|
|
|
|
# utility methods for tests
|
|
def touch(self, file, dir=None):
|
|
if dir is None:
|
|
dir = self.tmpdir
|
|
f = os.path.join(dir, file)
|
|
open(f, "w").close()
|
|
return f
|
|
|
|
def mkdirs(self, dir):
|
|
d = os.path.join(self.tmpdir, dir)
|
|
os.makedirs(d)
|
|
return d
|
|
|
|
def test_nsinstall_D(self):
|
|
"Test nsinstall -D <dir>"
|
|
testdir = os.path.join(self.tmpdir, "test")
|
|
self.assertEqual(nsinstall(["-D", testdir]), 0)
|
|
self.assert_(os.path.isdir(testdir))
|
|
|
|
def test_nsinstall_basic(self):
|
|
"Test nsinstall <file> <dir>"
|
|
testfile = self.touch("testfile")
|
|
testdir = self.mkdirs("testdir")
|
|
self.assertEqual(nsinstall([testfile, testdir]), 0)
|
|
self.assert_(os.path.isfile(os.path.join(testdir, "testfile")))
|
|
|
|
def test_nsinstall_basic_recursive(self):
|
|
"Test nsinstall <dir> <dest dir>"
|
|
sourcedir = self.mkdirs("sourcedir")
|
|
self.touch("testfile", sourcedir)
|
|
Xfile = self.touch("Xfile", sourcedir)
|
|
copieddir = self.mkdirs("sourcedir/copieddir")
|
|
self.touch("testfile2", copieddir)
|
|
Xdir = self.mkdirs("sourcedir/Xdir")
|
|
self.touch("testfile3", Xdir)
|
|
|
|
destdir = self.mkdirs("destdir")
|
|
|
|
self.assertEqual(nsinstall([sourcedir, destdir, "-X", Xfile, "-X", Xdir]), 0)
|
|
|
|
testdir = os.path.join(destdir, "sourcedir")
|
|
self.assert_(os.path.isdir(testdir))
|
|
self.assert_(os.path.isfile(os.path.join(testdir, "testfile")))
|
|
self.assert_(not os.path.exists(os.path.join(testdir, "Xfile")))
|
|
self.assert_(os.path.isdir(os.path.join(testdir, "copieddir")))
|
|
self.assert_(os.path.isfile(os.path.join(testdir, "copieddir", "testfile2")))
|
|
self.assert_(not os.path.exists(os.path.join(testdir, "Xdir")))
|
|
|
|
def test_nsinstall_multiple(self):
|
|
"Test nsinstall <three files> <dest dir>"
|
|
testfiles = [
|
|
self.touch("testfile1"),
|
|
self.touch("testfile2"),
|
|
self.touch("testfile3"),
|
|
]
|
|
testdir = self.mkdirs("testdir")
|
|
self.assertEqual(nsinstall(testfiles + [testdir]), 0)
|
|
for f in testfiles:
|
|
self.assert_(os.path.isfile(os.path.join(testdir, os.path.basename(f))))
|
|
|
|
def test_nsinstall_dir_exists(self):
|
|
"Test nsinstall <dir> <dest dir>, where <dest dir>/<dir> already exists"
|
|
srcdir = self.mkdirs("test")
|
|
destdir = self.mkdirs("testdir/test")
|
|
self.assertEqual(nsinstall([srcdir, os.path.dirname(destdir)]), 0)
|
|
self.assert_(os.path.isdir(destdir))
|
|
|
|
def test_nsinstall_t(self):
|
|
"Test that nsinstall -t works (preserve timestamp)"
|
|
testfile = self.touch("testfile")
|
|
testdir = self.mkdirs("testdir")
|
|
# set mtime to now - 30 seconds
|
|
t = int(time.time()) - 30
|
|
os.utime(testfile, (t, t))
|
|
self.assertEqual(nsinstall(["-t", testfile, testdir]), 0)
|
|
destfile = os.path.join(testdir, "testfile")
|
|
self.assert_(os.path.isfile(destfile))
|
|
self.assertEqual(os.stat(testfile).st_mtime, os.stat(destfile).st_mtime)
|
|
|
|
@unittest.skipIf(sys.platform == "win32", "Windows doesn't have real file modes")
|
|
def test_nsinstall_m(self):
|
|
"Test that nsinstall -m works (set mode)"
|
|
testfile = self.touch("testfile")
|
|
mode = 0o600
|
|
os.chmod(testfile, mode)
|
|
testdir = self.mkdirs("testdir")
|
|
self.assertEqual(
|
|
nsinstall(["-m", "{0:04o}".format(mode), testfile, testdir]), 0
|
|
)
|
|
destfile = os.path.join(testdir, "testfile")
|
|
self.assert_(os.path.isfile(destfile))
|
|
self.assertEqual(os.stat(testfile).st_mode, os.stat(destfile).st_mode)
|
|
|
|
def test_nsinstall_d(self):
|
|
"Test that nsinstall -d works (create directories in target)"
|
|
# -d makes no sense to me, but ok!
|
|
testfile = self.touch("testfile")
|
|
testdir = self.mkdirs("testdir")
|
|
destdir = os.path.join(testdir, "subdir")
|
|
self.assertEqual(nsinstall(["-d", testfile, destdir]), 0)
|
|
self.assert_(os.path.isdir(os.path.join(destdir, "testfile")))
|
|
|
|
@unittest.skipIf(not RUN_NON_ASCII_TESTS, "Skipping non ascii tests")
|
|
def test_nsinstall_non_ascii(self):
|
|
"Test that nsinstall handles non-ASCII files"
|
|
filename = u"\u2325\u3452\u2415\u5081"
|
|
testfile = self.touch(filename)
|
|
testdir = self.mkdirs(u"\u4241\u1D04\u1414")
|
|
self.assertEqual(
|
|
nsinstall([testfile.encode("utf-8"), testdir.encode("utf-8")]), 0
|
|
)
|
|
|
|
destfile = os.path.join(testdir, filename)
|
|
self.assert_(os.path.isfile(destfile))
|
|
|
|
# Executing nsinstall.py with python 2 is not supported.
|
|
@unittest.skipIf(
|
|
not RUN_NON_ASCII_TESTS or sys.version_info[0] == 2, "Skipping non ascii tests"
|
|
)
|
|
def test_nsinstall_non_ascii_subprocess(self):
|
|
"Test that nsinstall as a subprocess handles non-ASCII files"
|
|
filename = u"\u2325\u3452\u2415\u5081"
|
|
testfile = self.touch(filename)
|
|
testdir = self.mkdirs(u"\u4241\u1D04\u1414")
|
|
# We don't use subprocess because it can't handle Unicode on
|
|
# Windows <http://bugs.python.org/issue1759845>. mozprocess calls
|
|
# CreateProcessW directly so it's perfect.
|
|
p = processhandler.ProcessHandlerMixin(
|
|
[sys.executable, NSINSTALL_PATH, testfile, testdir]
|
|
)
|
|
p.run()
|
|
rv = p.wait()
|
|
|
|
self.assertEqual(rv, 0)
|
|
destfile = os.path.join(testdir, filename)
|
|
self.assert_(os.path.isfile(destfile))
|
|
|
|
# TODO: implement -R, -l, -L and test them!
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mozunit.main()
|