47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from __future__ import absolute_import
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
import six
|
|
import unittest
|
|
|
|
|
|
from koji_cli.lib import _format_size, _format_secs, _progress_callback
|
|
|
|
class TestUploadProgressCallBack(unittest.TestCase):
|
|
|
|
maxDiff = None
|
|
|
|
def test_format_size(self):
|
|
self.assertEqual(_format_size(2000000000), '1.86 GiB')
|
|
self.assertEqual(_format_size(1073741824), '1.00 GiB')
|
|
self.assertEqual(_format_size(3000000), '2.86 MiB')
|
|
self.assertEqual(_format_size(1048576), '1.00 MiB')
|
|
self.assertEqual(_format_size(4000), '3.91 KiB')
|
|
self.assertEqual(_format_size(1024), '1.00 KiB')
|
|
self.assertEqual(_format_size(500), '500.00 B')
|
|
|
|
def test_format_secs(self):
|
|
self.assertEqual(_format_secs(0), '00:00:00')
|
|
self.assertEqual(_format_secs(60), '00:01:00')
|
|
self.assertEqual(_format_secs(3600), '01:00:00')
|
|
self.assertEqual(_format_secs(7283294), '2023:08:14')
|
|
self.assertEqual(_format_secs(1234), '00:20:34')
|
|
self.assertEqual(_format_secs(4321), '01:12:01')
|
|
self.assertEqual(_format_secs(4321.567), '01:12:01')
|
|
|
|
@mock.patch('sys.stdout', new_callable=six.StringIO)
|
|
def test_progress_callback(self, stdout):
|
|
_progress_callback(12300, 234000, 5670, 80, 900)
|
|
_progress_callback(45600, 234000, 5670, 0, 900)
|
|
_progress_callback(234000, 234000, 5670, 80, 900)
|
|
self.assertMultiLineEqual(
|
|
stdout.getvalue(),
|
|
'[= ] 05% 00:15:00 12.01 KiB 70.88 B/sec\r'
|
|
'[======= ] 19% 00:15:00 44.53 KiB - B/sec\r'
|
|
'[====================================] 100% 00:15:00 228.52 KiB 260.00 B/sec\r')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|