Calculate timeout using max_wait

krassowski · web-flow · commit 2152eae06d13 · 2021-01-31T00:28:41.000Z
diff --git a/python_packages/jupyter_lsp/jupyter_lsp/tests/test_stdio.py b/python_packages/jupyter_lsp/jupyter_lsp/tests/test_stdio.py
@@ -72,7 +72,7 @@ async def test_reader(message, repeats, interval, communicator_spawner):
         message=message, repeats=repeats, interval=interval
     )
     reader = LspStdIoReader(stream=process.stdout, queue=queue)
-    timeout = 2 + (interval or 1) * repeats * 10
+    timeout = 3 + reader.max_wait * repeats * 10
 
     communicate_and_close(process)
     await asyncio.wait_for(reader.read(), timeout=timeout)

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ async def test_reader(message, repeats, interval, communicator_spawner):`
`72`	`72`	`message=message, repeats=repeats, interval=interval`
`73`	`73`	`)`
`74`	`74`	`reader = LspStdIoReader(stream=process.stdout, queue=queue)`
`75`		`- timeout = 2 + (interval or 1) * repeats * 10`
	`75`	`+ timeout = 3 + reader.max_wait * repeats * 10`
`76`	`76`
`77`	`77`	`communicate_and_close(process)`
`78`	`78`	`await asyncio.wait_for(reader.read(), timeout=timeout)`