Differential D3070 Diff 10924 swh/scanner/tests/test_model.py

Changeset View

Standalone View

swh/scanner/tests/test_model.py

	# Copyright (C) 2020 The Software Heritage developers			# Copyright (C) 2020 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution			# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version			# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information			# See top-level LICENSE file for more information


	def test_tree_add_node(example_tree, temp_folder):			def test_tree_add_node(example_tree, temp_folder):
	avail_paths = temp_folder["paths"].keys()			avail_paths = temp_folder["paths"].keys()

	for path, pid in temp_folder["paths"].items():			for path, pid in temp_folder["paths"].items():
	example_tree.addNode(path, pid)			example_tree.addNode(path, pid, False)

	for path, node in example_tree.children.items():			for path, node in example_tree.children.items():
	assert path in avail_paths			assert path in avail_paths
	if node.children:			if node.children:
	for subpath, subnode in node.children.items():			for subpath, subnode in node.children.items():
	assert subpath in avail_paths			assert subpath in avail_paths


	def test_get_json_tree_all_not_present(example_tree, temp_folder):			def test_to_json_no_one_present(example_tree, temp_folder):
	for path, pid in temp_folder["paths"].items():			for path, pid in temp_folder["paths"].items():
	example_tree.addNode(path)			example_tree.addNode(path, pid, False)

	json_tree = example_tree.getTree()			result = example_tree.toDict()

	assert len(json_tree) == 0			assert len(result) == 6

				for _, node_info in result.items():
				assert node_info["known"] is False


	def test_get_json_tree_all_present(example_tree, temp_folder):			def test_get_json_tree_all_present(example_tree, temp_folder):
	for path, pid in temp_folder["paths"].items():			for path, pid in temp_folder["paths"].items():
	example_tree.addNode(path, pid)			example_tree.addNode(path, pid, True)

				result = example_tree.toDict()

	tree_dict = example_tree.getTree()			assert len(result) == 6

	assert len(tree_dict) == 3			for _, node_info in result.items():
	# since subdir have a pid, it can't have a children path			assert node_info["known"] is True
	assert tree_dict["subdir0"] is not dict


	def test_get_json_tree_only_one_present(example_tree, temp_folder):			def test_get_json_tree_only_one_present(example_tree, temp_folder):
				root = temp_folder["root"]
	filesample_path = temp_folder["filesample"]			filesample_path = temp_folder["filesample"]

	for path, pid in temp_folder["paths"].items():			for path, pid in temp_folder["paths"].items():
	if path == filesample_path:			example_tree.addNode(path, pid, True if path == filesample_path else False)
	example_tree.addNode(path, pid)
	else:
	example_tree.addNode(path)

	tree_dict = example_tree.getTree()			result = example_tree.toDict()

	assert len(tree_dict) == 1			assert len(result) == 6
	assert tree_dict["subdir0"]["filesample.txt"]			assert result[str(root) + "/subdir0/filesample.txt"]["known"] is True
				vlorentzUnsubmitted Not Done Inline Actions should also check the other ones are False vlorentz: should also check the other ones are False


	def test_get_directories_info(example_tree, temp_folder):			def test_get_directories_info(example_tree, temp_folder):
	root_path = temp_folder["root"]			root_path = temp_folder["root"]
	filesample_path = temp_folder["filesample"]			filesample_path = temp_folder["filesample"]
	filesample2_path = temp_folder["filesample2"]			filesample2_path = temp_folder["filesample2"]
	subdir_path = temp_folder["subdir"].relative_to(root_path)			subdir_path = temp_folder["subdir"].relative_to(root_path)
	subsubdir_path = temp_folder["subsubdir"].relative_to(root_path)			subsubdir_path = temp_folder["subsubdir"].relative_to(root_path)

	for path, pid in temp_folder["paths"].items():			for path, pid in temp_folder["paths"].items():
	if path == filesample_path or path == filesample2_path:			if path == filesample_path or path == filesample2_path:
	example_tree.addNode(path, pid)			example_tree.addNode(path, pid, True)
	else:			else:
	example_tree.addNode(path)			example_tree.addNode(path, pid, False)

	directories = example_tree.getDirectoriesInfo(example_tree.path)			directories = example_tree.getDirectoriesInfo(example_tree.path)

	assert subsubdir_path not in directories			assert subsubdir_path not in directories
	assert directories[subdir_path] == (2, 2)			assert directories[subdir_path] == (2, 2)