def setUp(self):
self._burner = mock.Mock()
self._burner.disabled = mock.Mock()
self._burner.running = mock.Mock()
self._burner.get_fire_value = mock.Mock()
self._burner.waiting = mock.Mock()
self._burnerController = BurnerController(self._burner)
class BurnerControllerTests(unittest.TestCase):
_burnerController = None
_burner = None
def setUp(self):
self._burner = mock.Mock()
self._burner.disabled = mock.Mock()
self._burner.running = mock.Mock()
self._burner.get_fire_value = mock.Mock()
self._burner.waiting = mock.Mock()
self._burnerController = BurnerController(self._burner)
def test_adc_value_calculates_average_from_list_correctly(self):
self._burnerController._fireValue[0] = 60 # AVG should be 1 when list size is 60. (All rest are 0).
self.assertEquals(1, self._burnerController.light_sensor())
def test_tick_gets_burner_value_and_adds_it_to_history(self):
self._burner.get_fire_value = mock.Mock(return_value=40) # AVG should be 2, because 40+40+40 = 120.
self._burnerController.tick() # Reads one value and adds it to history.
self._burnerController.tick()
self._burnerController.tick()
self.assertEquals(self._burnerController.light_sensor(), 2)
def test_tick_works_correctly(self):
self._burnerController.start_cycle(screw_ticks=1, delay_ticks=1)
self.assertEquals(self._burner.mock_calls, [mock.call.running()])
self.assertEqual(self._burnerController.tick(), True) # Device is running.
self.assertEquals(self._burner.mock_calls, [mock.call.running(), mock.call.get_fire_value(), mock.call.waiting()])
self.assertEqual(self._burnerController.tick(), False) # Device is not running anymore.
self.assertEquals(self._burner.mock_calls,
[mock.call.running(),
mock.call.get_fire_value(),
mock.call.waiting(),
mock.call.get_fire_value(),
mock.call.disabled()])
def test_disable_calls_burner_disable(self):
self._burnerController.disable()
self._burner.disabled.assert_called_once()
