def test_module_utils_basic_get_distribution_version(self): from ansible.module_utils.common.sys_info import get_distribution_version with patch('platform.system', return_value='Foo'): self.assertEqual(get_distribution_version(), None) with patch('platform.system', return_value='Linux'): with patch('platform.linux_distribution', return_value=("foo", "1", "One")): self.assertEqual(get_distribution_version(), "1") with patch('os.path.isfile', return_value=True): def _dist(distname='', version='', id='', supported_dists=(), full_distribution_name=1): if supported_dists != (): return ("AmazonFooBar", "2", "") else: return ("", "", "") with patch('platform.linux_distribution', side_effect=_dist): self.assertEqual(get_distribution_version(), "2") with patch('platform.linux_distribution', side_effect=Exception("boo")): with patch('platform.dist', return_value=("bar", "3", "Three")): self.assertEqual(get_distribution_version(), "3")
def test_linux_distribution(self): distro_linux_dist = (get_distribution(), get_distribution_version(), get_distribution_codename()) platform_linux_dist = platform.linux_distribution() assert isinstance(distro_linux_dist, type(platform_linux_dist)), \ 'linux_distribution() returned type (%s) which is different from platform.linux_distribution type (%s)' % \ (type(distro_linux_dist), type(platform_linux_dist)) # TODO: add the cases where we expect them to differ # The third item in the tuple is different. assert distro_linux_dist[0] == platform_linux_dist[0] assert distro_linux_dist[1] == platform_linux_dist[1] if platform_linux_dist[0] == 'Fedora' and 20 < int( platform_linux_dist[1]) < 28: pytest.skip( "Fedora versions between 20 and 28 return the variant instead of the code name making this test unreliable" ) # Fedora considers the platform_linux behaviour to have been a bug as it's finding the # variant, not the code name. Fedora wants this to be the empty string. platform_linux_dist = platform_linux_dist[:2] + ('', ) assert distro_linux_dist[2] == platform_linux_dist[2]
def _guess_distribution(self): # try to find out which linux distribution this is dist = (get_distribution(), get_distribution_version(), get_distribution_codename()) distribution_guess = { 'distribution': dist[0] or 'NA', 'distribution_version': dist[1] or 'NA', # distribution_release can be the empty string 'distribution_release': 'NA' if dist[2] is None else dist[2] } distribution_guess['distribution_major_version'] = distribution_guess['distribution_version'].split('.')[0] or 'NA' return distribution_guess
def test_distro_found(): with patch('ansible.module_utils.distro.version', return_value="1"): assert get_distribution_version() == "1"
def test_get_distribution_version_not_linux(): """If it's not Linux, then it has no distribution""" with patch('platform.system', return_value='Foo'): assert get_distribution_version() is None
def test_get_distribution_version_not_linux(mocker, system, version): """If it's not Linux, then it has no distribution""" mocker.patch('platform.system', return_value=system) mocker.patch('ansible.module_utils.common.sys_info.distro.version', return_value=version) assert get_distribution_version() == version