def unfreeze():
# turn off switch
FunctionSwitch.objects.unfreeze_engine()
# resend signal
valve.open_valve(signals)
# unfreeze process
frozen_process_list = PipelineProcess.objects.filter(is_frozen=True)
for process in frozen_process_list:
process.unfreeze()
def test_open_valve(self):
kwargs_1 = {'1': 1}
kwargs_2 = {'2': 2}
def valve_closed():
return True
valve.unload_valve_function()
valve.set_valve_function(valve_closed)
valve.send(mock_signal, 'signal_1', **kwargs_1)
valve.send(mock_signal, 'signal_1', **kwargs_2)
self.assertEqual(len(mock_signal.signal_1.history), 0)
valve.open_valve(mock_signal)
self.assertEqual(mock_signal.signal_1.history[0], kwargs_1)
self.assertEqual(mock_signal.signal_1.history[1], kwargs_2)
mock_signal.clear()