def test_normal_flow_unknown_os(self, is_suse_family_mock,
is_redhat_family_mock, is_ubuntu_family_mock):
is_ubuntu_family_mock.return_value = False
is_redhat_family_mock.return_value = False
is_suse_family_mock.return_value = False
try:
list_ambari_managed_repos()
self.fail("Should throw a Fail")
except Fail:
pass # Expected
def test_normal_flow_ubuntu(self, is_suse_family_mock,
is_redhat_family_mock, is_ubuntu_family_mock,
glob_mock):
is_ubuntu_family_mock.return_value = True
is_redhat_family_mock.return_value = False
is_suse_family_mock.return_value = False
glob_mock.side_effect = \
[
[
"/etc/apt/sources.list.d/HDP-1.1.1.repo",
"/etc/apt/sources.list.d/HDP-1.1.2.repo",
"/etc/apt/sources.list.d/HDP-1.1.3.repo",
"/etc/apt/sources.list.d/HDP-UTILS-1.1.3.repo",
],
[
"/etc/apt/sources.list.d/HDP-UTILS-1.1.3.repo",
],
[]
]
res = list_ambari_managed_repos('HDP')
self.assertEquals(glob_mock.call_args_list[0][0][0],
"/etc/apt/sources.list.d/HDP*")
self.assertEquals(
res, ['HDP-1.1.1', 'HDP-1.1.2', 'HDP-1.1.3', 'HDP-UTILS-1.1.3'])
self.assertTrue(glob_mock.call_count > 1)
def test_normal_flow_sles(self, is_suse_family_mock,
is_redhat_family_mock, is_ubuntu_family_mock, glob_mock):
is_ubuntu_family_mock.return_value = False
is_redhat_family_mock.return_value = False
is_suse_family_mock.return_value = True
glob_mock.side_effect = \
[
[
"/etc/zypp/repos.d/HDP-1.1.1.repo",
"/etc/zypp/repos.d/HDP-1.1.2.repo",
"/etc/zypp/repos.d/HDP-1.1.3.repo",
"/etc/zypp/repos.d/HDP-UTILS-1.1.3.repo",
],
[
"/etc/zypp/repos.d/HDP-UTILS-1.1.3.repo",
],
[]
]
res = list_ambari_managed_repos()
self.assertEquals(glob_mock.call_args_list[0][0][0], "/etc/zypp/repos.d/HDP*")
self.assertEquals(res, ['HDP-1.1.1', 'HDP-1.1.2', 'HDP-1.1.3', 'HDP-UTILS-1.1.3'])
self.assertTrue(glob_mock.call_count > 1)
