def test_call_successfull_rpc_from_inside_rpc(self):
serving_client1 = self._checked_connect()
golden_return = {'blub': 'foo'}
def callback1(server_context, args_string):
call_result = self.library.swiboe_rpc_ok(json.dumps(golden_return))
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback1 = swiboe.RPC(callback1)
self._ok(
self.library.swiboe_new_rpc(serving_client1, 'test.test', 100,
rpc_callback1))
serving_client2 = self._checked_connect()
def callback2(server_context, args_string):
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_server_context_call_rpc(
server_context, 'test.test', args_string,
byref(client_context)))
call_result = swiboe.PtrRpcResult()
self._ok(
self.library.swiboe_client_context_wait(
client_context, byref(call_result)))
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback2 = swiboe.RPC(callback2)
self._ok(
self.library.swiboe_new_rpc(serving_client2, 'test.foo', 100,
rpc_callback2))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.foo', 'null',
byref(client_context)))
call_result = swiboe.PtrRpcResult()
self._ok(
self.library.swiboe_client_context_wait(client_context,
byref(call_result)))
self.assertTrue(self.library.swiboe_rpc_result_is_ok(call_result))
json_blob = c_char_p()
self.library.swiboe_rpc_result_unwrap(call_result, byref(json_blob))
self.assertEqual(golden_return, json.loads(json_blob.value))
self.library.swiboe_delete_string(json_blob)
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client1))
self._ok(self.library.swiboe_disconnect(serving_client2))
def test_call_not_handled_rpc(self):
serving_client = self._checked_connect()
def callback(server_context, args_string):
call_result = self.library.swiboe_rpc_not_handled()
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(serving_client, 'test.test', 100,
rpc_callback))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.test', 'null',
byref(client_context)))
call_result = swiboe.PtrRpcResult()
self._ok(
self.library.swiboe_client_context_wait(client_context,
byref(call_result)))
self.assertFalse(self.library.swiboe_rpc_result_is_ok(call_result))
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client))
def test_new_rpc(self):
client = self._checked_connect()
def callback(server_context, args_string):
return
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(client, 'list_rust_files', 100,
rpc_callback))
self._ok(self.library.swiboe_disconnect(client))
def test_streaming_cancelled_rpc(self):
serving_client = self._checked_connect()
done_event = threading.Event()
def callback(server_context, args_string):
i = 0
while not self.library.swiboe_server_context_cancelled(
server_context):
update = {'count': i}
rv = self.library.swiboe_server_context_update(
server_context, json.dumps(update))
self.assertTrue(rv == swiboe.SUCCESS
or rv == swiboe.ERR_RPC_DONE)
i += 1
done_event.set()
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(serving_client, 'test.test', 100,
rpc_callback))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.test', 'null',
byref(client_context)))
for i in range(10):
json_str = c_char_p()
self._ok(
self.library.swiboe_client_context_recv(
client_context, byref(json_str)))
self.assertEqual(i, json.loads(json_str.value)['count'])
self.library.swiboe_delete_string(json_str)
self._ok(self.library.swiboe_client_context_cancel(client_context))
done_event.wait()
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client))
def test_call_successfull_rpc(self):
serving_client = self._checked_connect()
golden_return = {'blub': 'foo'}
def callback(server_context, args_string):
call_result = self.library.swiboe_rpc_ok(json.dumps(golden_return))
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(serving_client, 'test.test', 100,
rpc_callback))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.test', 'null',
byref(client_context)))
call_result = swiboe.PtrRpcResult()
# We do not expect any data to be streamed, so try_recv should return nothing.
try_recv_result = c_char_p()
self._ok(
self.library.swiboe_client_context_try_recv(
client_context, byref(try_recv_result)))
self.assertEqual(None, try_recv_result.value)
self.library.swiboe_delete_string(try_recv_result)
self._ok(
self.library.swiboe_client_context_wait(client_context,
byref(call_result)))
self.assertTrue(self.library.swiboe_rpc_result_is_ok(call_result))
json_blob = c_char_p()
self.library.swiboe_rpc_result_unwrap(call_result, byref(json_blob))
self.assertEqual(golden_return, json.loads(json_blob.value))
self.library.swiboe_delete_string(json_blob)
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client))
def test_call_rpc_returning_error(self):
serving_client = self._checked_connect()
error = {'details': u'Needed föö, got blah.'}
def callback(server_context, args_string):
self.assertEqual(
False,
self.library.swiboe_server_context_cancelled(server_context))
call_result = self.library.swiboe_rpc_error(
swiboe.RPC_ERR_INVALID_ARGS, json.dumps(error))
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(serving_client, 'test.test', 100,
rpc_callback))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.test', 'null',
byref(client_context)))
call_result = swiboe.PtrRpcResult()
self.assertEqual(
False, self.library.swiboe_client_context_done(client_context))
self._ok(
self.library.swiboe_client_context_wait(client_context,
byref(call_result)))
self.assertFalse(self.library.swiboe_rpc_result_is_ok(call_result))
details = c_char_p()
rpc_error_code = self.library.swiboe_rpc_result_unwrap_err(
call_result, byref(details))
self.assertEqual(swiboe.RPC_ERR_INVALID_ARGS, rpc_error_code)
self.assertEqual(error, json.loads(details.value))
self.library.swiboe_delete_string(details)
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client))
def test_streaming_successfull_rpc(self):
serving_client = self._checked_connect()
last = {'blub': 'foo'}
def callback(server_context, args_string):
for i in range(1, 4):
update = {'count': i}
self._ok(
self.library.swiboe_server_context_update(
server_context, json.dumps(update)))
call_result = self.library.swiboe_rpc_ok(json.dumps(last))
self._ok(
self.library.swiboe_server_context_finish(
server_context, call_result))
rpc_callback = swiboe.RPC(callback)
self._ok(
self.library.swiboe_new_rpc(serving_client, 'test.test', 100,
rpc_callback))
client = self._checked_connect()
client_context = swiboe.PtrClientContext()
self._ok(
self.library.swiboe_client_call_rpc(client, 'test.test', 'null',
byref(client_context)))
json_str = c_char_p()
self._ok(
self.library.swiboe_client_context_recv(client_context,
byref(json_str)))
self.assertEqual(1, json.loads(json_str.value)['count'])
self.library.swiboe_delete_string(json_str)
json_str = c_char_p()
self._ok(
self.library.swiboe_client_context_recv(client_context,
byref(json_str)))
self.assertEqual(2, json.loads(json_str.value)['count'])
self.library.swiboe_delete_string(json_str)
json_str = c_char_p()
self._ok(
self.library.swiboe_client_context_recv(client_context,
byref(json_str)))
self.assertEqual(3, json.loads(json_str.value)['count'])
self.library.swiboe_delete_string(json_str)
json_str = c_char_p()
self._ok(
self.library.swiboe_client_context_recv(client_context,
byref(json_str)))
self.assertEqual(None, json_str.value)
self.library.swiboe_delete_string(json_str)
self.assertEqual(
True, self.library.swiboe_client_context_done(client_context))
call_result = swiboe.PtrRpcResult()
self._ok(
self.library.swiboe_client_context_wait(client_context,
byref(call_result)))
self.assertTrue(self.library.swiboe_rpc_result_is_ok(call_result))
json_blob = c_char_p()
self.library.swiboe_rpc_result_unwrap(call_result, byref(json_blob))
self.assertEqual(last, json.loads(json_blob.value))
self.library.swiboe_delete_string(json_blob)
self._ok(self.library.swiboe_disconnect(client))
self._ok(self.library.swiboe_disconnect(serving_client))