def ready(self):
from pipeline.engine.signals import dispatch
dispatch.dispatch()
from pipeline.django_signal_valve import valve
from pipeline.engine.models import FunctionSwitch
valve.set_valve_function(FunctionSwitch.objects.is_frozen)
FunctionSwitch.objects.init_db()
def test_set_valve_function(self):
self.assertRaises(Exception, valve.set_valve_function, args=[1])
def func():
return True
valve.unload_valve_function()
valve.set_valve_function(func)
self.assertEqual(valve.valve_function(), func)
self.assertRaises(Exception, valve.set_valve_function, args=[func])
valve.__valve_function = None
def test_send_on_closed(self):
kwargs_1 = {'1': 1}
kwargs_2 = {'2': 2}
def is_valve_closed():
return True
valve.unload_valve_function()
valve.set_valve_function(is_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)
mock_signal.clear()
Signal.objects.all().delete()
def test_send_on_valve_opened(self):
kwargs_1 = {'1': 1}
kwargs_2 = {'2': 2}
def is_valve_closed():
return False
valve.unload_valve_function()
valve.set_valve_function(is_valve_closed)
valve.send(mock_signal, 'signal_1', **kwargs_1)
valve.send(mock_signal, 'signal_1', **kwargs_2)
self.assertEqual(mock_signal.signal_1.history[0], kwargs_1)
self.assertEqual(mock_signal.signal_1.history[1], kwargs_2)
mock_signal.clear()
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()