mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-28 06:04:17 +01:00
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
#!/usr/bin/python
|
|
from __future__ import absolute_import
|
|
import unittest
|
|
import svtplay.output
|
|
|
|
class mockfile(object):
|
|
def __init__(self):
|
|
self.content = []
|
|
|
|
def write(self, string):
|
|
self.content.append(string)
|
|
|
|
def read(self):
|
|
return self.content.pop()
|
|
|
|
class progressbarTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.mockfile = mockfile()
|
|
svtplay.output.progress_stream = self.mockfile
|
|
|
|
def test_0_100(self):
|
|
svtplay.output.progressbar(100, 0)
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[000/100][..................................................] "
|
|
)
|
|
|
|
def test_progress_1_100(self):
|
|
svtplay.output.progressbar(100, 1)
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[001/100][..................................................] "
|
|
)
|
|
|
|
def test_progress_2_100(self):
|
|
svtplay.output.progressbar(100, 2)
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[002/100][=.................................................] "
|
|
)
|
|
|
|
def test_progress_50_100(self):
|
|
svtplay.output.progressbar(100, 50)
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[050/100][=========================.........................] "
|
|
)
|
|
|
|
def test_progress_100_100(self):
|
|
svtplay.output.progressbar(100, 100)
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[100/100][==================================================] "
|
|
)
|
|
|
|
def test_progress_20_100_msg(self):
|
|
svtplay.output.progressbar(100, 20, "msg")
|
|
self.assertEqual(
|
|
self.mockfile.read(),
|
|
"\r[020/100][==========........................................] msg"
|
|
)
|