46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import Mock
|
|
|
|
sys.path[:0] = [str(Path(__file__).resolve().parents[1] / 'apps'),
|
|
str(Path(__file__).resolve().parents[1] / 'apps/ad_toolbox')]
|
|
|
|
from informationscollector import InformationsCollector
|
|
|
|
|
|
class WarningCollectorTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.app = object.__new__(InformationsCollector)
|
|
self.app.warning_list = []
|
|
self.app.warning_count_sensor = Mock()
|
|
self.app.warning_count_sensor.get_state.return_value = '2'
|
|
|
|
def test_warning_is_counted_and_exposed_in_attributes(self):
|
|
self.app.on_log_warning(
|
|
'example', None, 'WARNING', None, 'Something happened', {}
|
|
)
|
|
|
|
self.app.warning_count_sensor.set_state.assert_called_once_with(
|
|
state=3,
|
|
attributes={
|
|
'last_warning': 'Something happened',
|
|
'warning_list': ['Something happened'],
|
|
},
|
|
)
|
|
|
|
def test_warning_reset_clears_count_and_attributes(self):
|
|
self.app.warning_list = ['Old warning']
|
|
|
|
self.app.on_reset_warning_count(None, None, None, None, {})
|
|
|
|
self.assertEqual(self.app.warning_list, [])
|
|
self.app.warning_count_sensor.set_state.assert_called_once_with(
|
|
state=0,
|
|
attributes={'last_warning': '', 'warning_list': []},
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|