def test_tinypng_process():
(fake_fs, fake_os, fake_open) = init_filesystem()
utils.os = fake_os
utils.open = fake_open
pytinypng.os = fake_os
with success_result():
handler = TestHandler()
pytinypng.process_directory('/input', '/output', '12345', handler)
assert fake_os.path.exists('/output/subdir/a/bullet.png')
assert handler.post_no == 2
def test_tinypng_process_allow_overwrite():
(fake_fs, fake_os, fake_open) = init_filesystem()
utils.os = fake_os
utils.open = fake_open
pytinypng.os = fake_os
pytinypng.open = fake_open
fake_fs.CreateFile("/output/subdir/a/bullet.png")
with success_result():
handler = TestHandler()
pytinypng.process_directory('/input', '/output', '12345', handler)
assert handler.post_no == 1
def test_tinypng_process_stop():
def fake_compress(*args):
return TinyPNGResponse(500, error='TooManyRequests', message='...')
shrink = fake_compress
(fake_fs, fake_os, fake_open) = init_filesystem()
utils.os = fake_os
utils.open = fake_open
pytinypng.os = fake_os
handler = TestHandler()
pytinypng.process_directory('/input', '/output', '12345', handler)
assert handler.post_no == 1
def test_tinypng_process_retry():
def fake_compress(*args):
return TinyPNGResponse(500, error='InternalServerError', message='...')
pytinypng.TINYPNG_SLEEP_SEC = 0
shrink = fake_compress
(fake_fs, fake_os, fake_open) = init_filesystem()
utils.os = fake_os
utils.open = fake_open
pytinypng.os = fake_os
handler = TestHandler()
pytinypng.process_directory('/input', '/output', '12345', handler)
assert handler.post_no == 1