icloud_photos_downloader/tests/test_summary.py at f0241a29c66ac3a49ca223c0bd1385ddb8e39b3e · icloud-photos-downloader/icloud_photos_downloader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""Tests for the SummaryTracker class"""

from unittest.mock import patch, Mock
from icloudpd.summary import SummaryTracker
from icloudpd.logger import setup_logger


def test_prints_summary():
    """Test that the summary tracker prints the overall time"""
    with patch("time.time", side_effect=[1000.0, 1005.0]):
        logger = setup_logger()
        summary_tracker = SummaryTracker(logger)
        summary_tracker.start_timer()
        summary_tracker.stop_timer()

        mock_logger_info = Mock()
        with patch.object(logger, "info", mock_logger_info):
            summary_tracker.print_summary()

        assert mock_logger_info.call_args_list[0][0][0] == "\nExecution Summary"
        assert mock_logger_info.call_args_list[1][0][0] == "================="
        assert "Overall time:" in mock_logger_info.call_args_list[2][0][0]
        assert mock_logger_info.call_args_list[2][0][1].seconds == 5