def test_defined_weeks():
progress_bar = ProgressBarYum(2000000000, 'file.iso')
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 1
assert 'file.iso 0% [ ] --- KiB/s | 1.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 2
assert 'file.i 0% [ ] 0.5 B/s | 2.0 B 1111111:06:36 ETA' == str(progress_bar)
def test_defined_rounded():
eta._NOW = lambda: 1411868723.5
progress_bar = ProgressBarYum(1023, 'long_file_name.iso')
assert 'long_fil 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 1022
assert 'long_fil 99% [=========-] --- KiB/s | 1.0 KiB ' == str(progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 1023
assert 'long_file_name.iso | 1.0 KiB 00:00:01 ' == str(progress_bar)
def test_defined_weeks():
progress_bar = ProgressBarYum(2000000000, 'file.iso')
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 1
assert 'file.iso 0% [ ] --- KiB/s | 1.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 2
assert 'file.i 0% [ ] 0.5 B/s | 2.0 B 1111111:06:36 ETA' == str(
progress_bar)
def progress_bar_yum():
files = {
'CentOS-7.0-1406-x86_64-DVD.iso': 10 if OPTIONS['--fast'] else 4148166656,
'CentOS-7.0-1406-x86_64-Everything.iso': 15 if OPTIONS['--fast'] else 7062159360,
'md5sum.txt': 5 if OPTIONS['--fast'] else 486,
}
for file_name, file_size in files.items():
bar = ProgressBarYum(0 if OPTIONS['--undefined'] else file_size, file_name)
for i in range(0, file_size + 1, 2 if OPTIONS['--fast'] else int(file_size / 100.0)):
bar.numerator = i
print(bar, end='\r')
sys.stdout.flush()
time.sleep(0.25)
bar.numerator = file_size
bar.force_done = True
print(bar)
def test_defined_rounded():
eta._NOW = lambda: 1411868723.5
progress_bar = ProgressBarYum(1023, 'long_file_name.iso')
assert 'long_fil 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 1022
assert 'long_fil 99% [=========-] --- KiB/s | 1.0 KiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 1023
assert 'long_file_name.iso | 1.0 KiB 00:00:01 ' == str(
progress_bar)
def test_defined():
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(2000, 'file.iso')
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 102
assert 'file.iso 5% [- ] --- KiB/s | 102 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 281
assert 'file.iso 14% [= ] 358 B/s | 281 B 00:00:05 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 593
assert 'file.iso 29% [==- ] 491 B/s | 593 B 00:00:03 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 1925
assert 'file.iso 96% [=========-] 1.1 KiB/s | 1.9 KiB 00:00:01 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 1999
assert 'file.iso 99% [=========-] 1.1 KiB/s | 2.0 KiB 00:00:01 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 2000
assert 'file.iso | 2.0 KiB 00:00:03 ' == str(progress_bar)
def test_defined_wont_fit():
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=33)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert 'file.iso | 0.0 B 00:00:00 ' == str(progress_bar)
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=30)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert 'file.i | 0.0 B 00:00:00 ' == str(progress_bar)
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=20)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert ' | 0.0 B 00:00:00 ' == str(progress_bar)
def progress_bar_yum():
files = {
'CentOS-7.0-1406-x86_64-DVD.iso':
10 if OPTIONS['--fast'] else 4148166656,
'CentOS-7.0-1406-x86_64-Everything.iso':
15 if OPTIONS['--fast'] else 7062159360,
'md5sum.txt': 5 if OPTIONS['--fast'] else 486,
}
for file_name, file_size in files.items():
bar = ProgressBarYum(0 if OPTIONS['--undefined'] else file_size,
file_name)
for i in range(0, file_size + 1,
2 if OPTIONS['--fast'] else int(file_size / 100.0)):
bar.numerator = i
print(bar, end='\r')
sys.stdout.flush()
time.sleep(0.25)
bar.numerator = file_size
bar.force_done = True
print(bar)
def test_defined():
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(2000, 'file.iso')
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert 'file.iso 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 102
assert 'file.iso 5% [- ] --- KiB/s | 102 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 281
assert 'file.iso 14% [= ] 358 B/s | 281 B 00:00:05 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 593
assert 'file.iso 29% [==- ] 491 B/s | 593 B 00:00:03 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 1925
assert 'file.iso 96% [=========-] 1.1 KiB/s | 1.9 KiB 00:00:01 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 1999
assert 'file.iso 99% [=========-] 1.1 KiB/s | 2.0 KiB 00:00:01 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 2000
assert 'file.iso | 2.0 KiB 00:00:03 ' == str(
progress_bar)
def test_defined_wont_fit():
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=33)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert 'file.iso | 0.0 B 00:00:00 ' == str(progress_bar)
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=30)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert 'file.i | 0.0 B 00:00:00 ' == str(progress_bar)
progress_bar = ProgressBarYum(2000, 'file.iso', max_width=20)
assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar)
progress_bar.force_done = True
assert ' | 0.0 B 00:00:00 ' == str(progress_bar)
def test_undefined():
misc.terminal_width = lambda: 60
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(None, '', max_width=55)
assert ' [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert ' [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert ' [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 10
assert ' [ ] --- KiB/s | 10 B ' == str(
progress_bar)
assert ' [ ] --- KiB/s | 10 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 100
assert ' [ ] 180 B/s | 100 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 1954727
assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(
progress_bar)
assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 4217583
assert ' [ ] 2.8 MiB/s | 4.0 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 6826725
assert ' [ ] 3.4 MiB/s | 6.5 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 8659265
assert ' [ ] 3.6 MiB/s | 8.3 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868725.0
progress_bar.numerator = 8659265
assert ' [ ] 3.3 MiB/s | 8.3 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868725.5
progress_bar.numerator = 21057295
assert ' [ ] 4.8 MiB/s | 20 MiB ' == str(
progress_bar)
eta._NOW = lambda: 1411868726.0
progress_bar.numerator = 65572196
assert ' [ ] 10 MiB/s | 62 MiB ' == str(
progress_bar)
progress_bar.force_done = True
assert ' | 62 MiB 00:00:05 ' == str(
progress_bar)
def test_defined_long():
misc.terminal_width = lambda: 50
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(20, 'a.iso')
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 1
assert 'a.is 5% [ ] --- KiB/s | 1.0 B ' == str(
progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 2
assert 'a.is 10% [ ] 2.0 B/s | 2.0 B 00:00:09 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 3
assert 'a.is 15% [- ] 2.0 B/s | 3.0 B 00:00:09 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 4
assert 'a.is 20% [- ] 2.0 B/s | 4.0 B 00:00:08 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 5
assert 'a.is 25% [= ] 2.0 B/s | 5.0 B 00:00:08 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 6
assert 'a.is 30% [= ] 2.0 B/s | 6.0 B 00:00:07 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868725.0
progress_bar.numerator = 7
assert 'a.is 35% [= ] 2.0 B/s | 7.0 B 00:00:07 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868725.5
progress_bar.numerator = 8
assert 'a.is 40% [=- ] 2.0 B/s | 8.0 B 00:00:06 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868726.0
progress_bar.numerator = 9
assert 'a.is 45% [=- ] 2.0 B/s | 9.0 B 00:00:06 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868726.5
progress_bar.numerator = 10
assert 'a.is 50% [== ] 2.0 B/s | 10 B 00:00:05 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868727.0
progress_bar.numerator = 11
assert 'a.is 55% [== ] 2.0 B/s | 11 B 00:00:05 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868727.5
progress_bar.numerator = 12
assert 'a.is 60% [== ] 2.0 B/s | 12 B 00:00:04 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868728.0
progress_bar.numerator = 13
assert 'a.is 65% [==- ] 2.0 B/s | 13 B 00:00:04 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868728.5
progress_bar.numerator = 14
assert 'a.is 70% [==- ] 2.0 B/s | 14 B 00:00:03 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868729.0
progress_bar.numerator = 15
assert 'a.is 75% [=== ] 2.0 B/s | 15 B 00:00:03 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868729.5
progress_bar.numerator = 16
assert 'a.is 80% [=== ] 2.0 B/s | 16 B 00:00:02 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868730.0
progress_bar.numerator = 17
assert 'a.is 85% [=== ] 2.0 B/s | 17 B 00:00:02 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868730.5
progress_bar.numerator = 18
assert 'a.is 90% [===-] 2.0 B/s | 18 B 00:00:01 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868731.0
progress_bar.numerator = 19
assert 'a.is 95% [===-] 2.0 B/s | 19 B 00:00:01 ETA' == str(
progress_bar)
eta._NOW = lambda: 1411868731.5
progress_bar.numerator = 20
assert 'a.iso | 20 B 00:00:10 ' == str(
progress_bar)
def test_undefined():
misc.terminal_width = lambda: 60
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(None, '', max_width=55)
assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 10
assert ' [ ] --- KiB/s | 10 B ' == str(progress_bar)
assert ' [ ] --- KiB/s | 10 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 100
assert ' [ ] 180 B/s | 100 B ' == str(progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 1954727
assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(progress_bar)
assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 4217583
assert ' [ ] 2.8 MiB/s | 4.0 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 6826725
assert ' [ ] 3.4 MiB/s | 6.5 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 8659265
assert ' [ ] 3.6 MiB/s | 8.3 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868725.0
progress_bar.numerator = 8659265
assert ' [ ] 3.3 MiB/s | 8.3 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868725.5
progress_bar.numerator = 21057295
assert ' [ ] 4.8 MiB/s | 20 MiB ' == str(progress_bar)
eta._NOW = lambda: 1411868726.0
progress_bar.numerator = 65572196
assert ' [ ] 10 MiB/s | 62 MiB ' == str(progress_bar)
progress_bar.force_done = True
assert ' | 62 MiB 00:00:05 ' == str(progress_bar)
def test_defined_long():
misc.terminal_width = lambda: 50
eta._NOW = lambda: 1411868721.5
progress_bar = ProgressBarYum(20, 'a.iso')
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
assert 'a.is 0% [ ] --- KiB/s | 0.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.0
progress_bar.numerator = 1
assert 'a.is 5% [ ] --- KiB/s | 1.0 B ' == str(progress_bar)
eta._NOW = lambda: 1411868722.5
progress_bar.numerator = 2
assert 'a.is 10% [ ] 2.0 B/s | 2.0 B 00:00:09 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868723.0
progress_bar.numerator = 3
assert 'a.is 15% [- ] 2.0 B/s | 3.0 B 00:00:09 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868723.5
progress_bar.numerator = 4
assert 'a.is 20% [- ] 2.0 B/s | 4.0 B 00:00:08 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868724.0
progress_bar.numerator = 5
assert 'a.is 25% [= ] 2.0 B/s | 5.0 B 00:00:08 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868724.5
progress_bar.numerator = 6
assert 'a.is 30% [= ] 2.0 B/s | 6.0 B 00:00:07 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868725.0
progress_bar.numerator = 7
assert 'a.is 35% [= ] 2.0 B/s | 7.0 B 00:00:07 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868725.5
progress_bar.numerator = 8
assert 'a.is 40% [=- ] 2.0 B/s | 8.0 B 00:00:06 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868726.0
progress_bar.numerator = 9
assert 'a.is 45% [=- ] 2.0 B/s | 9.0 B 00:00:06 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868726.5
progress_bar.numerator = 10
assert 'a.is 50% [== ] 2.0 B/s | 10 B 00:00:05 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868727.0
progress_bar.numerator = 11
assert 'a.is 55% [== ] 2.0 B/s | 11 B 00:00:05 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868727.5
progress_bar.numerator = 12
assert 'a.is 60% [== ] 2.0 B/s | 12 B 00:00:04 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868728.0
progress_bar.numerator = 13
assert 'a.is 65% [==- ] 2.0 B/s | 13 B 00:00:04 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868728.5
progress_bar.numerator = 14
assert 'a.is 70% [==- ] 2.0 B/s | 14 B 00:00:03 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868729.0
progress_bar.numerator = 15
assert 'a.is 75% [=== ] 2.0 B/s | 15 B 00:00:03 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868729.5
progress_bar.numerator = 16
assert 'a.is 80% [=== ] 2.0 B/s | 16 B 00:00:02 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868730.0
progress_bar.numerator = 17
assert 'a.is 85% [=== ] 2.0 B/s | 17 B 00:00:02 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868730.5
progress_bar.numerator = 18
assert 'a.is 90% [===-] 2.0 B/s | 18 B 00:00:01 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868731.0
progress_bar.numerator = 19
assert 'a.is 95% [===-] 2.0 B/s | 19 B 00:00:01 ETA' == str(progress_bar)
eta._NOW = lambda: 1411868731.5
progress_bar.numerator = 20
assert 'a.iso | 20 B 00:00:10 ' == str(progress_bar)
