def test_double_turn_off(self):
# setup
airily_dao = AirlyDao('url', 'key', 28)
airily_dao.caqi = MagicMock(return_value=10)
calendar_dao = CalendarDao('url')
air_purifier = AirPurifier(ConfigMock())
air_purifier.turn_off = MagicMock()
air_purifier.turn_on = MagicMock()
air_purifier.get_state = MagicMock(return_value=AirPurifierState.ON)
cleaning_pause = self.get_cleaning_pause(['Monday', 'Tuesday', 'Wednesday'], '8:00', '16:00')
clean_airtomation_service = CleanAirtomationService(70, cleaning_pause, airily_dao, calendar_dao,
air_purifier)
# test
clean_airtomation_service.clean_polluted_air()
air_purifier.get_state = MagicMock(return_value=AirPurifierState.OFF)
clean_airtomation_service.clean_polluted_air()
# verify
air_purifier.turn_on.assert_not_called()
air_purifier.turn_off.assert_called_once()
def test_pause_edge_right(self):
# setup
airily_dao = AirlyDao('url', 'key', 28)
airily_dao.caqi = MagicMock(return_value=10)
calendar_dao = CalendarDao('url')
air_purifier = AirPurifier(ConfigMock())
air_purifier.turn_off = MagicMock()
air_purifier.turn_on = MagicMock()
air_purifier.get_state = MagicMock(return_value=AirPurifierState.ON)
cleaning_pause = self.get_cleaning_pause(['Monday', 'Tuesday', 'Wednesday'], '8:00', '16:00')
clean_airtomation_service = CleanAirtomationService(70, cleaning_pause, airily_dao, calendar_dao,
air_purifier)
# test
result = clean_airtomation_service._is_not_in_pause_time()
# verify
self.assertFalse(result)
def test_switch_off_success_switch_on_failure(self):
air_purifier = AirPurifier(MockedConfig())
self.assertTrue(air_purifier.turn_off())
self.assertEqual(AirPurifierState.OFF, air_purifier.get_state())
self.assertFalse(air_purifier.turn_on())
self.assertEqual(AirPurifierState.OFF, air_purifier.get_state())
def test_switch_on_success(self):
air_purifier = AirPurifier(MockedConfig())
self.assertTrue(air_purifier.turn_on())
self.assertEqual(AirPurifierState.ON, air_purifier.get_state())