import mock
from unittest import TestCase

from dvc.progress import progress, progress_aware


class TestProgressAware(TestCase):
    @mock.patch("sys.stdout.isatty", return_value=True)
    @mock.patch("dvc.progress.Progress._print")
    def test(self, mock_print, mock_isatty):
        function = progress_aware(lambda: None)

        function()
        mock_print.assert_not_called()

        progress.update_target("testing", 0, 100)
        function()
        mock_print.assert_called()

        progress.finish_target("testing")