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)