Differential D8231 Diff 29706 swh/graphql/tests/functional/test_swhid_search.py

Changeset View

Standalone View

swh/graphql/tests/functional/test_swhid_search.py

This file was added.

				# Copyright (C) 2022 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

				import pytest

				from . import utils
				from ..data import get_directories, get_releases, get_revisions, get_snapshots


				def test_invalid_swhid(client):
				query_str = """
				{
				searchSwhid(swhid: "swh:1:dir:dae0d245988b472abd30a4f968b919d0019b6c7") {
				nodes {
				type
				}
				}
				}
				"""
				errors = utils.get_error_response(client, query_str)
				# API will throw an error in case of an invalid SWHID
				assert len(errors) == 1
				assert "Invalid SWHID: invalid syntax" in errors[0]["message"]


				@pytest.mark.parametrize(
				"swhid",
				[
				"swh:1:rel:0949d7a8c96347dba09be8d79085b8207f345412",
				"swh:1:rev:0949d7a8c96347dba09be8d79085b8207f345412",
				"swh:1:dir:0949d7a8c96347dba09be8d79085b8207f345412",
				"swh:1:cnt:0949d7a8c96347dba09be8d79085b8207f345412",
				"swh:1:snp:0949d7a8c96347dba09be8d79085b8207f345412",
				],
				)
				def test_missing_swhid(client, swhid):
				query_str = """
				{
				searchSwhid(swhid: "%s") {
				nodes {
				type
				}
				}
				}
				"""
				data, _ = utils.get_query_response(client, query_str % swhid)
				# API will return an empty list in case of a valid, non existing SWHID
				assert data == {"searchSwhid": {"nodes": []}}


				@pytest.mark.parametrize("snapshot", get_snapshots())
				def test_snapshot_swhid_search(client, snapshot):
				query_str = """
				{
				searchSwhid(swhid: "%s") {
				nodes {
				type
				match
				target {
				__typename
				... on Snapshot {
				swhid
				}
				}
				}
				}
				}
				"""
				data, _ = utils.get_query_response(client, query_str % snapshot.swhid())
				assert data == {
				"searchSwhid": {
				"nodes": [
				{
				"match": 100,
				"target": {
				"__typename": "Snapshot",
				"swhid": str(snapshot.swhid()),
				},
				"type": "snapshot",
				}
				]
				}
				}


				@pytest.mark.parametrize("revision", get_revisions())
				def test_revision_swhid_search(client, revision):
				query_str = """
				{
				searchSwhid(swhid: "%s") {
				nodes {
				type
				match
				target {
				__typename
				... on Revision {
				swhid
				}
				}
				}
				}
				}
				"""
				data, _ = utils.get_query_response(client, query_str % revision.swhid())
				assert data == {
				"searchSwhid": {
				"nodes": [
				{
				"match": 100,
				"target": {
				"__typename": "Revision",
				"swhid": str(revision.swhid()),
				},
				"type": "revision",
				}
				]
				}
				}


				@pytest.mark.parametrize("release", get_releases())
				def test_release_swhid_search(client, release):
				query_str = """
				{
				searchSwhid(swhid: "%s") {
				nodes {
				type
				match
				target {
				__typename
				... on Release {
				swhid
				}
				}
				}
				}
				}
				"""
				data, _ = utils.get_query_response(client, query_str % release.swhid())
				assert data == {
				"searchSwhid": {
				"nodes": [
				{
				"match": 100,
				"target": {
				"__typename": "Release",
				"swhid": str(release.swhid()),
				},
				"type": "release",
				}
				]
				}
				}


				@pytest.mark.parametrize("directory", get_directories())
				def test_directory_swhid_search(client, directory):
				query_str = """
				{
				searchSwhid(swhid: "%s") {
				nodes {
				type
				match
				target {
				__typename
				... on Directory {
				swhid
				}
				}
				}
				}
				}
				"""
				data, _ = utils.get_query_response(client, query_str % directory.swhid())
				assert data == {
				"searchSwhid": {
				"nodes": [
				{
				"match": 100,
				"target": {
				"__typename": "Directory",
				"swhid": str(directory.swhid()),
				},
				"type": "directory",
				}
				]
				}
				}