Current File : //snap/core22/current/usr/share/subiquity/subiquitycore/tests/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)
Mini Shell