1
0
Fork 0
mirror of synced 2024-12-13 02:45:45 -05:00
yadm/test/test_unit_query_distro.py

30 lines
1,000 B
Python

"""Unit tests: query_distro"""
import pytest
@pytest.mark.parametrize(
'condition', ['lsb_release', 'os-release', 'missing'])
def test_query_distro(runner, yadm, tst_distro, tmp_path, condition):
"""Match lsb_release -si when present"""
test_release = 'testrelease'
lsb_release = ''
os_release = tmp_path.joinpath('os-release')
if condition == 'os-release':
os_release.write_text(f"testing\nID={test_release}\nrelease")
if condition != 'lsb_release':
lsb_release = 'LSB_RELEASE_PROGRAM="missing_lsb_release"'
script = f"""
YADM_TEST=1 source {yadm}
{lsb_release}
OS_RELEASE="{os_release}"
query_distro
"""
run = runner(command=['bash'], inp=script)
assert run.success
assert run.err == ''
if condition == 'lsb_release':
assert run.out.rstrip() == tst_distro
elif condition == 'os-release':
assert run.out.rstrip() == test_release
else:
assert run.out.rstrip() == ''