class TestDefaultStateMachine(unittest.TestCase):
def setUp(self):
self.SM = DefaultStateMachine()
def test_init(self):
default_allowed_transitions = OrderedDict()
default_allowed_transitions['Resetting'] = {
'Ready', 'Fault', 'Disabling'
}
default_allowed_transitions['Ready'] = {"Fault", "Disabling"}
default_allowed_transitions['Fault'] = {"Resetting", "Disabling"}
default_allowed_transitions['Disabling'] = {"Disabled", "Fault"}
default_allowed_transitions['Disabled'] = {"Resetting"}
self.assertEqual(default_allowed_transitions,
self.SM.allowed_transitions)
self.assertEqual(["Resetting"], self.SM.busy_states)
def test_is_allowed(self):
self.SM.allowed_transitions["Ready"] = {"Resetting", "Seeking"}
response = self.SM.is_allowed("Ready", "Resetting")
self.assertTrue(response)
response = self.SM.is_allowed("Ready", "Paused")
self.assertFalse(response)
def test_set_allowed(self):
self.SM.set_allowed("Ready", "Prerun")
self.assertEqual({"Prerun", "Disabling", "Fault"},
self.SM.allowed_transitions['Ready'])
self.SM.set_allowed("Ready", "Resetting")
self.assertEqual({"Prerun", "Disabling", "Fault", "Resetting"},
self.SM.allowed_transitions['Ready'])
def test_set_busy(self):
self.assertEqual(["Resetting"], self.SM.busy_states)
self.SM.set_busy("Resetting", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=True)
self.assertEqual(["Ready"], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
def test_is_busy(self):
self.assertEqual(['Resetting'], self.SM.busy_states)
response = self.SM.is_busy("Resetting")
self.assertTrue(response)
response = self.SM.is_busy("Ready")
self.assertFalse(response)
class TestDefaultStateMachine(unittest.TestCase):
def setUp(self):
self.SM = DefaultStateMachine()
def test_init(self):
default_allowed_transitions = OrderedDict()
default_allowed_transitions['Resetting'] = {'Ready', 'Fault', 'Disabling'}
default_allowed_transitions['Ready'] = {"Fault", "Disabling"}
default_allowed_transitions['Fault'] = {"Resetting", "Disabling"}
default_allowed_transitions['Disabling'] = {"Disabled", "Fault"}
default_allowed_transitions['Disabled'] = {"Resetting"}
self.assertEqual(default_allowed_transitions,
self.SM.allowed_transitions)
self.assertEqual(["Resetting"], self.SM.busy_states)
def test_is_allowed(self):
self.SM.allowed_transitions["Ready"] = {"Resetting", "Seeking"}
response = self.SM.is_allowed("Ready", "Resetting")
self.assertTrue(response)
response = self.SM.is_allowed("Ready", "Paused")
self.assertFalse(response)
def test_set_allowed(self):
self.SM.set_allowed("Ready", "Prerun")
self.assertEqual({"Prerun", "Disabling", "Fault"},
self.SM.allowed_transitions['Ready'])
self.SM.set_allowed("Ready", "Resetting")
self.assertEqual({"Prerun", "Disabling", "Fault", "Resetting"},
self.SM.allowed_transitions['Ready'])
def test_set_busy(self):
self.assertEqual(["Resetting"], self.SM.busy_states)
self.SM.set_busy("Resetting", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=True)
self.assertEqual(["Ready"], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
def test_is_busy(self):
self.assertEqual(['Resetting'], self.SM.busy_states)
response = self.SM.is_busy("Resetting")
self.assertTrue(response)
response = self.SM.is_busy("Ready")
self.assertFalse(response)
class TestDefaultStateMachine(unittest.TestCase):
def setUp(self):
self.SM = DefaultStateMachine("test_state_machine")
def test_init(self):
default_allowed_transitions = OrderedDict()
default_allowed_transitions['Resetting'] = {
'Ready', 'Fault', 'Disabling'
}
default_allowed_transitions['Ready'] = {"Fault", "Disabling"}
default_allowed_transitions['Fault'] = {"Resetting", "Disabling"}
default_allowed_transitions['Disabling'] = {"Disabled", "Fault"}
default_allowed_transitions['Disabled'] = {"Resetting"}
self.assertEqual("test_state_machine", self.SM.name)
self.assertEqual(default_allowed_transitions,
self.SM.allowed_transitions)
self.assertEqual(["Resetting"], self.SM.busy_states)
def test_is_allowed(self):
self.SM.allowed_transitions.update(
dict(Ready={"Resetting", "Rewinding"}))
response = self.SM.is_allowed("Ready", "Resetting")
self.assertTrue(response)
response = self.SM.is_allowed("Ready", "Paused")
self.assertFalse(response)
def test_set_allowed(self):
self.SM.set_allowed("Ready", "Prerun")
self.assertEqual({"Prerun", "Disabling", "Fault"},
self.SM.allowed_transitions['Ready'])
self.SM.set_allowed("Ready", "Resetting")
self.assertEqual({"Prerun", "Disabling", "Fault", "Resetting"},
self.SM.allowed_transitions['Ready'])
def test_set_busy(self):
self.assertEqual(["Resetting"], self.SM.busy_states)
self.SM.set_busy("Resetting", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
self.SM.set_busy("Ready", busy=True)
self.assertEqual(["Ready"], self.SM.busy_states)
self.SM.set_busy("Ready", busy=False)
self.assertEqual([], self.SM.busy_states)
def test_is_busy(self):
self.assertEqual(['Resetting'], self.SM.busy_states)
response = self.SM.is_busy("Resetting")
self.assertTrue(response)
response = self.SM.is_busy("Ready")
self.assertFalse(response)
def test_insert(self):
@DefaultStateMachine.insert
class DummyController(object):
pass
d = DummyController()
self.assertIsInstance(d.stateMachine, DefaultStateMachine)
self.assertEqual("DefaultStateMachine", d.stateMachine.name)