760 lines
25 KiB
Python
760 lines
25 KiB
Python
from __future__ import generators
|
|
import os, sys, time, signal
|
|
import py
|
|
from py.__.execnet.gateway_base import Message, Channel, ChannelFactory
|
|
from py.__.execnet.gateway_base import ExecnetAPI, queue, Popen2IO
|
|
from py.__.execnet import gateway_base, gateway
|
|
|
|
from py.__.execnet.gateway import startup_modules, getsource
|
|
pytest_plugins = "pytester"
|
|
|
|
TESTTIMEOUT = 10.0 # seconds
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
if 'pythonpath' in metafunc.funcargnames:
|
|
for name in 'python2.4', 'python2.5', 'python2.6', 'python3.1':
|
|
metafunc.addcall(id=name, param=name)
|
|
|
|
def pytest_funcarg__pythonpath(request):
|
|
name = request.param
|
|
executable = py.path.local.sysfind(name)
|
|
if executable is None:
|
|
py.test.skip("no %s found" % (name,))
|
|
return executable
|
|
|
|
def test_io_message(pythonpath, tmpdir):
|
|
check = tmpdir.join("check.py")
|
|
check.write(py.code.Source(gateway_base, """
|
|
try:
|
|
from io import BytesIO
|
|
except ImportError:
|
|
from StringIO import StringIO as BytesIO
|
|
import tempfile
|
|
temp_out = BytesIO()
|
|
temp_in = BytesIO()
|
|
io = Popen2IO(temp_out, temp_in)
|
|
for i, msg_cls in Message._types.items():
|
|
print ("checking %s %s" %(i, msg_cls))
|
|
for data in "hello", "hello".encode('ascii'):
|
|
msg1 = msg_cls(i, data)
|
|
msg1.writeto(io)
|
|
x = io.outfile.getvalue()
|
|
io.outfile.truncate(0)
|
|
io.outfile.seek(0)
|
|
io.infile.seek(0)
|
|
io.infile.write(x)
|
|
io.infile.seek(0)
|
|
msg2 = Message.readfrom(io)
|
|
assert msg1.channelid == msg2.channelid, (msg1, msg2)
|
|
assert msg1.data == msg2.data
|
|
print ("all passed")
|
|
"""))
|
|
#out = py.process.cmdexec("%s %s" %(executable,check))
|
|
out = pythonpath.sysexec(check)
|
|
print (out)
|
|
assert "all passed" in out
|
|
|
|
def test_popen_io(pythonpath, tmpdir):
|
|
check = tmpdir.join("check.py")
|
|
check.write(py.code.Source(gateway_base, """
|
|
do_exec(Popen2IO.server_stmt, globals())
|
|
io.write("hello".encode('ascii'))
|
|
s = io.read(1)
|
|
assert s == "x".encode('ascii')
|
|
"""))
|
|
from subprocess import Popen, PIPE
|
|
args = [str(pythonpath), str(check)]
|
|
proc = Popen(args, stdin=PIPE, stdout=PIPE, stderr=PIPE)
|
|
proc.stdin.write("x".encode('ascii'))
|
|
stdout, stderr = proc.communicate()
|
|
print (stderr)
|
|
ret = proc.wait()
|
|
assert "hello".encode('ascii') in stdout
|
|
|
|
def test_rinfo_source(pythonpath, tmpdir):
|
|
check = tmpdir.join("check.py")
|
|
check.write(py.code.Source("""
|
|
class Channel:
|
|
def send(self, data):
|
|
assert eval(repr(data), {}) == data
|
|
channel = Channel()
|
|
""", gateway.rinfo_source, """
|
|
print ('all passed')
|
|
"""))
|
|
out = pythonpath.sysexec(check)
|
|
print (out)
|
|
assert "all passed" in out
|
|
|
|
def test_geterrortext(pythonpath, tmpdir):
|
|
check = tmpdir.join("check.py")
|
|
check.write(py.code.Source(gateway_base, """
|
|
class Arg:
|
|
pass
|
|
errortext = geterrortext((Arg, "1", 4))
|
|
assert "Arg" in errortext
|
|
import sys
|
|
try:
|
|
raise ValueError("17")
|
|
except ValueError:
|
|
excinfo = sys.exc_info()
|
|
s = geterrortext(excinfo)
|
|
assert "17" in s
|
|
print ("all passed")
|
|
"""))
|
|
out = pythonpath.sysexec(check)
|
|
print (out)
|
|
assert "all passed" in out
|
|
|
|
class TestExecnetEvents:
|
|
def test_popengateway(self, _pytest):
|
|
rec = _pytest.gethookrecorder(ExecnetAPI)
|
|
gw = py.execnet.PopenGateway()
|
|
call = rec.popcall("pyexecnet_gateway_init")
|
|
assert call.gateway == gw
|
|
gw.exit()
|
|
call = rec.popcall("pyexecnet_gateway_exit")
|
|
assert call.gateway == gw
|
|
|
|
|
|
def test_getsource_import_modules():
|
|
for dottedname in startup_modules:
|
|
yield getsource, dottedname
|
|
|
|
def test_getsource_no_colision():
|
|
seen = {}
|
|
for dottedname in startup_modules:
|
|
mod = __import__(dottedname, None, None, ['__doc__'])
|
|
for name, value in vars(mod).items():
|
|
if py.std.inspect.isclass(value):
|
|
if name in seen:
|
|
olddottedname, oldval = seen[name]
|
|
if oldval is not value:
|
|
py.test.fail("duplicate class %r in %s and %s" %
|
|
(name, dottedname, olddottedname))
|
|
seen[name] = (dottedname, value)
|
|
|
|
def test_stdouterrin_setnull():
|
|
cap = py.io.StdCaptureFD()
|
|
from py.__.execnet.gateway import stdouterrin_setnull
|
|
stdouterrin_setnull()
|
|
import os
|
|
os.write(1, "hello".encode('ascii'))
|
|
if os.name == "nt":
|
|
os.write(2, "world")
|
|
os.read(0, 1)
|
|
out, err = cap.reset()
|
|
assert not out
|
|
assert not err
|
|
|
|
|
|
class TestMessage:
|
|
def test_wire_protocol(self):
|
|
for cls in Message._types.values():
|
|
one = py.io.BytesIO()
|
|
data = '23'.encode('ascii')
|
|
cls(42, data).writeto(one)
|
|
two = py.io.BytesIO(one.getvalue())
|
|
msg = Message.readfrom(two)
|
|
assert isinstance(msg, cls)
|
|
assert msg.channelid == 42
|
|
assert msg.data == data
|
|
assert isinstance(repr(msg), str)
|
|
# == "<Message.%s channelid=42 '23'>" %(msg.__class__.__name__, )
|
|
|
|
class TestPureChannel:
|
|
def setup_method(self, method):
|
|
self.fac = ChannelFactory(None)
|
|
|
|
def test_factory_create(self):
|
|
chan1 = self.fac.new()
|
|
assert chan1.id == 1
|
|
chan2 = self.fac.new()
|
|
assert chan2.id == 3
|
|
|
|
def test_factory_getitem(self):
|
|
chan1 = self.fac.new()
|
|
assert self.fac._channels[chan1.id] == chan1
|
|
chan2 = self.fac.new()
|
|
assert self.fac._channels[chan2.id] == chan2
|
|
|
|
def test_channel_timeouterror(self):
|
|
channel = self.fac.new()
|
|
py.test.raises(IOError, channel.waitclose, timeout=0.01)
|
|
|
|
def test_channel_makefile_incompatmode(self):
|
|
channel = self.fac.new()
|
|
py.test.raises(ValueError, 'channel.makefile("rw")')
|
|
|
|
|
|
class PopenGatewayTestSetup:
|
|
def setup_class(cls):
|
|
cls.gw = py.execnet.PopenGateway()
|
|
|
|
#def teardown_class(cls):
|
|
# cls.gw.exit()
|
|
|
|
class BasicRemoteExecution:
|
|
def test_correct_setup(self):
|
|
assert self.gw._receiverthread.isAlive()
|
|
|
|
def test_repr_doesnt_crash(self):
|
|
assert isinstance(repr(self.gw), str)
|
|
|
|
def test_attribute__name__(self):
|
|
channel = self.gw.remote_exec("channel.send(__name__)")
|
|
name = channel.receive()
|
|
assert name == "__channelexec__"
|
|
|
|
def test_correct_setup_no_py(self):
|
|
channel = self.gw.remote_exec("""
|
|
import sys
|
|
channel.send(list(sys.modules))
|
|
""")
|
|
remotemodules = channel.receive()
|
|
assert 'py' not in remotemodules, (
|
|
"py should not be imported on remote side")
|
|
|
|
def test_remote_exec_waitclose(self):
|
|
channel = self.gw.remote_exec('pass')
|
|
channel.waitclose(TESTTIMEOUT)
|
|
|
|
def test_remote_exec_waitclose_2(self):
|
|
channel = self.gw.remote_exec('def gccycle(): pass')
|
|
channel.waitclose(TESTTIMEOUT)
|
|
|
|
def test_remote_exec_waitclose_noarg(self):
|
|
channel = self.gw.remote_exec('pass')
|
|
channel.waitclose()
|
|
|
|
def test_remote_exec_error_after_close(self):
|
|
channel = self.gw.remote_exec('pass')
|
|
channel.waitclose(TESTTIMEOUT)
|
|
py.test.raises(IOError, channel.send, 0)
|
|
|
|
def test_remote_exec_channel_anonymous(self):
|
|
channel = self.gw.remote_exec('''
|
|
obj = channel.receive()
|
|
channel.send(obj)
|
|
''')
|
|
channel.send(42)
|
|
result = channel.receive()
|
|
assert result == 42
|
|
|
|
def test_channel_close_and_then_receive_error(self):
|
|
channel = self.gw.remote_exec('raise ValueError')
|
|
py.test.raises(channel.RemoteError, channel.receive)
|
|
|
|
def test_channel_finish_and_then_EOFError(self):
|
|
channel = self.gw.remote_exec('channel.send(42)')
|
|
x = channel.receive()
|
|
assert x == 42
|
|
py.test.raises(EOFError, channel.receive)
|
|
py.test.raises(EOFError, channel.receive)
|
|
py.test.raises(EOFError, channel.receive)
|
|
|
|
def test_channel_close_and_then_receive_error_multiple(self):
|
|
channel = self.gw.remote_exec('channel.send(42) ; raise ValueError')
|
|
x = channel.receive()
|
|
assert x == 42
|
|
py.test.raises(channel.RemoteError, channel.receive)
|
|
|
|
def test_channel__local_close(self):
|
|
channel = self.gw._channelfactory.new()
|
|
self.gw._channelfactory._local_close(channel.id)
|
|
channel.waitclose(0.1)
|
|
|
|
def test_channel__local_close_error(self):
|
|
channel = self.gw._channelfactory.new()
|
|
self.gw._channelfactory._local_close(channel.id,
|
|
channel.RemoteError("error"))
|
|
py.test.raises(channel.RemoteError, channel.waitclose, 0.01)
|
|
|
|
def test_channel_error_reporting(self):
|
|
channel = self.gw.remote_exec('def foo():\n return foobar()\nfoo()\n')
|
|
try:
|
|
channel.receive()
|
|
except channel.RemoteError:
|
|
e = sys.exc_info()[1]
|
|
assert str(e).startswith('Traceback (most recent call last):')
|
|
assert str(e).find('NameError: global name \'foobar\' '
|
|
'is not defined') > -1
|
|
else:
|
|
py.test.fail('No exception raised')
|
|
|
|
def test_channel_syntax_error(self):
|
|
# missing colon
|
|
channel = self.gw.remote_exec('def foo()\n return 1\nfoo()\n')
|
|
try:
|
|
channel.receive()
|
|
except channel.RemoteError:
|
|
e = sys.exc_info()[1]
|
|
assert str(e).startswith('Traceback (most recent call last):')
|
|
assert str(e).find('SyntaxError') > -1
|
|
|
|
def test_channel_iter(self):
|
|
channel = self.gw.remote_exec("""
|
|
for x in range(3):
|
|
channel.send(x)
|
|
""")
|
|
l = list(channel)
|
|
assert l == [0, 1, 2]
|
|
|
|
def test_channel_passing_over_channel(self):
|
|
channel = self.gw.remote_exec('''
|
|
c = channel.gateway.newchannel()
|
|
channel.send(c)
|
|
c.send(42)
|
|
''')
|
|
c = channel.receive()
|
|
x = c.receive()
|
|
assert x == 42
|
|
|
|
# check that the both sides previous channels are really gone
|
|
channel.waitclose(TESTTIMEOUT)
|
|
#assert c.id not in self.gw._channelfactory
|
|
newchan = self.gw.remote_exec('''
|
|
assert %d not in channel.gateway._channelfactory._channels
|
|
''' % (channel.id))
|
|
newchan.waitclose(TESTTIMEOUT)
|
|
assert channel.id not in self.gw._channelfactory._channels
|
|
|
|
def test_channel_receiver_callback(self):
|
|
l = []
|
|
#channel = self.gw.newchannel(receiver=l.append)
|
|
channel = self.gw.remote_exec(source='''
|
|
channel.send(42)
|
|
channel.send(13)
|
|
channel.send(channel.gateway.newchannel())
|
|
''')
|
|
channel.setcallback(callback=l.append)
|
|
py.test.raises(IOError, channel.receive)
|
|
channel.waitclose(TESTTIMEOUT)
|
|
assert len(l) == 3
|
|
assert l[:2] == [42,13]
|
|
assert isinstance(l[2], channel.__class__)
|
|
|
|
def test_channel_callback_after_receive(self):
|
|
l = []
|
|
channel = self.gw.remote_exec(source='''
|
|
channel.send(42)
|
|
channel.send(13)
|
|
channel.send(channel.gateway.newchannel())
|
|
''')
|
|
x = channel.receive()
|
|
assert x == 42
|
|
channel.setcallback(callback=l.append)
|
|
py.test.raises(IOError, channel.receive)
|
|
channel.waitclose(TESTTIMEOUT)
|
|
assert len(l) == 2
|
|
assert l[0] == 13
|
|
assert isinstance(l[1], channel.__class__)
|
|
|
|
def test_waiting_for_callbacks(self):
|
|
l = []
|
|
def callback(msg):
|
|
import time; time.sleep(0.2)
|
|
l.append(msg)
|
|
channel = self.gw.remote_exec(source='''
|
|
channel.send(42)
|
|
''')
|
|
channel.setcallback(callback)
|
|
channel.waitclose(TESTTIMEOUT)
|
|
assert l == [42]
|
|
|
|
def test_channel_callback_stays_active(self):
|
|
self.check_channel_callback_stays_active(earlyfree=True)
|
|
|
|
def check_channel_callback_stays_active(self, earlyfree=True):
|
|
# with 'earlyfree==True', this tests the "sendonly" channel state.
|
|
l = []
|
|
channel = self.gw.remote_exec(source='''
|
|
try:
|
|
import thread
|
|
except ImportError:
|
|
import _thread as thread
|
|
import time
|
|
def producer(subchannel):
|
|
for i in range(5):
|
|
time.sleep(0.15)
|
|
subchannel.send(i*100)
|
|
channel2 = channel.receive()
|
|
thread.start_new_thread(producer, (channel2,))
|
|
del channel2
|
|
''')
|
|
subchannel = self.gw.newchannel()
|
|
subchannel.setcallback(l.append)
|
|
channel.send(subchannel)
|
|
if earlyfree:
|
|
subchannel = None
|
|
counter = 100
|
|
while len(l) < 5:
|
|
if subchannel and subchannel.isclosed():
|
|
break
|
|
counter -= 1
|
|
print(counter)
|
|
if not counter:
|
|
py.test.fail("timed out waiting for the answer[%d]" % len(l))
|
|
time.sleep(0.04) # busy-wait
|
|
assert l == [0, 100, 200, 300, 400]
|
|
return subchannel
|
|
|
|
def test_channel_callback_remote_freed(self):
|
|
channel = self.check_channel_callback_stays_active(earlyfree=False)
|
|
channel.waitclose(TESTTIMEOUT) # freed automatically at the end of producer()
|
|
|
|
def test_channel_endmarker_callback(self):
|
|
l = []
|
|
channel = self.gw.remote_exec(source='''
|
|
channel.send(42)
|
|
channel.send(13)
|
|
channel.send(channel.gateway.newchannel())
|
|
''')
|
|
channel.setcallback(l.append, 999)
|
|
py.test.raises(IOError, channel.receive)
|
|
channel.waitclose(TESTTIMEOUT)
|
|
assert len(l) == 4
|
|
assert l[:2] == [42,13]
|
|
assert isinstance(l[2], channel.__class__)
|
|
assert l[3] == 999
|
|
|
|
def test_channel_endmarker_callback_error(self):
|
|
q = queue.Queue()
|
|
channel = self.gw.remote_exec(source='''
|
|
raise ValueError()
|
|
''')
|
|
channel.setcallback(q.put, endmarker=999)
|
|
val = q.get(TESTTIMEOUT)
|
|
assert val == 999
|
|
err = channel._getremoteerror()
|
|
assert err
|
|
assert str(err).find("ValueError") != -1
|
|
|
|
@py.test.mark.xfail
|
|
def test_remote_redirect_stdout(self):
|
|
out = py.io.TextIO()
|
|
handle = self.gw._remote_redirect(stdout=out)
|
|
c = self.gw.remote_exec("print 42")
|
|
c.waitclose(TESTTIMEOUT)
|
|
handle.close()
|
|
s = out.getvalue()
|
|
assert s.strip() == "42"
|
|
|
|
@py.test.mark.xfail
|
|
def test_remote_exec_redirect_multi(self):
|
|
num = 3
|
|
l = [[] for x in range(num)]
|
|
channels = [self.gw.remote_exec("print %d" % i,
|
|
stdout=l[i].append)
|
|
for i in range(num)]
|
|
for x in channels:
|
|
x.waitclose(TESTTIMEOUT)
|
|
|
|
for i in range(num):
|
|
subl = l[i]
|
|
assert subl
|
|
s = subl[0]
|
|
assert s.strip() == str(i)
|
|
|
|
def test_channel_file_write(self):
|
|
channel = self.gw.remote_exec("""
|
|
f = channel.makefile()
|
|
f.write("hello world\\n")
|
|
f.close()
|
|
channel.send(42)
|
|
""")
|
|
first = channel.receive()
|
|
assert first.strip() == 'hello world'
|
|
second = channel.receive()
|
|
assert second == 42
|
|
|
|
def test_channel_file_write_error(self):
|
|
channel = self.gw.remote_exec("pass")
|
|
f = channel.makefile()
|
|
channel.waitclose(TESTTIMEOUT)
|
|
py.test.raises(IOError, f.write, 'hello')
|
|
|
|
def test_channel_file_proxyclose(self):
|
|
channel = self.gw.remote_exec("""
|
|
f = channel.makefile(proxyclose=True)
|
|
f.write("hello world")
|
|
f.close()
|
|
channel.send(42)
|
|
""")
|
|
first = channel.receive()
|
|
assert first.strip() == 'hello world'
|
|
py.test.raises(EOFError, channel.receive)
|
|
|
|
def test_channel_file_read(self):
|
|
channel = self.gw.remote_exec("""
|
|
f = channel.makefile(mode='r')
|
|
s = f.read(2)
|
|
channel.send(s)
|
|
s = f.read(5)
|
|
channel.send(s)
|
|
""")
|
|
channel.send("xyabcde")
|
|
s1 = channel.receive()
|
|
s2 = channel.receive()
|
|
assert s1 == "xy"
|
|
assert s2 == "abcde"
|
|
|
|
def test_channel_file_read_empty(self):
|
|
channel = self.gw.remote_exec("pass")
|
|
f = channel.makefile(mode="r")
|
|
s = f.read(3)
|
|
assert s == ""
|
|
s = f.read(5)
|
|
assert s == ""
|
|
|
|
def test_channel_file_readline_remote(self):
|
|
channel = self.gw.remote_exec("""
|
|
channel.send('123\\n45')
|
|
""")
|
|
channel.waitclose(TESTTIMEOUT)
|
|
f = channel.makefile(mode="r")
|
|
s = f.readline()
|
|
assert s == "123\n"
|
|
s = f.readline()
|
|
assert s == "45"
|
|
|
|
def test_channel_makefile_incompatmode(self):
|
|
channel = self.gw.newchannel()
|
|
py.test.raises(ValueError, 'channel.makefile("rw")')
|
|
|
|
def test_confusion_from_os_write_stdout(self):
|
|
channel = self.gw.remote_exec("""
|
|
import os
|
|
os.write(1, 'confusion!'.encode('ascii'))
|
|
channel.send(channel.receive() * 6)
|
|
channel.send(channel.receive() * 6)
|
|
""")
|
|
channel.send(3)
|
|
res = channel.receive()
|
|
assert res == 18
|
|
channel.send(7)
|
|
res = channel.receive()
|
|
assert res == 42
|
|
|
|
def test_confusion_from_os_write_stderr(self):
|
|
channel = self.gw.remote_exec("""
|
|
import os
|
|
os.write(2, 'test'.encode('ascii'))
|
|
channel.send(channel.receive() * 6)
|
|
channel.send(channel.receive() * 6)
|
|
""")
|
|
channel.send(3)
|
|
res = channel.receive()
|
|
assert res == 18
|
|
channel.send(7)
|
|
res = channel.receive()
|
|
assert res == 42
|
|
|
|
def test__rinfo(self):
|
|
rinfo = self.gw._rinfo()
|
|
assert rinfo.executable
|
|
assert rinfo.cwd
|
|
assert rinfo.version_info
|
|
s = repr(rinfo)
|
|
old = self.gw.remote_exec("""
|
|
import os.path
|
|
cwd = os.getcwd()
|
|
channel.send(os.path.basename(cwd))
|
|
os.chdir('..')
|
|
""").receive()
|
|
try:
|
|
rinfo2 = self.gw._rinfo()
|
|
assert rinfo2.cwd == rinfo.cwd
|
|
rinfo3 = self.gw._rinfo(update=True)
|
|
assert rinfo3.cwd != rinfo2.cwd
|
|
finally:
|
|
self.gw._cache_rinfo = rinfo
|
|
self.gw.remote_exec("import os ; os.chdir(%r)" % old).waitclose()
|
|
|
|
class BasicCmdbasedRemoteExecution(BasicRemoteExecution):
|
|
def test_cmdattr(self):
|
|
assert hasattr(self.gw, '_cmd')
|
|
|
|
#class TestBlockingIssues:
|
|
# def test_join_blocked_execution_gateway(self):
|
|
# gateway = py.execnet.PopenGateway()
|
|
# channel = gateway.remote_exec("""
|
|
# time.sleep(5.0)
|
|
# """)
|
|
# def doit():
|
|
# gateway.exit()
|
|
# gateway.join(joinexec=True)
|
|
# return 17
|
|
#
|
|
# pool = py._thread.WorkerPool()
|
|
# reply = pool.dispatch(doit)
|
|
# x = reply.get(timeout=1.0)
|
|
# assert x == 17
|
|
|
|
class TestPopenGateway(PopenGatewayTestSetup, BasicRemoteExecution):
|
|
def test_rinfo_popen(self):
|
|
rinfo = self.gw._rinfo()
|
|
assert rinfo.executable == py.std.sys.executable
|
|
assert rinfo.cwd == py.std.os.getcwd()
|
|
assert rinfo.version_info == py.std.sys.version_info
|
|
|
|
def test_chdir_separation(self, tmpdir):
|
|
old = tmpdir.chdir()
|
|
try:
|
|
gw = py.execnet.PopenGateway()
|
|
finally:
|
|
waschangedir = old.chdir()
|
|
c = gw.remote_exec("import os ; channel.send(os.getcwd())")
|
|
x = c.receive()
|
|
assert x == str(waschangedir)
|
|
|
|
def test_many_popen(self):
|
|
num = 4
|
|
l = []
|
|
for i in range(num):
|
|
l.append(py.execnet.PopenGateway())
|
|
channels = []
|
|
for gw in l:
|
|
channel = gw.remote_exec("""channel.send(42)""")
|
|
channels.append(channel)
|
|
## try:
|
|
## while channels:
|
|
## channel = channels.pop()
|
|
## try:
|
|
## ret = channel.receive()
|
|
## assert ret == 42
|
|
## finally:
|
|
## channel.gateway.exit()
|
|
## finally:
|
|
## for x in channels:
|
|
## x.gateway.exit()
|
|
while channels:
|
|
channel = channels.pop()
|
|
ret = channel.receive()
|
|
assert ret == 42
|
|
|
|
@py.test.mark.xfail # "fix needed: dying remote process does not cause waitclose() to fail"
|
|
def test_waitclose_on_remote_killed(self):
|
|
gw = py.execnet.PopenGateway()
|
|
channel = gw.remote_exec("""
|
|
import os
|
|
import time
|
|
channel.send(os.getpid())
|
|
while 1:
|
|
channel.send("#" * 100)
|
|
""")
|
|
remotepid = channel.receive()
|
|
py.process.kill(remotepid)
|
|
py.test.raises(channel.RemoteError, "channel.waitclose(TESTTIMEOUT)")
|
|
py.test.raises(EOFError, channel.send, None)
|
|
py.test.raises(EOFError, channel.receive)
|
|
|
|
def test_endmarker_delivery_on_remote_killterm():
|
|
if not hasattr(py.std.os, 'kill'):
|
|
py.test.skip("no os.kill()")
|
|
gw = py.execnet.PopenGateway()
|
|
try:
|
|
q = queue.Queue()
|
|
channel = gw.remote_exec(source='''
|
|
import os
|
|
os.kill(os.getpid(), 15)
|
|
''')
|
|
channel.setcallback(q.put, endmarker=999)
|
|
val = q.get(TESTTIMEOUT)
|
|
assert val == 999
|
|
err = channel._getremoteerror()
|
|
finally:
|
|
gw.exit()
|
|
py.test.skip("provide information on causes/signals "
|
|
"of dying remote gateways")
|
|
|
|
|
|
class SocketGatewaySetup:
|
|
def setup_class(cls):
|
|
# open a gateway to a fresh child process
|
|
cls.proxygw = py.execnet.PopenGateway()
|
|
cls.gw = py.execnet.SocketGateway.new_remote(cls.proxygw,
|
|
("127.0.0.1", 0)
|
|
)
|
|
|
|
## def teardown_class(cls):
|
|
## cls.gw.exit()
|
|
## cls.proxygw.exit()
|
|
|
|
class TestSocketGateway(SocketGatewaySetup, BasicRemoteExecution):
|
|
pass
|
|
|
|
class TestSshGateway(BasicRemoteExecution):
|
|
def setup_class(cls):
|
|
from py.__.conftest import getspecssh
|
|
cls.sshhost = getspecssh().ssh
|
|
cls.gw = py.execnet.SshGateway(cls.sshhost)
|
|
|
|
def test_sshconfig_functional(self, tmpdir):
|
|
ssh_config = tmpdir.join("ssh_config")
|
|
ssh_config.write(
|
|
"Host alias123\n"
|
|
" HostName %s\n" % self.sshhost)
|
|
gw = py.execnet.SshGateway("alias123", ssh_config=ssh_config)
|
|
assert gw._cmd.find("-F") != -1
|
|
assert gw._cmd.find(str(ssh_config)) != -1
|
|
pid = gw.remote_exec("import os ; channel.send(os.getpid())").receive()
|
|
gw.exit()
|
|
|
|
def test_sshaddress(self):
|
|
assert self.gw.remoteaddress == self.sshhost
|
|
|
|
@py.test.mark.xfail # XXX ssh-gateway error handling
|
|
def test_connexion_failes_on_non_existing_hosts(self):
|
|
py.test.raises(IOError,
|
|
"py.execnet.SshGateway('nowhere.codespeak.net')")
|
|
|
|
@py.test.mark.xfail # "XXX ssh-gateway error handling"
|
|
def test_deprecated_identity(self):
|
|
py.test.deprecated_call(
|
|
py.test.raises, IOError,
|
|
py.execnet.SshGateway,
|
|
'nowhere.codespeak.net', identity='qwe')
|
|
|
|
|
|
def test_threads():
|
|
gw = py.execnet.PopenGateway()
|
|
gw.remote_init_threads(3)
|
|
c1 = gw.remote_exec("channel.send(channel.receive())")
|
|
c2 = gw.remote_exec("channel.send(channel.receive())")
|
|
c2.send(1)
|
|
res = c2.receive()
|
|
assert res == 1
|
|
c1.send(42)
|
|
res = c1.receive()
|
|
assert res == 42
|
|
gw.exit()
|
|
|
|
def test_threads_twice():
|
|
gw = py.execnet.PopenGateway()
|
|
gw.remote_init_threads(3)
|
|
py.test.raises(IOError, gw.remote_init_threads, 3)
|
|
gw.exit()
|
|
|
|
|
|
def test_nodebug():
|
|
from py.__.execnet import gateway_base
|
|
assert not gateway_base.debug
|
|
|
|
def test_channel_endmarker_remote_killterm():
|
|
gw = py.execnet.PopenGateway()
|
|
try:
|
|
q = queue.Queue()
|
|
channel = gw.remote_exec('''
|
|
import os
|
|
os.kill(os.getpid(), 15)
|
|
''')
|
|
channel.setcallback(q.put, endmarker=999)
|
|
val = q.get(TESTTIMEOUT)
|
|
assert val == 999
|
|
err = channel._getremoteerror()
|
|
finally:
|
|
gw.exit()
|
|
py.test.skip("provide information on causes/signals "
|
|
"of dying remote gateways")
|
|
|