def test_exception_handling(local_config):
"""Tests that OperationalError is handled by all methods without throwing an exception."""
manager = StateManager(local_config)
manager._persistor._cursor_factory = BreakingCursor
manager.set_checkpoint('ABCDEFGH', 'default', 'ALPHA')
assert manager.get_previous_hashes('default') == []
manager.prune("2020-01-12T00:00:00")
manager.add_report_item(6, 'default', {'keyval': 1})
assert manager.get_current_report_items(6, 'default') == []
manager.clear_report_items(6, 'default')
def test_clear_report_items(local_config):
"""Test the clear_report_items() API."""
manager = StateManager(local_config)
manager.clear_report_items(6, 'default')
assert getattr(manager._persistor, "_sc", 0) == 0
assert getattr(manager._persistor, "_gph", 0) == 0
assert getattr(manager._persistor, "_guh", 0) == 0
assert getattr(manager._persistor, "_p", 0) == 0
assert getattr(manager._persistor, "_ari", 0) == 0
assert getattr(manager._persistor, "_gcri", 0) == 0
assert getattr(manager._persistor, "_cri", 0) == 1
def test_report_items(local_config):
"""Tests the management of report items."""
manager = StateManager(local_config)
manager.add_report_item(6, 'default', {'keyval': 1})
manager.add_report_item(6, 'default', {'keyval': 4})
manager.add_report_item(6, 'default', {'keyval': 9})
manager.add_report_item(2, 'default', {'keyval': 2})
manager.add_report_item(2, 'default', {'keyval': 3})
_test_check_report_items(manager.get_current_report_items(6, 'default'), 'keyval', [1, 4, 9])
_test_check_report_items(manager.get_current_report_items(2, 'default'), 'keyval', [2, 3])
_test_check_report_items(manager.get_current_report_items(9, 'default'), 'keyval', [])
manager.clear_report_items(6, 'default')
_test_check_report_items(manager.get_current_report_items(6, 'default'), 'keyval', [])
_test_check_report_items(manager.get_current_report_items(2, 'default'), 'keyval', [2, 3])
_test_check_report_items(manager.get_current_report_items(9, 'default'), 'keyval', [])