mirror of
https://github.com/hakril/PythonForWindows
synced 2026-06-08 14:31:45 +00:00
hakril
67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import windows.pipe
|
|
from .pfwtest import *
|
|
|
|
import time
|
|
|
|
PIPE_NAME = "PFW_Test_Pipe"
|
|
|
|
rcode_test_ipc_pipe = """
|
|
import windows; import windows.pipe
|
|
windows.pipe.send_object("{pipe}", {{'Hello': 2}})
|
|
"""
|
|
|
|
|
|
@python_injection
|
|
def test_ipc_pipe(proc32_64):
|
|
with windows.pipe.create(PIPE_NAME) as np:
|
|
proc32_64.execute_python(rcode_test_ipc_pipe.format(pipe=PIPE_NAME))
|
|
obj = np.recv()
|
|
assert obj == {'Hello': 2}
|
|
|
|
|
|
rcode_test_echo_pipe = """
|
|
import windows; import windows.pipe
|
|
|
|
with windows.pipe.create("{pipe}") as np:
|
|
np.wait_connection()
|
|
obj = np.recv()
|
|
np.send(obj)
|
|
"""
|
|
|
|
|
|
@python_injection
|
|
def test_pipe_echo_server(proc32_64):
|
|
# Verify that injection / load of Pfw in remote works with a first safe stub
|
|
proc32_64.execute_python("import windows")
|
|
t = proc32_64.execute_python_unsafe(rcode_test_echo_pipe.format(pipe=PIPE_NAME))
|
|
time.sleep(0.5)
|
|
assert not t.is_exit
|
|
obj = {'MYPID': windows.current_process.pid}
|
|
pipe = windows.pipe.connect(PIPE_NAME)
|
|
pipe.send(obj)
|
|
echoobj = pipe.recv()
|
|
assert obj == echoobj
|
|
|
|
|
|
@python_injection
|
|
def test_pipe_recv_object(proc32_64):
|
|
# Verify that injection / load of Pfw in remote works with a first safe stub
|
|
proc32_64.execute_python("import windows")
|
|
# not the good way to do the exchange (race possible)
|
|
# Just for the sake of the test
|
|
proc32_64.execute_python_unsafe(rcode_test_ipc_pipe.format(pipe=PIPE_NAME))
|
|
obj = windows.pipe.recv_object(PIPE_NAME)
|
|
assert obj == {'Hello': 2}
|
|
|
|
|
|
UNICODE_PIPE_NAME = u"Wyczyść moją rurę"
|
|
|
|
def test_pipe_unicode_name():
|
|
with windows.pipe.create(UNICODE_PIPE_NAME) as np:
|
|
# also Try the connect API with the unicode name
|
|
np2 = windows.pipe.connect(UNICODE_PIPE_NAME)
|
|
assert np.name == np2.name
|
|
assert np.name.endswith(UNICODE_PIPE_NAME)
|
|
assert np2.name.endswith(UNICODE_PIPE_NAME) |