+
+ def check_dispatch_classes(self, config, module):
+ setup, shutdown, update, monitor = config.dispatch_classes()
+ self.assertIs(setup, module.ComputeNodeSetupActor)
+ self.assertIs(shutdown, module.ComputeNodeShutdownActor)
+ self.assertIs(update, module.ComputeNodeUpdateActor)
+ self.assertIs(monitor, module.ComputeNodeMonitorActor)
+
+ def test_default_dispatch(self):
+ config = self.load_config()
+ self.check_dispatch_classes(config, dispatch)
+
+ def test_custom_dispatch(self):
+ config = self.load_config(
+ config_str=self.TEST_CONFIG + "[Daemon]\ndispatcher=slurm\n")
+ self.check_dispatch_classes(config, slurm_dispatch)