test_scheduler: Add SchedulerMonitor

Use it in test_pending_priority
check_status to see the update of schedulers is expected or not

Author: Deepskyhunter

artiq/test/test_scheduler.py

@@ -83,6 +83,27 @@ def get(self):
         self._next_rid += 1
         return rid
 
+class SchedulerMonitor():
+    def __init__(self, expect_status):
+        self.expect_status = expect_status
+        self.schedulers = {}
+        self.current_status = {}
+
+    def check_status(self, test):
+        self.check_counter += 1
+        for key, value in self.schedulers.items():
+            if key not in self.current_status.keys():
+                self.current_status[key] = ""
+            if self.schedulers[key]["status"] != self.current_status[key]:
+                test.assertEqual(self.schedulers[key]["status"], self.expect_status[key][0])
+                self.current_status[key] = self.expect_status[key].pop(0)
+
+    def finished(self, test):
+        for value in self.expect_status.values():
+            if value:
+                return False
+        return True
+
 
 class SchedulerCase(unittest.TestCase):
     def setUp(self):
@@ -145,46 +166,23 @@ def test_pending_priority(self):
         late = time() + 100000
         early = time() + 1
 
-        schedule = {}
-
-        expect = [
-            {0: "pending"},
-            {0: "pending", 1: "pending"},
-            {0: "pending", 1: "pending", 2: "pending"},
-            {0: "preparing", 1: "pending", 2: "pending"},
-            {0: "prepare_done", 1: "pending", 2: "pending"},
-            {0: "running", 1: "pending", 2: "preparing"},
-            {0: "running", 1: "pending", 2: "prepare_done"},
-            {0: "paused", 1: "pending", 2: "running"},
-            {0: "paused", 1: "pending", 2: "run_done"},
-            {0: "running", 1: "pending", 2: "analyzing"},
-            {0: "running", 1: "pending", 2: "deleting"},
-        ]
+        expect_status = {
+            0: ["pending", "preparing", "prepare_done",
+                "running", "paused", "running"],
+            1: ["pending"],
+            2: ["pending", "preparing", "prepare_done",
+                "running", "run_done", "analyzing", "deleting"],
+        }
+
+        scheduler_mon = SchedulerMonitor(expect_status)
 
         done = asyncio.Event()
-        expect_idx = 0
-        skip_next = False
 
         def notify(mod):
-            nonlocal expect_idx, skip_next
-            process_mod(schedule, mod)
-            
-            # gather status of each RID
-            current_status = {}
-            for rid, info in schedule.items():
-                current_status[rid] = info["status"]
-
-            # skip once after prepare_done or run_done
-            if skip_next:
-                skip_next = False
-            else:
-                self.assertEqual(current_status, expect[expect_idx])
-                expect_idx += 1
-                if "prepare_done" in current_status.values() or\
-                    "run_done" in current_status.values():
-                    skip_next = True
+            process_mod(scheduler_mon.schedulers, mod)
+            scheduler_mon.check_status(self)
 
-            if expect_idx >= len(expect):
+            if scheduler_mon.finished(self):
                 done.set()
 
         scheduler.notifier.publish = notify