archey4/archey/test/entries/test_archey_distro.py

"""Test module for Archey's distribution detection module"""

import unittest
from unittest.mock import MagicMock, patch

from archey.configuration import DEFAULT_CONFIG
from archey.entries.distro import Distro
from archey.test.entries import HelperMethods


class TestDistroEntry(unittest.TestCase):
    """`Distro` entry simple test cases"""

    @patch(
        "archey.entries.distro.check_output",
        return_value="10\n",  # Imitate `getprop` output on Android 10.
    )
    def test_fetch_android_release(self, _):
        """Test `_fetch_android_release` static method"""
        self.assertEqual(
            Distro._fetch_android_release(),  # pylint: disable=protected-access
            "Android 10",
        )

    @patch(
        "archey.entries.distro.platform.mac_ver",
        side_effect=[
            ("", ("", "", ""), ""),  # Darwin case.
            ("11.1", ("foo", "bar", "baz"), "x86_64"),  # macOS case.
        ],
    )
    @patch(
        "archey.entries.distro.platform.release",
        return_value="20.2.0",  # Darwin release.
    )
    def test_fetch_darwin_release(self, _, __):
        """Test `_fetch_darwin_release` static method"""
        self.assertEqual(
            Distro._fetch_darwin_release(),  # pylint: disable=protected-access
            "Darwin 20.2.0",
        )
        self.assertEqual(
            Distro._fetch_darwin_release(),  # pylint: disable=protected-access
            "macOS 11.1",
        )

    @HelperMethods.patch_clean_configuration
    def test_unknown_distro_output(self):
        """Test for `output` method when distribution name couldn't be found"""
        distro_intance_mock = HelperMethods.entry_mock(Distro)
        output_mock = MagicMock()

        distro_intance_mock.value = {
            "name": None,
            "arch": "ARCHITECTURE",
        }

        Distro.output(distro_intance_mock, output_mock)
        self.assertEqual(
            output_mock.append.call_args[0][1],
            f"{DEFAULT_CONFIG['default_strings']['not_detected']} ARCHITECTURE",
        )


if __name__ == "__main__":
    unittest.main()