def test_persistent_header(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
client.set_persistent_header(TEST_HEADER_KEY,
TEST_HEADER_VALUE)
self.assertEqual(TEST_HEADER_VALUE,
client.readHeader(TEST_HEADER_KEY))
def test_unexpected_exception(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
with self.assertRaisesRegex(ApplicationError,
"Surprise!") as ex:
client.surprise()
self.assertEqual(ApplicationErrorType.UNKNOWN,
ex.exception.type)
def test_oneway_with_rocket(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService,
path=path,
client_type=ClientType.
THRIFT_ROCKET_CLIENT_TYPE) as client:
self.assertIsNone(client.oneway())
time.sleep(1) # wait for server to clear the queue
def test_client_type_and_protocol(self) -> None:
with server_in_another_process() as path:
with get_sync_client(
TestService,
path=path,
client_type=ClientType.THRIFT_ROCKET_CLIENT_TYPE,
protocol=Protocol.BINARY,
) as client:
sum = client.add(1, 2)
self.assertEqual(3, sum)
def test_reuse_client(self) -> None:
with server_in_another_process() as path:
client = get_sync_client(TestService, path=path)
with client:
self.assertEqual(3, client.add(1, 2))
with self.assertRaises(RuntimeError):
with client:
pass
with self.assertRaises(RuntimeError):
client.add(1, 2)
def test_onetime_header(self) -> None:
with server_in_another_process() as path:
with get_client(TestService.Client, path=path) as client:
client.set_onetime_header("ONETIME_HEADER_KEY",
"ONETIME_HEADER_VALUE")
self.assertEqual(
"ONETIME_HEADER_VALUE",
client.readHeader("ONETIME_HEADER_KEY"),
)
self.assertEqual(
"",
client.readHeader("ONETIME_HEADER_KEY"),
)
def test_exit_callback(self) -> None:
class Callback:
def __init__(self):
self.triggered = False
def trigger(self):
self.triggered = True
cb1 = Callback()
cb2 = Callback()
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
client._at_exit(cb1.trigger)
client._at_exit(cb2.trigger)
self.assertTrue(cb1.triggered)
self.assertTrue(cb2.triggered)
def test_persistent_header(self) -> None:
with server_in_another_process() as path:
with get_client(TestService.Client, path=path) as client:
client.set_persistent_header("PERSISTENT_HEADER_KEY",
"PERSISTENT_HEADER_VALUE")
self.assertEqual(
"PERSISTENT_HEADER_VALUE",
client.get_persistent_headers()["PERSISTENT_HEADER_KEY"],
)
self.assertEqual(
"PERSISTENT_HEADER_VALUE",
client.readHeader("PERSISTENT_HEADER_KEY"),
)
client.clear_persistent_headers()
self.assertFalse(client.get_persistent_headers())
self.assertEqual(
"",
client.readHeader("PERSISTENT_HEADER_KEY"),
)
def test_onetime_header_override_persistent_header(self) -> None:
with server_in_another_process() as path:
with get_client(TestService.Client, path=path) as client:
client.set_persistent_header("HEADER_KEY",
"PERSISTENT_HEADER_VALUE")
self.assertEqual(
"PERSISTENT_HEADER_VALUE",
client.readHeader("HEADER_KEY"),
)
client.set_onetime_header("HEADER_KEY", "ONETIME_HEADER_VALUE")
self.assertEqual(
"ONETIME_HEADER_VALUE",
client.readHeader("HEADER_KEY"),
)
self.assertEqual(
"PERSISTENT_HEADER_VALUE",
client.get_persistent_headers()["HEADER_KEY"],
)
self.assertEqual(
"PERSISTENT_HEADER_VALUE",
client.readHeader("HEADER_KEY"),
)
def test_derived_service(self) -> None:
with server_in_another_process() as path:
with get_sync_client(EchoService, path=path) as client:
self.assertEqual("hello", client.echo("hello"))
self.assertEqual(3, client.add(1, 2))
def test_oneway(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
self.assertIsNone(client.oneway())
time.sleep(1) # wait for server to clear the queue
def test_void_return_with_exception(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
with self.assertRaises(EmptyException):
client.oops()
def test_exception(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
self.assertAlmostEqual(2, client.divide(6, 3))
with self.assertRaises(ArithmeticException):
client.divide(1, 0)
def test_void_return(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
self.assertIsNone(client.noop())
def test_basic(self) -> None:
with server_in_another_process() as path:
with get_sync_client(TestService, path=path) as client:
self.assertEqual(3, client.add(1, 2))
def test_deriving_from_external_service(self) -> None:
with server_in_another_process() as path:
with get_sync_client(LeafService, path=path) as client:
self.assertEqual([3, 2, 1], list(client.reverse([1, 2, 3])))
self.assertEqual("hello", client.echo("hello"))
self.assertEqual(3, client.add(1, 2))