comp-chem/cp2k
1
0
Fork 0
mirror of https://github.com/cp2k/cp2k.git synced 2026-07-28 06:05:29 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

regtesting: Add timeouts to shell communication

Browse source
This commit is contained in:
Ole Schütt 2022-01-11 17:24:42 +01:00 • committed by Ole Schütt
parent e076040b12
commit cbaf0f4606
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tools/regtesting/do_regtest.py
View file

@ -131,9 +131,10 @@ async def main() -> None:
if not tasks:
print("\nNo test directories selected, check --restrictdir filter.")
sys.exit(1)
print(f"Launched {len(tasks)} test directories and {cfg.num_workers} worker...\n")
sys.stdout.flush()
# Wait for tasks to finish and print their results.
print(f"Launched {len(tasks)} test directories and {cfg.num_workers} worker...\n")
all_results: List[TestResult] = []
with open(cfg.error_summary, "wt", encoding="utf8", errors="replace") as err_fh:
for num_done, task in enumerate(asyncio.as_completed(tasks)):
@ -374,13 +375,14 @@ class Cp2kShell:
data = (line + "\n").encode("utf8")
self._child.stdin.write(data)
try:
await self._child.stdin.drain()
await asyncio.wait_for(self._child.stdin.drain(), timeout=self.cfg.timeout)
except ConnectionResetError:
pass
async def readline(self) -> str:
assert self._child and self._child.stdout
data = await self._child.stdout.readline()
data_future = self._child.stdout.readline()
data = await asyncio.wait_for(data_future, timeout=self.cfg.timeout)
line = data.decode("utf8", errors="replace")
if self.cfg.debug:
print("Received: " + line, end="")