From 048782e26cd99f0fd34af7e3cd1d82e3e97db5f6 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Fri, 29 May 2015 20:16:47 +0800
Subject: [PATCH] test/scheduler: test flush

---
 artiq/test/scheduler.py | 40 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

diff --git a/artiq/test/scheduler.py b/artiq/test/scheduler.py
index 36e11b302..436f9316a 100644
--- a/artiq/test/scheduler.py
+++ b/artiq/test/scheduler.py
@@ -27,11 +27,11 @@ def _get_expid(name):
     }
 
 
-def _get_basic_steps(rid, expid):
+def _get_basic_steps(rid, expid, priority=0, flush=False):
     return [
         {"action": "setitem", "key": rid, "value": 
-            {"pipeline": "main", "status": "pending", "priority": 0,
-             "expid": expid, "due_date": None, "flush": False},
+            {"pipeline": "main", "status": "pending", "priority": priority,
+             "expid": expid, "due_date": None, "flush": flush},
             "path": []},
         {"action": "setitem", "key": "status", "value": "preparing",
             "path": [rid]},
@@ -120,3 +120,37 @@ class SchedulerCase(unittest.TestCase):
         scheduler.submit("main", expid, 0, None, False)
         loop.run_until_complete(done.wait())
         loop.run_until_complete(scheduler.stop())
+
+    def test_flush(self):
+        scheduler = Scheduler(0, _handlers)
+        expid = _get_expid("EmptyExperiment")
+
+        expect = _get_basic_steps(1, expid, 1, True)
+        expect.insert(1, {"key": "status",
+                          "path": [1],
+                          "value": "flushing",
+                          "action": "setitem"})
+        first_preparing = asyncio.Event()
+        done = asyncio.Event()
+        expect_idx = 0
+        def notify(notifier, mod):
+            nonlocal expect_idx
+            if mod == {"path": [0],
+                       "value": "preparing",
+                       "key": "status",
+                       "action": "setitem"}:
+                first_preparing.set()
+            if mod["path"] == [1] or (mod["path"] == [] and mod["key"] == 1):
+                self.assertEqual(mod, expect[expect_idx])
+                expect_idx += 1
+                if expect_idx >= len(expect):
+                    done.set()
+        scheduler.notifier.publish = notify
+
+        loop = asyncio.get_event_loop()
+        scheduler.start()
+        scheduler.submit("main", expid, 0, None, False)
+        loop.run_until_complete(first_preparing.wait())
+        scheduler.submit("main", expid, 1, None, True)
+        loop.run_until_complete(done.wait())
+        loop.run_until_complete(scheduler.stop())