Change rpc server example to use simple restore function

This commit is contained in:
Jason Paryani
2013-12-11 22:48:59 -08:00
parent a880994a6b
commit 81388f82f6
2 changed files with 39 additions and 13 deletions

View File

@@ -0,0 +1,35 @@
import capnp
import os
import socket
import gc
import sys # add examples dir to sys.path
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'examples'))
import calculator_client
import calculator_server
def test_calculator():
read, write = socket.socketpair(socket.AF_UNIX)
server = capnp.TwoPartyServer(write, calculator_server.restore)
calculator_client.main(read)
def test_calculator_gc():
def new_evaluate_impl(old_evaluate_impl):
def call(*args, **kwargs):
gc.collect()
return old_evaluate_impl(*args, **kwargs)
return call
read, write = socket.socketpair(socket.AF_UNIX)
# inject a gc.collect to the beginning of every evaluate_impl call
evaluate_impl_orig = calculator_server.evaluate_impl
calculator_server.evaluate_impl = new_evaluate_impl(evaluate_impl_orig)
server = capnp.TwoPartyServer(write, calculator_server.restore)
calculator_client.main(read)
calculator_server.evaluate_impl = evaluate_impl_orig