def test_callback(mocker):
a = EigerCommand(name="test")
x = mocker.Mock()
a.set_callback_start(x)
y = mocker.Mock()
a.set_callback_call(y)
z = mocker.Mock()
a.set_callback_end(z)
a()
x.assert_called()
y.assert_called()
z.assert_called()
def add_command():
command = EigerTask("command")
initialize = EigerCommand("initialize")
command.add_child(initialize)
return command
def add_disarm_command(state_machine):
disarm = EigerCommand("disarm")
disarm.set_callback_start(state_machine.configure)
disarm.set_callback_call(state_machine.delay)
disarm.set_callback_end(state_machine.idle)
return disarm
def add_trigger_command(state_machine):
trigger = EigerCommand("trigger")
trigger.set_callback_start(state_machine.acquire)
trigger.set_callback_call(state_machine.delay)
trigger.set_callback_end(state_machine.ready)
return trigger
def add_arm_command(state_machine):
arm = EigerCommand("arm")
arm.set_callback_start(state_machine.configure)
arm.set_callback_call(state_machine.delay)
arm.set_callback_end(state_machine.ready)
return arm
def add_initialize_command(state_machine):
initialize = EigerCommand("initialize")
initialize.set_callback_start(state_machine.initialize)
initialize.set_callback_call(state_machine.init)
initialize.set_callback_end(state_machine.idle)
return initialize