def test_send_on_valve_is_none(self):
kwargs_1 = {'1': 1}
kwargs_2 = {'2': 2}
valve.unload_valve_function()
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_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()