e7b0c1b554
Fixing issues with the ResourceFinder and the CollectionPlugin not finding files if test-case-level pytest paths were getting used. Change-Id: I317749a0d5f5908105c525f4f33ee786b0ff1a1b Signed-off-by: croy <Christian.Roy@windriver.com>
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
import os
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from framework.database.objects.testcase import TestCase
|
|
|
|
|
|
class CollectionPlugin:
|
|
"""
|
|
Plugin that allows us to get all tests after running a collect only in pytest.
|
|
"""
|
|
|
|
PRIORITIES = ["p0", "p1", "p2", "p3"]
|
|
|
|
def __init__(self, repo_root: str):
|
|
"""
|
|
Constructor.
|
|
|
|
Args:
|
|
repo_root (str): The Absolute path to the root of the repo.
|
|
"""
|
|
self.repo_root: str = repo_root
|
|
self.tests: [TestCase] = []
|
|
|
|
def _get_full_nodeid(self, test: Any) -> str:
|
|
"""
|
|
Ensures the nodeid is relative to the repository root.
|
|
|
|
Args:
|
|
test (Any): The pytest test item.
|
|
|
|
Returns:
|
|
str: Full nodeid relative to repository root.
|
|
"""
|
|
# Get the absolute path of the test file
|
|
abs_path = Path(test.path).absolute()
|
|
repo_root_path = Path(self.repo_root)
|
|
|
|
# Make it relative to repo root
|
|
rel_path = abs_path.relative_to(repo_root_path).as_posix()
|
|
|
|
# Replace the file path portion of nodeid with the repository-relative path
|
|
parts = test.nodeid.split("::")
|
|
return "::".join([rel_path] + parts[1:])
|
|
|
|
def pytest_report_collectionfinish(self, items: Any):
|
|
"""
|
|
Run after collection is finished.
|
|
|
|
Args:
|
|
items (Any): list of Pytest test items.
|
|
"""
|
|
for test in items:
|
|
markers = list(map(lambda marker: marker.name, test.own_markers))
|
|
priority = self.get_testcase_priority(markers)
|
|
if priority:
|
|
markers.remove(priority)
|
|
|
|
full_node_id = self._get_full_nodeid(test)
|
|
testcase = TestCase(test.name, os.path.basename(test.location[0]), priority, test.location[0], full_node_id)
|
|
testcase.set_markers(markers)
|
|
self.tests.append(testcase)
|
|
|
|
def get_tests(self) -> list[TestCase]:
|
|
"""
|
|
Returns the tests.
|
|
|
|
Returns:
|
|
list[TestCase]: List of test cases collected during pytest collection.
|
|
"""
|
|
return self.tests
|
|
|
|
def get_testcase_priority(self, markers: Any) -> str:
|
|
"""
|
|
Gets the testcase priority from the list of markers.
|
|
|
|
Args:
|
|
markers (Any): the pytest markers to find the priority from.
|
|
|
|
Returns:
|
|
str: Testcase priority.
|
|
"""
|
|
for mark in markers:
|
|
if mark in self.PRIORITIES:
|
|
return mark
|