def test_pkg_get(self): """ Test if grains switching the pkg get method. :return: """ debian_list = """ g++ g++-4.9 g++-5 gawk gcc gcc-4.9 gcc-4.9-base:amd64 gcc-4.9-base:i386 gcc-5 gcc-5-base:amd64 gcc-5-base:i386 gcc-6-base:amd64 gcc-6-base:i386 """ inspector = Inspector(cachedir="/test", piddir="/test", pidfilename="bar.pid") inspector.grains_core = MagicMock() inspector.grains_core.os_data = MagicMock() inspector.grains_core.os_data.get = MagicMock(return_value="Debian") with patch.object(inspector, "_Inspector__get_cfg_pkgs_dpkg", MagicMock(return_value="dpkg")): with patch.object(inspector, "_Inspector__get_cfg_pkgs_rpm", MagicMock(return_value="rpm")): inspector.grains_core = MagicMock() inspector.grains_core.os_data = MagicMock() inspector.grains_core.os_data().get = MagicMock( return_value="Debian") self.assertEqual(inspector._get_cfg_pkgs(), "dpkg") inspector.grains_core.os_data().get = MagicMock( return_value="Suse") self.assertEqual(inspector._get_cfg_pkgs(), "rpm") inspector.grains_core.os_data().get = MagicMock( return_value="redhat") self.assertEqual(inspector._get_cfg_pkgs(), "rpm")
def test_file_tree(self): ''' Test file tree. :return: ''' inspector = Inspector(cachedir='/test', piddir='/test', pidfilename='bar.pid') tree_root = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'inspectlib', 'tree_test') expected_tree = (['/a/a/dummy.a', '/a/b/dummy.b', '/b/b.1', '/b/b.2', '/b/b.3'], ['/a', '/a/a', '/a/b', '/a/c', '/b', '/c'], ['/a/a/dummy.ln.a', '/a/b/dummy.ln.b', '/a/c/b.1', '/b/b.4', '/b/b.5', '/c/b.1', '/c/b.2', '/c/b.3']) tree_result = [] for chunk in inspector._get_all_files(tree_root): buff = [] for pth in chunk: buff.append(pth.replace(tree_root, '')) tree_result.append(buff) tree_result = tuple(tree_result) self.assertEqual(expected_tree, tree_result)
def test_get_unmanaged_files(self): """ Test get_unmanaged_files. :return: """ inspector = Inspector(cachedir="/test", piddir="/test", pidfilename="bar.pid") managed = ( ["a", "b", "c"], ["d", "e", "f"], ["g", "h", "i"], ) system_all = ( ["a", "b", "c"], ["d", "E", "f"], ["G", "H", "i"], ) self.assertEqual( inspector._get_unmanaged_files(managed=managed, system_all=system_all), ([], ["E"], ["G", "H"]), )