Differential D5105 Diff 25646 python_modules/dagster/dagster_tests/general_tests/grpc_tests/test_ping.py

Changeset View

Standalone View

python_modules/dagster/dagster_tests/general_tests/grpc_tests/test_ping.py

	Show First 20 Lines • Show All 129 Lines • ▼ Show 20 Lines

	def test_streaming():			def test_streaming():
	with ephemeral_grpc_api_client() as api_client:			with ephemeral_grpc_api_client() as api_client:
	results = [result for result in api_client.streaming_ping(sequence_length=10, echo="foo")]			results = [result for result in api_client.streaming_ping(sequence_length=10, echo="foo")]
	assert len(results) == 10			assert len(results) == 10
	for sequence_number, result in enumerate(results):			for sequence_number, result in enumerate(results):
	assert result["sequence_number"] == sequence_number			assert result["sequence_number"] == sequence_number
	assert result["echo"] == "foo"			assert result["echo"] == "foo"


				def test_get_server_id():
				with ephemeral_grpc_api_client() as api_client:
				assert api_client.get_server_id()


				def create_server_process():
				port = find_free_port()
				server_process = open_server_process(port=port, socket=None)
				assert server_process is not None
				return port, server_process

				dgibsonUnsubmitted Not Done Inline Actions any particular reason this is an inner function? dgibson: any particular reason this is an inner function?

				def test_detect_server_restart():

				# Create first server and query ID
				port, server_process = create_server_process()
				dgibsonUnsubmitted Not Done Inline Actions you could also use GrpcServerProcess for this, but this is fine too dgibson: you could also use GrpcServerProcess for this, but this is fine too
				try:
				api_client = DagsterGrpcClient(port=port)
				server_id_one = api_client.get_server_id()
				assert server_id_one
				dgibsonUnsubmitted Not Done Inline Actions api_client.shutdown_server() would also work dgibson: api_client.shutdown_server() would also work
				finally:
				interrupt_ipc_subprocess_pid(server_process.pid)

				with pytest.raises(grpc._channel._InactiveRpcError): # pylint: disable=protected-access
				api_client.get_server_id()

				# Create second server and query ID
				port, server_process = create_server_process()
				try:
				api_client = DagsterGrpcClient(port=port)
				server_id_two = api_client.get_server_id()
				assert server_id_two
				finally:
				interrupt_ipc_subprocess_pid(server_process.pid)

				assert server_id_one != server_id_two
				alangenfeldUnsubmitted Not Done Inline Actions random failure https://buildkite.com/dagster/dagster-diffs/builds/11591#0002866e-00de-4a84-92d4-b1a8f6bcac1f alangenfeld: random failure https://buildkite.com/dagster/dagster-diffs/builds/11591#0002866e-00de-4a84-92d4…