Skip to content

Commit d41744b

Browse files
committed
Add retry config for unit test
faster failure
1 parent 1832a9e commit d41744b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

graphdatascience/tests/unit/test_arrow_runner.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,14 @@ def test_create_with_provided_connection(runner: CollectingQueryRunner) -> None:
3838
listenAddress="localhost:1234", enabled=True, running=True, versions=[ArrowEndpointVersion.V1.version()]
3939
)
4040

41-
arrow_runner = ArrowQueryRunner.create(runner, arrow_info, connection_string_override="localhost:4321")
41+
retry_config = RetryConfigV2(
42+
retryable_exceptions=[],
43+
stop_config=StopConfig(after_attempt=1),
44+
)
45+
46+
arrow_runner = ArrowQueryRunner.create(
47+
runner, arrow_info, retry_config=retry_config, connection_string_override="localhost:4321"
48+
)
4249

4350
assert isinstance(arrow_runner, ArrowQueryRunner)
4451

0 commit comments

Comments
 (0)