test_data.py
No OneTemporary
Actions

Size

2 KB

Subscribers

None

test_data.py
View Options

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

	from pathlib import Path

	from flask import url_for
	import pytest

	from swh.model.exceptions import ValidationError
	from swh.scanner.client import Client
	from swh.scanner.data import (
	MerkleNodeInfo,
	add_origin,
	directory_content,
	get_directory_data,
	has_dirs,
	init_merkle_node_info,
	)

	from .data import fake_origin


	def test_merkle_node_data_wrong_args():
	nodes_data = MerkleNodeInfo()

	with pytest.raises(ValidationError):
	nodes_data["wrong key"] = {"known": True}

	with pytest.raises(ValidationError):
	nodes_data["swh:1:dir:17d207da3804cc60a77cba58e76c3b2f767cb112"] = "wrong value"


	def test_init_merkle_supported_node_info(source_tree):
	nodes_data = MerkleNodeInfo()
	init_merkle_node_info(source_tree, nodes_data, {"known", "origin"})
	for _, node_attrs in nodes_data.items():
	assert "known" and "origin" in node_attrs.keys()


	def test_init_merkle_not_supported_node_info(source_tree):
	nodes_data = MerkleNodeInfo()
	with pytest.raises(Exception):
	init_merkle_node_info(source_tree, nodes_data, {"unsupported_info"})


	def test_add_origin(event_loop, live_server, aiosession, source_tree, nodes_data):
	api_url = url_for("index", _external=True)
	init_merkle_node_info(source_tree, nodes_data, {"known", "origin"})
	client = Client(api_url, aiosession)

	event_loop.run_until_complete(add_origin(source_tree, nodes_data, client))
	for node, attrs in nodes_data.items():
	assert attrs["origin"] == fake_origin[str(source_tree.swhid())]


	def test_get_directory_data(source_tree, nodes_data):
	root = Path(source_tree.data["path"].decode())
	dirs_data = get_directory_data(root, source_tree, nodes_data)

	assert len(dirs_data) == 5


	def test_directory_content(source_tree, nodes_data):
	foo_dir = source_tree[b"foo"]
	foo_content = directory_content(foo_dir, nodes_data)
	assert foo_content[0] == 3
	assert foo_content[1] == 3


	def test_has_dirs(source_tree):
	assert has_dirs(source_tree)

File Metadata

Mime Type: text/x-python
Expires: Thu, Jul 3, 10:48 AM (1 w, 5 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3451972

test_data.pyNo OneTemporaryActions

test_data.pyView Options

File Metadata

Event Timeline

test_data.py
No OneTemporary
Actions

test_data.py
View Options