mirror of
https://github.com/facebook/sapling.git
synced 2024-10-06 14:58:03 +03:00
44343769f8
Summary: We want to rename away from "mercurial". Rather than rename the "mercurial" Python package, we opted to just collapse it into the parent "edenscm" package. This is also a step towards further organizing we want to do around the new project name. To ease the transition wrt hotfixes, we now replace "edenscm.mercurial" with "mercurial" to fix imports within base64-python extensions. Reviewed By: sggutier Differential Revision: D38943169 fbshipit-source-id: 03fa18079c51e2f7fac05d65b127095da3ab7c99
96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import errno
|
|
import os
|
|
import time
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from edenscm import error, util, vfs
|
|
from edenscm.ext import extutil
|
|
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.EPERM)
|
|
|
|
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__)
|