From f158711f7edab2437555e5d006637e9be77b5da0 Mon Sep 17 00:00:00 2001 From: Sebastien Bourdeauducq Date: Wed, 11 Mar 2015 19:07:04 +0100 Subject: [PATCH] test/worker: test watchdog in build() --- artiq/test/worker.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/artiq/test/worker.py b/artiq/test/worker.py index 5d3aad34d..80e76735a 100644 --- a/artiq/test/worker.py +++ b/artiq/test/worker.py @@ -4,7 +4,7 @@ import sys from time import sleep from artiq import * -from artiq.master.worker import * +from artiq.master.worker import * class WatchdogNoTimeout(Experiment, AutoDB): @@ -20,6 +20,15 @@ class WatchdogTimeout(Experiment, AutoDB): sleep(100.0) +class WatchdogTimeoutInBuild(Experiment, AutoDB): + def build(self): + with self.scheduler.watchdog(0.1*s): + sleep(100.0) + + def run(self): + pass + + @asyncio.coroutine def _call_worker(worker, run_params): yield from worker.prepare(0, run_params) @@ -52,3 +61,7 @@ class WatchdogCase(unittest.TestCase): def test_watchdog_timeout(self): with self.assertRaises(WorkerWatchdogTimeout): _run_experiment("WatchdogTimeout") + + def test_watchdog_timeout_in_build(self): + with self.assertRaises(WorkerWatchdogTimeout): + _run_experiment("WatchdogTimeoutInBuild")