name : test_lsb_release.py # Copyright 2021 Canonical, Ltd. # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, version 3. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. import unittest from unittest.mock import patch, mock_open from subiquitycore.lsb_release import lsb_release class TestLSBRelease(unittest.TestCase): def setUp(self): self.target = "subiquitycore.lsb_release.open" def test_lsb_release(self): lsb_str = """ DISTRIB_ID=Ubuntu DISTRIB_RELEASE=21.10 DISTRIB_CODENAME=impish DISTRIB_DESCRIPTION="Ubuntu 21.10" """ with patch(self.target, mock_open(read_data=lsb_str)) as patched: distro = lsb_release(path="dummy") patched.assert_called_once_with("dummy", "r") self.assertEqual(distro["id"], "Ubuntu") self.assertEqual(distro["release"], "21.10") self.assertEqual(distro["codename"], "impish") self.assertEqual(distro["description"], "Ubuntu 21.10") def test_lsb_release_inexistent(self): with patch(self.target, side_effect=FileNotFoundError): self.assertEqual(lsb_release("/inexistent"), {}) def test_lsb_release_default(self): with patch(self.target, side_effect=FileNotFoundError) as patched: lsb_release(path=None) patched.assert_called_once_with("/etc/lsb-release", "r") def test_lsb_release_dry_run(self): with patch(self.target, side_effect=FileNotFoundError) as patched: lsb_release(dry_run=True) patched.assert_called_once_with("examples/lsb-release-focal", "r") def test_lsb_release_mutually_exclusive(self): with self.assertRaises(ValueError): lsb_release(path="dummy", dry_run=True)