def setUp(self) -> None:
self.init_job()
self.rpc = get_debug_test_context(True)
self.rsc_guest = get_debug_rs_context(self.rpc, self.job_id,
guest_options)
self.rsc_host = get_debug_rs_context(self.rpc, self.job_id,
host_options)
self.rptc = RptContext(self.rpc)
def setUpClass(cls) -> None:
if is_debug:
cls.ctx = get_debug_test_context()
else:
if is_standalone:
cls.ctx = get_standalone_context()
else:
cls.ctx = get_cluster_context()
set_default_option('include_key', False)
set_default_option('total_partitions', total_partitions)
def get_debug_test_context(is_standalone=False,
manager_port=4670,
egg_port=20001,
transfer_port=20002,
session_id='testing',
role='host',
props_file=default_props_file):
rp_context = rpta.get_debug_test_context(is_standalone=is_standalone,
manager_port=manager_port,
egg_port=egg_port,
transfer_port=transfer_port,
session_id=session_id)
rs_context = RollSiteContext(roll_site_session_id, rp_ctx=rp_context,
options=get_option(role, props_file))
return rs_context
def test_get(self):
rp_context = get_debug_test_context(is_standalone, manager_port_host,
egg_port_host, transfer_port_host,
'testing')
context = RollSiteContext("atest",
self_role='host',
self_partyId=host_partyId,
rs_ip=host_ip,
rs_port=host_rs_port,
rp_ctx=rp_context)
_tag = "Hello2"
rs = context.load(name="RsaIntersectTransferVariable.rsa_pubkey",
tag="{}".format(_tag))
futures = rs.pull(get_parties)
for future in futures:
obj = future.result()
if isinstance(obj, RollPair):
key = 'hello'
print("obj:", obj.get(key))
else:
print("obj:", obj)
def test_get_rollpair(self):
rp_context = get_debug_test_context(is_standalone, manager_port_host,
egg_port_host, transfer_port_host,
'testing')
context = RollSiteContext("atest2",
self_role='host',
self_partyId=host_partyId,
rs_ip=host_ip,
rs_port=host_rs_port,
rp_ctx=rp_context)
_tag = "roll_pair_tag"
rs = context.load(name="roll_pair_name.table", tag="{}".format(_tag))
futures = rs.pull(get_parties)
for future in futures:
obj = future.result()
if isinstance(obj, RollPair):
key = "k1"
print("obj:", obj.get(key))
else:
print("obj:", obj)
def test_remote(self):
rp_context = get_debug_test_context(is_standalone, manager_port_guest,
egg_port_guest,
transfer_port_guest,
'testing_guest')
context = RollSiteContext("atest",
self_role='guest',
self_partyId=guest_partyId,
rs_ip=guest_ip,
rs_port=guest_rs_port,
rp_ctx=rp_context)
_tag = "Hello2"
rs = context.load(name="RsaIntersectTransferVariable.rsa_pubkey",
tag="{}".format(_tag))
fp = open("testA.model", 'r')
obj = fp.read(35)
futures = rs.push(obj, remote_parties)
fp.close()
for future in futures:
role, party = future.result()
print("result:", role, party)
def test_remote_rollpair(self):
rp_context = get_debug_test_context(is_standalone, manager_port_guest,
egg_port_guest,
transfer_port_guest,
'testing_guest')
data = [("k1", "v1"), ("k2", "v2"), ("k3", "v3"), ("k4", "v4"),
("k5", "v5"), ("k6", "v6")]
context = RollSiteContext("atest2",
self_role='guest',
self_partyId=guest_partyId,
rs_ip=guest_ip,
rs_port=guest_rs_port,
rp_ctx=rp_context)
rp_options = {'include_key': True}
rp = rp_context.load("namespace", "name").put_all(data,
options=rp_options)
_tag = "Hello"
rs = context.load(name="roll_pair_name.table", tag="roll_pair_tag")
futures = rs.push(rp, remote_parties)
for future in futures:
role, party = future.result()
print("result:", role, party)
def get_debug_test_context(is_standalone=False,
manager_port=4670,
command_port=20001,
transfer_port=20002,
role='host',
self_party_id=None,
props_file=default_props_file,
roll_site_session_id=None):
rp_context = rpta.get_debug_test_context(
is_standalone=is_standalone,
manager_port=manager_port,
egg_port=command_port,
transfer_port=transfer_port,
session_id='_'.join([roll_site_session_id, role,
str(self_party_id)]))
rs_context = RollSiteContext(roll_site_session_id=roll_site_session_id,
rp_ctx=rp_context,
options=get_option(
role=role,
self_party_id=self_party_id,
conf_file=props_file))
return rs_context
def setUp(self):
self.ctx = get_debug_test_context()
def setUpClass(cls) -> None:
#cls.rpc = get_debug_test_context()
cls.rpc = get_debug_test_context(True)
cls.rptc = RptContext(cls.rpc)
def setUpClass(cls) -> None:
cls.rpc = get_debug_test_context(True)
cls.rptc = RptContext(cls.rpc)
cls.rsc = get_debug_rs_context(cls.rpc, "rs_sid1", host_options)
