datatracker/ietf/doc/tests_models.py
2023-11-01 12:41:15 -05:00

114 lines
3.8 KiB
Python

# Copyright The IETF Trust 2016-2023, All Rights Reserved
# -*- coding: utf-8 -*-
import itertools
from ietf.doc.factories import WgRfcFactory
from ietf.doc.models import RelatedDocument
from ietf.utils.test_utils import TestCase
class RelatedDocumentTests(TestCase):
def test_is_downref(self):
rfcs = [
WgRfcFactory(std_level_id=lvl)
for lvl in ["inf", "exp", "bcp", "ps", "ds", "std", "unkn"]
]
result_matrix = {
# source
"inf": {
"inf": None, # target
"exp": None, # target
"bcp": None, # target
"ps": None, # target
"ds": None, # target
"std": None, # target
"unkn": None, # target
},
# source
"exp": {
"inf": None, # target
"exp": None, # target
"bcp": None, # target
"ps": None, # target
"ds": None, # target
"std": None, # target
"unkn": None, # target
},
# source
"bcp": {
"inf": "Downref", # target
"exp": "Downref", # target
"bcp": None, # target
"ps": None, # target
"ds": None, # target
"std": None, # target
"unkn": "Possible Downref", # target
},
# source
"ps": {
"inf": "Downref", # target
"exp": "Downref", # target
"bcp": None, # target
"ps": None, # target
"ds": None, # target
"std": None, # target
"unkn": "Possible Downref", # target
},
# source
"ds": {
"inf": "Downref", # target
"exp": "Downref", # target
"bcp": None, # target
"ps": "Downref", # target
"ds": None, # target
"std": None, # target
"unkn": "Possible Downref", # target
},
# source
"std": {
"inf": "Downref", # target
"exp": "Downref", # target
"bcp": None, # target
"ps": "Downref", # target
"ds": "Downref", # target
"std": None, # target
"unkn": "Possible Downref", # target
},
# source
"unkn": {
"inf": None, # target
"exp": None, # target
"bcp": None, # target
"ps": "Possible Downref", # target
"ds": "Possible Downref", # target
"std": None, # target
"unkn": "Possible Downref", # target
},
}
for rel in ["refnorm", "refinfo", "refunk", "refold"]:
for source, target in itertools.product(rfcs, rfcs):
ref = RelatedDocument.objects.create(
source=source,
target=target,
relationship_id=rel,
)
result = ref.is_downref()
desired_result = (
result_matrix[source.std_level_id][target.std_level_id]
if ref.relationship.slug in ["refnorm", "refunk"]
else None
)
if (
ref.relationship.slug == "refunk"
and desired_result is not None
and not desired_result.startswith("Possible")
):
desired_result = f"Possible {desired_result}"
self.assertEqual(desired_result, result)