mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-25 12:45:42 +01:00
a6ec593c0a
The naming conventions of the unittest module does not adhere to Python coding standards as specified in PEP8. This isn't our fault however...
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
#!/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.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"
|
|
)
|