mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +03:00
f42333f17c
Summary: Formats a subset of opted-in Python files in fbsource. Black formatting was applied first, which is guaranteed safe as the AST will not have changed during formatting. Pyfmt was then run, which also includes import sorting. The changes from isort were manually reviewed, and some potentially dangerous changes were reverted, and the directive was added to those files. A final run of pyfmt shows no more changes to be applied. Reviewed By: zertosh Differential Revision: D24101830 fbshipit-source-id: 0f2616873117a821dbc6cfb6d8e4f64f4420312b
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
# Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import errno
|
|
import os
|
|
import time
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from edenscm.hgext import extutil
|
|
from edenscm.mercurial import error, util, vfs
|
|
from hghave import require
|
|
|
|
|
|
locktimeout = 25
|
|
locksuccess = 24
|
|
|
|
|
|
def _exitstatus(code):
|
|
"""convert a posix exit status into the same form returned by
|
|
os.spawnv
|
|
|
|
returns None if the process was stopped instead of exiting"""
|
|
if os.WIFEXITED(code):
|
|
return os.WEXITSTATUS(code)
|
|
elif os.WIFSIGNALED(code):
|
|
return -(os.WTERMSIG(code))
|
|
|
|
|
|
class ExtutilTests(unittest.TestCase):
|
|
def testspawndetachednoblock(self):
|
|
"""spawndetached() should return without waiting for the process to
|
|
finish."""
|
|
start = time.time()
|
|
util.spawndetached(["sleep", "5"])
|
|
end = time.time()
|
|
if end - start >= 1.0:
|
|
self.fail(
|
|
"spawndetached() took took %s seconds, should have "
|
|
"returned immediately" % (end - start)
|
|
)
|
|
|
|
def testspawndetachedfailure1(self):
|
|
"""spawndetached() should throw if executing the process fails."""
|
|
try:
|
|
util.spawndetached(["no_such_program", "arg1", "arg2"])
|
|
self.fail("expected spawndetached to fail with ENOENT")
|
|
except (OSError, IOError) as ex:
|
|
self.assertEqual(ex.errno, errno.ENOENT)
|
|
|
|
def testspawndetachedfailure2(self):
|
|
"""spawndetached() should throw if executing the process fails."""
|
|
try:
|
|
util.spawndetached([os.devnull, "arg1", "arg2"])
|
|
self.fail("expected spawndetached to fail with EACCES")
|
|
except (OSError, IOError) as ex:
|
|
self.assertEqual(ex.errno, errno.EACCES)
|
|
|
|
def testflock(self):
|
|
testtmp = os.environ["TESTTMP"]
|
|
opener = vfs.vfs(testtmp)
|
|
name = "testlock"
|
|
|
|
with extutil.flock(opener.join(name), "testing a lock", timeout=0):
|
|
otherlock = self.otherprocesslock(opener, name)
|
|
self.assertEqual(
|
|
otherlock, locktimeout, "other process should not have taken the lock"
|
|
)
|
|
|
|
otherlock = self.otherprocesslock(opener, name)
|
|
self.assertEqual(
|
|
otherlock, locksuccess, "other process should have taken the lock"
|
|
)
|
|
|
|
def otherprocesslock(self, opener, name):
|
|
pid = os.fork()
|
|
if pid == 0:
|
|
try:
|
|
with extutil.flock(opener.join(name), "other process lock", timeout=0):
|
|
os._exit(locksuccess)
|
|
except error.LockHeld:
|
|
os._exit(locktimeout)
|
|
else:
|
|
time.sleep(0.1) # Avoids a crash on OSX
|
|
p, st = os.waitpid(pid, 0)
|
|
st = _exitstatus(st) # Convert back to an int
|
|
return st
|
|
|
|
|
|
if __name__ == "__main__":
|
|
silenttestrunner.main(__name__)
|