sapling/eden/scm/tests/test-fb-hgext-sshaskpass.py
Xavier Deguillard fe8855e9fb hgmain: initialize fbinit
Summary:
The memcache code assumes it's initialized, let's actually do it.

The test change is required due to fbinit installing its own SIGTERM handler
that shows a backtrace. Installing our own SIGTERM handler makes it closer to
what Mercurial does already.

Reviewed By: quark-zju

Differential Revision: D19518698

fbshipit-source-id: bc8c2311e65c9c00678756abae3979ccda4453ea
2020-01-23 08:57:22 -08:00

46 lines
1.1 KiB
Python

from __future__ import absolute_import
import os
import signal
import sys
# Make sure we use sshaskpass.py in this repo, unaffected by PYTHONPATH
from edenscm.hgext import sshaskpass
from edenscm.mercurial import error
if not sys.platform.startswith("linux"):
sys.stderr.write("this test only supports linux\n")
sys.exit(80)
# stdin, stderr have to be tty to run test
pid, master = os.forkpty()
if pid:
# parent, test some I/O
os.write(master, "(input)\n")
with os.fdopen(master, "r") as f:
sys.stdout.write("pty receives: %r" % f.read())
os.waitpid(pid, 0)
sys.exit(0)
sigterm = getattr(signal, "SIGTERM", None)
if sigterm:
def catchterm(*args):
raise error.SignalInterrupt
signal.signal(sigterm, catchterm)
# child, start a ttyserver and do some I/O
ttysrvpid, sockpath = sshaskpass._startttyserver()
try:
r, w = sshaskpass._receivefds(sockpath)
with os.fdopen(r) as f:
line = f.readline()
os.write(w, "client receives: " + line)
finally:
sshaskpass._killprocess(ttysrvpid)
os.unlink(sockpath)