sapling/eden/scm/tests/test-fb-hgext-sshaskpass.py
Simon Farnsworth 5232dcf9d1 py3: make .py tests continue to run on python 2
Summary:
Use sed to convert testutil.ddot requires to hghave.

ignore-conflict-markers

for test-simplemerge.py

Reviewed By: simpkins

Differential Revision: D19658355

fbshipit-source-id: afae73eb1e43ead79514dfaf9f911f51ac25972e
2020-01-31 00:18:22 -08:00

50 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
from hghave import require
require(["py2"])
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)