diff --git a/src/tests/test_stats_manager.py b/src/tests/test_stats_manager.py new file mode 100644 index 00000000..fd222e0a --- /dev/null +++ b/src/tests/test_stats_manager.py @@ -0,0 +1,24 @@ +from unittest.mock import Mock +import pytest +from seedpass.core.stats_manager import StatsManager + +def test_stats_manager_reset(): + """Test that StatsManager.reset correctly resets the displayed flag.""" + stats_manager = StatsManager() + manager = Mock() + + # Initial call should display stats + stats_manager.display_stats_once(manager) + manager.display_stats.assert_called_once() + + # Second call should not display stats again + manager.reset_mock() + stats_manager.display_stats_once(manager) + manager.display_stats.assert_not_called() + + # Reset the stats manager + stats_manager.reset() + + # Call after reset should display stats again + stats_manager.display_stats_once(manager) + manager.display_stats.assert_called_once()