Use /etc/os-release if lsb_release is missing (#175)

test/test_unit_query_distro.py

@@ -1,26 +1,30 @@
 """Unit tests: query_distro"""
+import pytest
 
 
-def test_lsb_release_present(runner, yadm, tst_distro):
+@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 == ''
-    assert run.out.rstrip() == tst_distro
-
-
-def test_lsb_release_missing(runner, yadm):
-    """Empty value when missing"""
-    script = f"""
-        YADM_TEST=1 source {yadm}
-        LSB_RELEASE_PROGRAM="missing_lsb_release"
-        query_distro
-    """
-    run = runner(command=['bash'], inp=script)
-    assert run.success
-    assert run.err == ''
-    assert run.out.rstrip() == ''
+    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() == ''