svtplay-dl/lib/svtplay_dl/tests/output.py

#!/usr/bin/python
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-

# The unittest framwork doesn't play nice with pylint:
#   pylint: disable-msg=C0103

from __future__ import absolute_import
import unittest
import svtplay_dl.output
from mock import patch


# FIXME: use mock framework instead of this hack
class mockfile(object):
    def __init__(self):
        self.content = []

    def write(self, string):
        self.content.append(string)

    def read(self):
        return self.content.pop()


class progressTest(unittest.TestCase):
    def setUp(self):
        self.mockfile = mockfile()
        svtplay_dl.output.progress_stream = self.mockfile

    @patch('svtplay_dl.output.progressbar')
    def test_0_0(self, pbar):
        svtplay_dl.output.progress(0, 0)
        self.assertFalse(pbar.called)

    @patch('svtplay_dl.output.progressbar')
    def test_0_100(self, pbar):
        svtplay_dl.output.progress(0, 100)
        pbar.assert_any_call(100, 0, "")


class progressbarTest(unittest.TestCase):
    def setUp(self):
        self.old_termsiz = svtplay_dl.output.get_terminal_size
        svtplay_dl.output.get_terminal_size = lambda: (50, 25)

        self.mockfile = mockfile()
        svtplay_dl.output.progress_stream = self.mockfile

    def tearDown(self):
        svtplay_dl.output.get_terminal_size = self.old_termsiz

    def test_0_100(self):
        svtplay_dl.output.progressbar(100, 0)
        self.assertEqual(
            self.mockfile.read(), "\r[000/100][..........] "
        )

    def test_progress_1_100(self):
        svtplay_dl.output.progressbar(100, 1)
        self.assertEqual(
            self.mockfile.read(), "\r[001/100][..........] "
        )

    def test_progress_2_100(self):
        svtplay_dl.output.progressbar(100, 2)
        self.assertEqual(
            self.mockfile.read(), "\r[002/100][..........] "
        )

    def test_progress_50_100(self):
        svtplay_dl.output.progressbar(100, 50)
        self.assertEqual(
            self.mockfile.read(), "\r[050/100][=====.....] "
        )

    def test_progress_100_100(self):
        svtplay_dl.output.progressbar(100, 100)
        self.assertEqual(
            self.mockfile.read(), "\r[100/100][==========] "
        )

    def test_progress_20_100_msg(self):
        svtplay_dl.output.progressbar(100, 20, "msg")
        self.assertEqual(
            self.mockfile.read(), "\r[020/100][==........] msg"
        )

    def test_progress_20_100_termwidth(self):
        svtplay_dl.output.get_terminal_size = lambda: (75, 25)
        svtplay_dl.output.progressbar(100, 20)
        self.assertEqual(
            self.mockfile.read(),
            "\r[020/100][=======............................] "
        )


class EtaTest(unittest.TestCase):
    @patch('time.time')
    def test_eta_0_100(self, mock_time):
        mock_time.return_value = float(0)

        # Let's make this simple; we'll create something that
        # processes one item per second, and make the size be
        # 100.
        eta = svtplay_dl.output.ETA(100)
        self.assertEqual(eta.left, 100) # no progress yet
        self.assertEqual(str(eta), "(unknown)") # no progress yet

        mock_time.return_value = float(10) # sleep(10)
        eta.update(10)
        self.assertEqual(eta.left, 90)
        self.assertEqual(str(eta), "0:01:30") # 90 items left, 90s left

        mock_time.return_value += 1
        eta.increment() # another item completed in one second!
        self.assertEqual(eta.left, 89)
        self.assertEqual(str(eta), "0:01:29")

        mock_time.return_value += 9
        eta.increment(9) # another item completed in one second!
        self.assertEqual(eta.left, 80)
        self.assertEqual(str(eta), "0:01:20")

        mock_time.return_value = float(90) # sleep(79)
        eta.update(90)
        self.assertEqual(eta.left, 10)
        self.assertEqual(str(eta), "0:00:10")