diff --git a/test/test_unit_query_distro.py b/test/test_unit_query_distro.py index e83ff4e..4f46501 100644 --- a/test/test_unit_query_distro.py +++ b/test/test_unit_query_distro.py @@ -3,14 +3,16 @@ import pytest @pytest.mark.parametrize( - 'condition', ['lsb_release', 'os-release', 'missing']) + 'condition', ['lsb_release', 'os-release', 'os-release-quotes', '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 'os-release' in condition: + quotes = '"' if 'quotes' in condition else '' + os_release.write_text( + f"testing\nID={quotes}{test_release}{quotes}\nrelease") if condition != 'lsb_release': lsb_release = 'LSB_RELEASE_PROGRAM="missing_lsb_release"' script = f""" @@ -24,7 +26,7 @@ def test_query_distro(runner, yadm, tst_distro, tmp_path, condition): assert run.err == '' if condition == 'lsb_release': assert run.out.rstrip() == tst_distro - elif condition == 'os-release': + elif 'os-release' in condition: assert run.out.rstrip() == test_release else: assert run.out.rstrip() == ''