def test_read_error(tmpdir):
"""Test disc opening error handling.
:param py.path.local tmpdir: pytest fixture.
"""
output = tmpdir.ensure_dir('output')
with pytest.raises(subprocess.CalledProcessError) as exc:
pytest.run(output=output)
assert b'Failed to open disc' in exc.value.output
pytest.verify_failed_file(output)
def test_rip_error(request, tmpdir):
"""Test failed rips error handling.
:param request: pytest fixture.
:param py.path.local tmpdir: pytest fixture.
"""
# Duplicate ISO.
iso = tmpdir.join('truncated.iso')
py.path.local(__file__).dirpath().join('sample.iso').copy(iso)
# Load the ISO.
pytest.cdload(iso)
# Execute.
output = tmpdir.ensure_dir('output')
command = [
'docker', 'run', '--device=/dev/cdrom', '-v',
'{}:/output'.format(output), '-e', 'DEBUG=true', 'robpol86/makemkv'
]
master, slave = pty.openpty()
request.addfinalizer(lambda: [os.close(master), os.close(slave)])
proc = subprocess.Popen(command,
bufsize=1,
cwd=HERE,
stderr=subprocess.STDOUT,
stdin=slave,
stdout=subprocess.PIPE)
# Read output.
caught = False
while proc.poll() is None or proc.stdout.peek(1):
for line in proc.stdout:
# Watch for specific line, then truncate ISO.
if b'Analyzing seamless segments' in line:
iso.open('w').close()
elif b'ERROR: One or more titles failed.' in line:
caught = True
print(line) # Write to stdout, pytest will print if test fails.
# Verify.
assert proc.poll() > 0
assert caught is True
pytest.verify_failed_file(output)
def test_low_space(request, tmpdir):
"""Test low free space handling.
:param request: pytest fixture.
:param py.path.local tmpdir: pytest fixture.
"""
# Create a 5 MiB filesystem container.
fs_bin = tmpdir.join('fs.bin')
with fs_bin.open('w') as handle:
handle.truncate(1024 * 1024 * 5)
pytest.run(['mkfs.ext4', '-F', fs_bin])
# Mount the filesystem.
output = tmpdir.ensure_dir('output')
pytest.run(['sudo', 'mount', '-t', 'ext4', '-o', 'loop', fs_bin, output])
request.addfinalizer(lambda: pytest.run(['sudo', 'umount', output]))
# Docker run.
with pytest.raises(subprocess.CalledProcessError) as exc:
pytest.run(output=output)
assert b'Terminating MakeMKV due to low disk space.' in exc.value.stderr
pytest.verify_failed_file(output)
def test_failed_eject(tmpdir, failed_eject):
"""Test FAILED_EJECT environment variable.
:param py.path.local tmpdir: pytest fixture.
:param bool failed_eject: Set environment variable to 'true', 'false', or don't set.
"""
output = tmpdir.ensure_dir('output')
if failed_eject is True:
args = ['-e', 'FAILED_EJECT=true']
elif failed_eject is False:
args = ['-e', 'FAILED_EJECT=false']
else:
args = list()
# Docker run.
with pytest.raises(subprocess.CalledProcessError) as exc:
pytest.run(args=args, output=output)
# Verify.
if failed_eject is True:
assert b'\nEjecting due to failure...' in exc.value.output
else:
assert b'\nEjecting' not in exc.value.output
pytest.verify_failed_file(output)