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', [])